def test_init_context_and_converstion(self):
client = TestClient()
client_context = client.create_client_context("tesuser")
convo = Convo(client_context)
question = Question.create_from_text(client_context, "Hello world")
question.current_sentence()._response = "Hello matey"
convo.record_dialog(question)
conversation = Conversation(client_context, convo)
self.assertIsNotNone(conversation)
self.assertEqual(client_context.client.id, conversation.clientid)
self.assertEqual(client_context.userid, conversation.userid)
self.assertEqual(client_context.bot.id, conversation.botid)
self.assertEqual(client_context.brain.id, conversation.brainid)
self.assertIsNotNone(conversation.conversation)
doc = conversation.to_document()
self.assertEqual({'clientid': 'testclient', 'userid': 'tesuser', 'botid': 'bot', 'brainid': 'brain', 'conversation': {'client_context': {'clientid': 'testclient', 'userid': 'tesuser', 'botid': 'bot', 'brainid': 'brain', 'depth': 0}, 'questions': [{'sentences': [{'question': 'Hello world', 'response': 'Hello matey'}], 'srai': False, 'properties': {}, 'current_sentence_no': -1}], 'max_histories': 100, 'properties': {'topic': '*'}}},
doc)
conversation2 = Conversation.from_document(client_context, doc)
self.assertIsNotNone(conversation2)
self.assertEqual(conversation2.clientid, conversation.clientid)
self.assertEqual(conversation2.userid, conversation.userid)
self.assertEqual(conversation2.botid, conversation.botid)
self.assertEqual(conversation2.brainid, conversation.brainid)
def test_to_document_with_id(self):
client = TestClient()
client_context = client.create_client_context("testuser")
convo = Convo(client_context)
question = Question.create_from_text(client_context, "Hello world")
question.current_sentence()._response = "Hello matey"
convo.record_dialog(question)
conversation = Conversation(client_context, convo)
conversation.id = '666'
doc = conversation.to_document()
self.assertEqual(
{
'_id': '666',
'clientid': 'testclient',
'userid': 'testuser',
'botid': 'bot',
'brainid': 'brain',
'conversation': {
'client_context': {
'clientid': 'testclient',
'userid': 'testuser',
'botid': 'bot',
'brainid': 'brain',
'depth': 0
},
'questions': [{
'srai':
False,
'sentences': [{
'words': ['Hello', 'world'],
'response': 'Hello matey',
'positivity': 0.0,
'subjectivity': 0.5
}],
'current_sentence_no':
-1,
'properties': {}
}],
'max_histories':
100,
'properties': {
'topic': '*'
}
}
}, doc)
def test_repr_with_id(self):
client = TestClient()
client_context = client.create_client_context("testuser")
convo = Convo(client_context)
question = Question.create_from_text(client_context, "Hello world")
question.current_sentence()._response = "Hello matey"
convo.record_dialog(question)
conversation = Conversation(client_context, convo)
conversation.id = '1'
self.assertEquals(
"<Conversation(id='1', client='testclient', user='******', bot='bot', brain='brain')",
str(conversation))
def test_init_blank(self):
conversation1 = Conversation(None, None)
self.assertIsNotNone(conversation1)
self.assertIsNone(conversation1.clientid)
self.assertIsNone(conversation1.userid)
self.assertIsNone(conversation1.botid)
self.assertIsNone(conversation1.brainid)
self.assertIsNone(conversation1.conversation)
def test_from_document_no_data(self):
client = TestClient()
client_context = client.create_client_context("testuser")
data = {}
dao = Conversation.from_document(client_context, data)
self.assertIsNotNone(dao)
self.assertIsNone(dao.id)
self.assertEquals("testclient", dao.clientid)
self.assertEquals("testuser", dao.userid)
self.assertEquals("bot", dao.botid)
self.assertEquals("brain", dao.brainid)
self.assertIsNone(dao.conversation)
def test_from_document_data(self):
client = TestClient()
client_context = client.create_client_context("testuser")
data = {
"_id": "1",
"clientid": "client1",
"userid": "user1",
"botid": "bot1",
"brainid": "brain1"
}
dao = Conversation.from_document(client_context, data)
self.assertIsNotNone(dao)
self.assertIsNone(dao.conversation)
self.assertEquals("client1", dao.clientid)
self.assertEquals("user1", dao.userid)
self.assertEquals("bot1", dao.botid)
self.assertEquals("brain1", dao.brainid)
self.assertIsNone(dao.conversation)
def store_conversation(self, client_context, conversation):
YLogger.info(client_context, "Storing conversation to Mongo [%s] [%s]", client_context.client.id, client_context.userid)
return self.add_document(Conversation(client_context, conversation))
