def test_match_context_in_db(self):
config = SQLStorageConfiguration()
engine = SQLStorageEngine(config)
engine.initialise()
store = SQLConversationStore(engine)
store.empty()
client = TestClient()
client_context = client.create_client_context("user1")
matched_context1 = MatchContext(100,
100,
sentence="Hello",
response="Hi There")
matched_context1._matched_nodes = []
matched_context1._template_node = None
store._write_match_context_to_db(client_context, 1, matched_context1)
store.commit()
matched_context2 = MatchContext(100, 100)
store._read_match_context_from_db(client_context, 1, matched_context2)
self.assertEqual(100, matched_context2.max_search_timeout)
self.assertEqual(100, matched_context2.max_search_depth)
self.assertEqual("Hello", matched_context2.sentence)
self.assertEqual("Hi There", matched_context2.response)
store.empty()
def test_get_set_matched_nodes(self):
context = MatchContext(max_search_depth=100, max_search_timeout=60)
context._matched_nodes = [
Match(Match.WORD, PatternWordNode("Hello"), "Hello"),
Match(Match.WORD, PatternWordNode("There"), "There")
]
self.assertEqual(2, len(context.matched_nodes))
matched = context.matched_nodes
self.assertEqual(2, len(matched))