def test_get_all(self):
config = SQLStorageConfiguration()
engine = SQLStorageEngine(config)
engine.initialise()
store = SQLConversationStore(engine)
with self.assertRaises(Exception):
store._get_all()
def test_storage_where_no_property(self):
config = SQLStorageConfiguration()
engine = SQLStorageEngine(config)
engine.initialise()
store = SQLConversationStore(engine)
self.assertEqual(store.storage_engine, engine)
self.assert_just_conversation_storage(store)
def conversation_store(self):
return SQLConversationStore(self)
def test_read_write_question_properties_in_db(self):
config = SQLStorageConfiguration()
engine = SQLStorageEngine(config)
engine.initialise()
store = SQLConversationStore(engine)
client = TestClient()
client_context = client.create_client_context("user1")
store.empty()
properties1 = {"key1": "value1", "key2": "value2"}
store._write_properties_to_db(client_context, 1, 2,
ConversationPropertyDAO.QUESTION,
properties1)
store.commit()
properties2 = {}
store._read_properties_from_db(client_context, 1, 2,
ConversationPropertyDAO.QUESTION,
properties2)
self.assertEqual({"key1": "value1", "key2": "value2"}, properties2)
store.empty()
def test_conversation_in_db(self):
config = SQLStorageConfiguration()
engine = SQLStorageEngine(config)
engine.initialise()
store = SQLConversationStore(engine)
store.empty()
client = TestClient()
client_context = client.create_client_context("user1")
conversation1 = Conversation(client_context)
conversation1.properties['ckey1'] = "cvalue1"
conversation1.properties['ckey2'] = "cvalue2"
question1 = Question.create_from_text(client_context, "Hello There")
question1.sentence(0).response = "Hi"
question1.sentence(0)._positivity = 0.5
question1.sentence(0)._subjectivity = 0.6
question1.properties['qkey1'] = "qvalue1"
question1.properties['qkey2'] = "qvalue2"
conversation1.record_dialog(question1)
store.store_conversation(client_context, conversation1)
store.commit()
conversation2 = Conversation(client_context)
store.load_conversation(client_context, conversation2)
self.assertEqual(conversation2.properties['ckey1'], "cvalue1")
self.assertEqual(conversation2.properties['ckey2'], "cvalue2")
self.assertEqual(conversation2.questions[0].sentence(0).response, "Hi")
self.assertEqual(conversation2.questions[0].sentence(0)._positivity,
0.5)
self.assertEqual(conversation2.questions[0].sentence(0)._subjectivity,
0.6)
self.assertEqual(conversation2.questions[0].properties['qkey1'],
"qvalue1")
self.assertEqual(conversation2.questions[0].properties['qkey2'],
"qvalue2")
store.empty()
def test_initialise(self):
config = SQLStorageConfiguration()
engine = SQLStorageEngine(config)
engine.initialise()
store = SQLConversationStore(engine)
self.assertEqual(store.storage_engine, engine)
def test_questions_in_db(self):
config = SQLStorageConfiguration()
engine = SQLStorageEngine(config)
engine.initialise()
store = SQLConversationStore(engine)
store.empty()
client = TestClient()
client_context = client.create_client_context("user1")
conversation1 = Conversation(client_context)
question1 = Question.create_from_text(client_context, "Hello There")
question1.sentence(0).response = "Hi"
question1.sentence(0)._positivity = 0.5
question1.sentence(0)._subjectivity = 0.6
conversation1.record_dialog(question1)
store._write_questions_to_db(client_context, 1, conversation1)
store.commit()
conversation2 = Conversation(client_context)
store._read_questions_from_db(client_context, 1, conversation2)
self.assertEqual(1, len(conversation2.questions))
self.assertEqual(1, len(conversation2.questions[0].sentences))
self.assertEqual(0.5,
conversation2.questions[0].sentences[0].positivity)
self.assertEqual(0.6,
conversation2.questions[0].sentences[0].subjectivity)
self.assertEqual(["Hello", "There"],
conversation2.questions[0].sentences[0].words)
self.assertEqual("Hi",
conversation2.questions[0].sentences[0].response)
store.empty()
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_read_write_matches_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.append(
Match(Match.WORD, PatternWordNode("Hello"), "Hello"))
store._write_matches_to_db(client_context, matched_context1, 1)
store.commit()
matched_context2 = MatchContext(0, 0)
store._read_matches_from_db(client_context, matched_context2, 1)
self.assertEqual(1, len(matched_context2.matched_nodes))
self.assertEqual(Match.WORD,
matched_context2.matched_nodes[0].matched_node_type)
self.assertEqual("WORD [Hello]",
matched_context2.matched_nodes[0].matched_node_str)
self.assertFalse(
matched_context2.matched_nodes[0].matched_node_multi_word)
self.assertFalse(
matched_context2.matched_nodes[0].matched_node_wildcard)
self.assertEqual(
1, len(matched_context2.matched_nodes[0].matched_node_words))
self.assertEqual(["Hello"],
matched_context2.matched_nodes[0].matched_node_words)
store.empty()
