fix: message reply test

ostis-ai · Jun 11, 2024 · b636771 · b636771
1 parent 0c773f9
commit b636771
Showing 1 changed file with 29 additions and 5 deletions.
diff --git a/...t-components/problem-solver/cxx/message-reply-module/test/units/testMessageReplyAgent.cpp b/...t-components/problem-solver/cxx/message-reply-module/test/units/testMessageReplyAgent.cpp
@@ -3,9 +3,11 @@
 #include "sc-memory/kpm/sc_agent.hpp"
 #include "sc-agents-common/keynodes/coreKeynodes.hpp"
 #include "sc-agents-common/utils/AgentUtils.hpp"
+#include "sc-agents-common/utils/IteratorUtils.hpp"
 #include "agent/MessageReplyAgent.hpp"
 #include "keynodes/MessageReplyKeynodes.hpp"
 #include "test/agent/GenerateReplyMessageAgent.hpp"
+#include "utils/sc_log.hpp"
 
 namespace messageReplyModuleTest
 {
@@ -29,6 +31,28 @@ void shutdown()
   SC_AGENT_UNREGISTER(messageReplyModule::MessageReplyAgent)
 }
 
+bool generatedMessageIsValid(ScMemoryContext * context, ScAddr const & textLinkAddr)
+{
+  ScTemplate scTemplate;
+  scTemplate.Triple(
+      messageReplyModule::MessageReplyKeynodes::concept_message,
+      ScType::EdgeAccessVarPosPerm,
+      ScType::NodeVar >> "_user_message");
+  scTemplate.Quintuple(
+      ScType::NodeVar >> "_translation_node",
+      ScType::EdgeDCommonVar,
+      "_user_message",
+      ScType::EdgeAccessVarPosPerm,
+      scAgentsCommon::CoreKeynodes::nrel_sc_text_translation);
+  scTemplate.Triple(
+      "_translation_node",
+      ScType::EdgeAccessVarPosPerm,
+      textLinkAddr);
+  ScTemplateSearchResult searchResult;
+  context->HelperSearchTemplate(scTemplate, searchResult);
+  return searchResult.Size() == 1;
+}
+
 TEST_F(MessageReplyAgentTest, messageProcessingWithTextLinkSuccessful)
 {
   ScMemoryContext & context = *m_ctx;
@@ -42,17 +66,17 @@ TEST_F(MessageReplyAgentTest, messageProcessingWithTextLinkSuccessful)
   ScAddr test_action_node = context.HelperFindBySystemIdtf("test_action_node");
   EXPECT_TRUE(test_action_node.IsValid());
 
-  context.CreateEdge(
-          ScType::EdgeAccessConstPosPerm,
-          scAgentsCommon::CoreKeynodes::question_initiated,
-          test_action_node);
-
   EXPECT_TRUE(utils::AgentUtils::applyAction(&context, test_action_node, WAIT_TIME));
   EXPECT_TRUE(context.HelperCheckEdge(
           scAgentsCommon::CoreKeynodes::question_finished_successfully,
           test_action_node,
           ScType::EdgeAccessConstPosPerm));
 
+  EXPECT_TRUE(generatedMessageIsValid(&context, utils::IteratorUtils::getAnyByOutRelation(
+          &context,
+          test_action_node,
+          scAgentsCommon::CoreKeynodes::rrel_1)));
+
   SC_AGENT_UNREGISTER(GenerateReplyMessageAgent)
   shutdown();
 }