| // Copyright 2018 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "peridot/bin/suggestion_engine/ranking_features/mod_pair_ranking_feature.h" |
| |
| #include <lib/fxl/files/file.h> |
| #include <lib/fxl/files/path.h> |
| #include <lib/fxl/files/scoped_temp_dir.h> |
| |
| #include "gtest/gtest.h" |
| |
| namespace modular { |
| namespace { |
| |
| constexpr char kTestData[] = R"({ |
| "mod1": { |
| "mod2": 0.5, |
| "mod3": 0.5 |
| }, |
| "mod2": { |
| "mod3": 1.0 |
| }, |
| "mod3": { |
| "mod1": 0.2, |
| "mod4": 0.8 |
| } |
| })"; |
| |
| class ModPairRankingFeatureTest : public ::testing::Test { |
| public: |
| void SetUp() override { |
| std::string tmp_file; |
| ASSERT_TRUE(CreateFile(kTestData, &tmp_file)); |
| mod_pair_feature.LoadDataFromFile(tmp_file); |
| } |
| |
| protected: |
| ModPairRankingFeature mod_pair_feature{false}; |
| fuchsia::modular::UserInput query; |
| |
| private: |
| bool CreateFile(const std::string& content, std::string* const tmp_file) { |
| if (!tmp_dir_.NewTempFile(tmp_file)) { |
| return false; |
| } |
| return files::WriteFile(*tmp_file, content.c_str(), content.size()); |
| } |
| |
| files::ScopedTempDir tmp_dir_; |
| }; |
| |
| // Creates the values from a context query to mock the modules in a focused |
| // story based on which this ranking feature computes its value. |
| void AddValueToContextUpdate( |
| fidl::VectorPtr<fuchsia::modular::ContextValue>& context_update, |
| const std::string& mod) { |
| fuchsia::modular::ContextValue value; |
| value.meta.mod = fuchsia::modular::ModuleMetadata::New(); |
| value.meta.mod->url = mod; |
| context_update.push_back(std::move(value)); |
| } |
| |
| TEST_F(ModPairRankingFeatureTest, ComputeFeatureAddModuleAction) { |
| fuchsia::modular::Intent intent; |
| intent.handler = "mod4"; |
| fuchsia::modular::AddMod add_mod; |
| add_mod.intent = std::move(intent); |
| fuchsia::modular::StoryCommand command; |
| command.set_add_mod(std::move(add_mod)); |
| fuchsia::modular::Proposal proposal; |
| proposal.on_selected.push_back(std::move(command)); |
| SuggestionPrototype prototype; |
| prototype.proposal = std::move(proposal); |
| RankedSuggestion suggestion; |
| suggestion.prototype = &prototype; |
| |
| fidl::VectorPtr<fuchsia::modular::ContextValue> context_update; |
| AddValueToContextUpdate(context_update, "mod3"); |
| mod_pair_feature.UpdateContext(std::move(context_update)); |
| double value = mod_pair_feature.ComputeFeature(query, suggestion); |
| EXPECT_EQ(value, 0.8); |
| } |
| |
| TEST_F(ModPairRankingFeatureTest, ComputeFeatureNoModule) { |
| fuchsia::modular::Intent intent; |
| intent.handler = "mod-fiction"; |
| fuchsia::modular::AddMod add_mod; |
| add_mod.intent = std::move(intent); |
| fuchsia::modular::StoryCommand command; |
| command.set_add_mod(std::move(add_mod)); |
| fuchsia::modular::Proposal proposal; |
| proposal.on_selected.push_back(std::move(command)); |
| SuggestionPrototype prototype; |
| prototype.proposal = std::move(proposal); |
| RankedSuggestion suggestion; |
| suggestion.prototype = &prototype; |
| |
| fidl::VectorPtr<fuchsia::modular::ContextValue> context_update; |
| AddValueToContextUpdate(context_update, "mod1"); |
| mod_pair_feature.UpdateContext(std::move(context_update)); |
| double value = mod_pair_feature.ComputeFeature(query, suggestion); |
| EXPECT_EQ(value, kMinConfidence); |
| } |
| |
| TEST_F(ModPairRankingFeatureTest, ComputeFeatureMultipleActions) { |
| fuchsia::modular::AddMod add_mod; |
| add_mod.intent.handler = "mod-fiction"; |
| fuchsia::modular::StoryCommand command; |
| command.set_add_mod(std::move(add_mod)); |
| fuchsia::modular::Proposal proposal; |
| proposal.on_selected.push_back(std::move(command)); |
| SuggestionPrototype prototype; |
| prototype.proposal = std::move(proposal); |
| RankedSuggestion suggestion; |
| suggestion.prototype = &prototype; |
| |
| add_mod = fuchsia::modular::AddMod(); |
| add_mod.intent.handler = "mod3"; |
| fuchsia::modular::StoryCommand command2; |
| command2.set_add_mod(std::move(add_mod)); |
| suggestion.prototype->proposal.on_selected.push_back(std::move(command2)); |
| |
| fidl::VectorPtr<fuchsia::modular::ContextValue> context_update; |
| AddValueToContextUpdate(context_update, "mod1"); |
| AddValueToContextUpdate(context_update, "mod2"); |
| mod_pair_feature.UpdateContext(std::move(context_update)); |
| double value = mod_pair_feature.ComputeFeature(query, suggestion); |
| EXPECT_EQ(value, 1.0); |
| } |
| |
| TEST_F(ModPairRankingFeatureTest, CreateContextSelector) { |
| auto selector = mod_pair_feature.CreateContextSelector(); |
| EXPECT_NE(selector, nullptr); |
| EXPECT_EQ(selector->type, fuchsia::modular::ContextValueType::MODULE); |
| EXPECT_EQ(selector->meta->story->focused->state, |
| fuchsia::modular::FocusedStateState::FOCUSED); |
| } |
| |
| } // namespace |
| } // namespace modular |