def test_bot_chat_loop(self):
client = TestClient()
bot = Bot(BotConfiguration(), client)
self.assertIsNotNone(bot)
self.assertIsInstance(bot, Bot)
bot.configuration._default_response = "Sorry, I don't have an answer for that right now"
response = bot.ask_question(self._client_context, "hello")
self.assertIsNotNone(response)
self.assertEqual(response, "Sorry, I don't have an answer for that right now.")
response = bot.ask_question(self._client_context, "hello again")
self.assertIsNotNone(response)
self.assertEqual(response, "Sorry, I don't have an answer for that right now.")
response = bot.ask_question(self._client_context, "goodbye")
self.assertIsNotNone(response)
self.assertEqual(response, "Sorry, I don't have an answer for that right now.")
conversation = bot.get_conversation(self._client_context)
self.assertIsNotNone(conversation)
self.assertEqual(conversation.previous_nth_question(2).sentence(0).text(), "hello")
self.assertEqual(conversation.previous_nth_question(2).sentence(0).response, "Sorry, I don't have an answer for that right now")
self.assertEqual(conversation.previous_nth_question(1).sentence(0).text(), "hello again")
self.assertEqual(conversation.previous_nth_question(1).sentence(0).response, "Sorry, I don't have an answer for that right now")
self.assertEqual(conversation.previous_nth_question(0).sentence(0).text(), "goodbye")
self.assertEqual(conversation.previous_nth_question(0).sentence(0).response, "Sorry, I don't have an answer for that right now")
def test_ask_question_without_triggermgr(self):
bot_config = BotConfiguration()
client = TestClient()
client._trigger_mgr = None
bot = Bot(bot_config, client)
bot.ask_question(self._client_context, "HELLO WORLD")
def test_bot_chat_loop(self):
test_brain = Brain(BrainConfiguration())
self.assertIsNotNone(test_brain)
self.assertIsInstance(test_brain, Brain)
bot = Bot(test_brain, BotConfiguration())
self.assertIsNotNone(bot)
self.assertIsInstance(bot, Bot)
bot.configuration._default_response = "Sorry, I don't have an answer for that right now"
response = bot.ask_question("testid", "hello")
self.assertIsNotNone(response)
self.assertEqual(response, "Sorry, I don't have an answer for that right now")
response = bot.ask_question("testid", "hello again")
self.assertIsNotNone(response)
self.assertEqual(response, "Sorry, I don't have an answer for that right now")
response = bot.ask_question("testid", "goodbye")
self.assertIsNotNone(response)
self.assertEqual(response, "Sorry, I don't have an answer for that right now")
conversation = bot.get_conversation("testid")
self.assertIsNotNone(conversation)
self.assertEqual(conversation.nth_question(3).sentence(0).text(), "hello")
self.assertEqual(conversation.nth_question(3).sentence(0).response, "Sorry, I don't have an answer for that right now")
self.assertEqual(conversation.nth_question(2).sentence(0).text(), "hello again")
self.assertEqual(conversation.nth_question(2).sentence(0).response, "Sorry, I don't have an answer for that right now")
self.assertEqual(conversation.nth_question(1).sentence(0).text(), "goodbye")
self.assertEqual(conversation.nth_question(1).sentence(0).response, "Sorry, I don't have an answer for that right now")
def test_bot_chat_loop(self):
bot = Bot(BotConfiguration())
self.assertIsNotNone(bot)
self.assertIsInstance(bot, Bot)
bot.configuration._default_response = "Sorry, I don't have an answer for that right now"
response = bot.ask_question(self._client_context, "hello")
self.assertIsNotNone(response)
self.assertEqual(response, "Sorry, I don't have an answer for that right now")
response = bot.ask_question(self._client_context, "hello again")
self.assertIsNotNone(response)
self.assertEqual(response, "Sorry, I don't have an answer for that right now")
response = bot.ask_question(self._client_context, "goodbye")
self.assertIsNotNone(response)
self.assertEqual(response, "Sorry, I don't have an answer for that right now")
conversation = bot.get_conversation(self._client_context)
self.assertIsNotNone(conversation)
self.assertEqual(conversation.previous_nth_question(2).sentence(0).text(), "hello")
self.assertEqual(conversation.previous_nth_question(2).sentence(0).response, "Sorry, I don't have an answer for that right now")
self.assertEqual(conversation.previous_nth_question(1).sentence(0).text(), "hello again")
self.assertEqual(conversation.previous_nth_question(1).sentence(0).response, "Sorry, I don't have an answer for that right now")
self.assertEqual(conversation.previous_nth_question(0).sentence(0).text(), "goodbye")
self.assertEqual(conversation.previous_nth_question(0).sentence(0).response, "Sorry, I don't have an answer for that right now")
def test_bot_chat_loop(self):
test_brain = TestBrain(BrainConfiguration())
self.assertIsNotNone(test_brain)
self.assertIsInstance(test_brain, Brain)
bot = Bot(test_brain, BotConfiguration())
self.assertIsNotNone(bot)
self.assertIsInstance(bot, Bot)
test_brain.setresponse ("response1")
response = bot.ask_question("testid", "hello")
self.assertIsNotNone(response)
self.assertEqual(response, "response1")
test_brain.setresponse ("response2")
response = bot.ask_question("testid", "hello again")
self.assertIsNotNone(response)
self.assertEqual(response, "response2")
test_brain.setresponse("response3")
response = bot.ask_question("testid", "goodbye")
self.assertIsNotNone(response)
self.assertEqual(response, "response3")
conversation = bot.get_conversation("testid")
self.assertIsNotNone(conversation)
self.assertEqual(conversation.nth_question(3).sentence(0).text(), "hello")
self.assertEqual(conversation.nth_question(3).sentence(0).response, "response1")
self.assertEqual(conversation.nth_question(2).sentence(0).text(), "hello again")
self.assertEqual(conversation.nth_question(2).sentence(0).response, "response2")
self.assertEqual(conversation.nth_question(1).sentence(0).text(), "goodbye")
self.assertEqual(conversation.nth_question(1).sentence(0).response, "response3")
def test_handle_none_respons_withtriggermgr(self):
bot_config = BotConfiguration()
client = TestClient()
client._trigger_mgr = MockTriggerManager()
bot = Bot(bot_config, client)
bot.ask_question(self._client_context, "HELLO WORLD")
self.assertTrue(client._trigger_mgr._triggered)
def test_max_recusion(self):
bot = Bot(BotConfiguration())
self.assertIsNotNone(bot)
bot.configuration._default_response = "Sorry, I don't have an answer for that right now"
bot.configuration._max_question_recursion = 0
with self.assertRaises(Exception):
bot.ask_question(self._client_context, "hello")
def test_max_recusion(self):
bot = Bot(BotConfiguration())
self.assertIsNotNone(bot)
bot.configuration._default_response = "Sorry, I don't have an answer for that right now"
bot.configuration._max_question_recursion = 0
with self.assertRaises(Exception):
bot.ask_question(self._client_context, "hello")
def test_max_recusion(self):
client = TestClient()
bot = Bot(BotConfiguration(), client)
self.assertIsNotNone(bot)
bot.configuration._default_response = "Sorry, I don't have an answer for that right now"
bot.configuration._max_question_recursion = 0
bot.ask_question(self._client_context, "hello")
self.assertEqual("Maximum recursion limit [0] exceeded", bot.get_conversation(self._client_context)._exception)
def test_delete_valiables_srai_false(self):
client = TestClient()
bot = Bot(BotConfiguration(), client)
self.assertIsNotNone(bot)
self._client_context.deleteVariable = True
conversation = bot.get_conversation(self._client_context)
conversation.set_data_property("test", "value")
bot.ask_question(self._client_context, "hello", srai=False)
self.assertIsNone(conversation.data_property("test"))
def test_max_recusion(self):
test_brain = Brain(BrainConfiguration())
self.assertIsNotNone(test_brain)
bot = Bot(test_brain, BotConfiguration())
self.assertIsNotNone(bot)
bot.configuration._default_response = "Sorry, I don't have an answer for that right now"
bot.configuration._max_question_recursion = 0
with self.assertRaises(Exception):
bot.ask_question("testid", "hello")
def test_bot_with_conversation(self):
test_brain = Brain(BrainConfiguration())
self.assertIsNotNone(test_brain)
bot = Bot(test_brain, BotConfiguration())
self.assertIsNotNone(bot)
self.assertFalse(bot.has_conversation("testid"))
response = bot.ask_question("testid", "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation("testid"))
response = bot.ask_question("testid", "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation("testid"))
response = bot.ask_question("testid2", "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation("testid2"))
def test_bot_with_conversation(self):
test_brain = TestBrain(BrainConfiguration())
self.assertIsNotNone(test_brain)
bot = Bot(test_brain, BotConfiguration())
self.assertIsNotNone(bot)
self.assertFalse(bot.has_conversation("testid"))
response = bot.ask_question("testid", "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation("testid"))
response = bot.ask_question("testid", "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation("testid"))
response = bot.ask_question("testid2", "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation("testid2"))
def test_bot_with_conversation(self):
client = TestClient()
bot = Bot(BotConfiguration(), client)
self.assertIsNotNone(bot)
self.assertFalse(bot.has_conversation(self._client_context))
response = bot.ask_question(self._client_context, "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation(self._client_context))
response = bot.ask_question(self._client_context, "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation(self._client_context))
client_context2 = ClientContext(TestClient(), "testid2")
client_context2._bot = bot
client_context2._brain = self._client_context.bot.brain
response = bot.ask_question(client_context2, "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation(client_context2))
def test_bot_with_conversation(self):
bot = Bot(BotConfiguration())
self.assertIsNotNone(bot)
self.assertFalse(bot.has_conversation(self._client_context))
response = bot.ask_question(self._client_context, "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation(self._client_context))
response = bot.ask_question(self._client_context, "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation(self._client_context))
client_context2 = ClientContext(TestClient(), "testid2")
client_context2._bot = bot
client_context2._brain = self._client_context.bot.brain
response = bot.ask_question(client_context2, "hello")
self.assertIsNotNone(response)
self.assertTrue(bot.has_conversation(client_context2))
class TemplateGraphLearnTests(unittest.TestCase):
def setUp(self):
self.parser = TemplateGraph(AIMLParser())
self.assertIsNotNone(self.parser)
self.test_brain = None
self.test_sentence = Sentence("test sentence")
self.test_sentence._stars = [
'one', 'two', 'three', 'four', 'five', 'six'
]
self.test_sentence._thatstars = ["*"]
self.test_sentence._topicstars = ["*"]
test_config = ClientConfiguration()
self.test_bot = Bot(Brain(BrainConfiguration()),
config=test_config.bot_configuration)
self.test_clientid = "testid"
conversation = self.test_bot.get_conversation(self.test_clientid)
question = Question.create_from_sentence(self.test_sentence)
conversation._questions.append(question)
def test_eval_simple(self):
template = ET.fromstring("""
<template>
<eval>sometext</eval>
</template>
""")
ast = self.parser.parse_template_expression(template)
self.assertIsNotNone(ast)
self.assertIsInstance(ast, TemplateNode)
self.assertIsNotNone(ast.children)
self.assertEqual(len(ast.children), 1)
eval_node = ast.children[0]
self.assertIsNotNone(eval_node)
self.assertIsInstance(eval_node, TemplateEvalNode)
self.assertIsNotNone(eval_node.children)
self.assertEqual(len(eval_node.children), 1)
self.assertIsInstance(eval_node.children[0], TemplateWordNode)
self.assertEqual(eval_node.resolve(self.test_bot, self.test_clientid),
"sometext")
def test_learn_simple(self):
template = ET.fromstring("""
<template>
<learn>
<category>
<pattern>HELLO <eval>WORLD</eval> THERE</pattern>
<template>HIYA</template>
</category>
</learn>
</template>
""")
ast = self.parser.parse_template_expression(template)
self.assertIsNotNone(ast)
self.assertIsInstance(ast, TemplateNode)
self.assertIsNotNone(ast.children)
self.assertEqual(len(ast.children), 1)
learn_node = ast.children[0]
self.assertIsNotNone(learn_node)
self.assertIsInstance(learn_node, TemplateLearnNode)
self.assertIsNotNone(learn_node._pattern)
self.assertIsInstance(learn_node._pattern, ET.Element)
self.assertIsNotNone(learn_node._topic)
self.assertIsInstance(learn_node._topic, ET.Element)
self.assertIsNotNone(learn_node._that)
self.assertIsInstance(learn_node._that, ET.Element)
self.assertIsNotNone(learn_node._template)
self.assertIsInstance(learn_node._template, TemplateNode)
resolved = learn_node.resolve(self.test_bot, self.test_clientid)
self.assertEqual(resolved, "")
response = self.test_bot.ask_question(self.test_clientid,
"HELLO WORLD THERE")
self.assertEqual("HIYA", response)
def test_learnf_simple(self):
template = ET.fromstring("""
<template>
<learnf>
<category>
<pattern>HELLO <eval>WORLD</eval> THERE</pattern>
<template>HIYA</template>
</category>
</learnf>
</template>
""")
self.test_bot.brain._configuration._aiml_files = BrainFileConfiguration(
"/tmp", ".aiml", False)
ast = self.parser.parse_template_expression(template)
self.assertIsNotNone(ast)
self.assertIsInstance(ast, TemplateNode)
self.assertIsNotNone(ast.children)
self.assertEqual(len(ast.children), 1)
learn_node = ast.children[0]
self.assertIsNotNone(learn_node)
self.assertIsInstance(learn_node, TemplateLearnNode)
self.assertIsNotNone(learn_node._pattern)
self.assertIsInstance(learn_node._pattern, ET.Element)
self.assertIsNotNone(learn_node._topic)
self.assertIsInstance(learn_node._topic, ET.Element)
self.assertIsNotNone(learn_node._that)
self.assertIsInstance(learn_node._that, ET.Element)
self.assertIsNotNone(learn_node._template)
self.assertIsInstance(learn_node._template, TemplateNode)
resolved = learn_node.resolve(self.test_bot, self.test_clientid)
self.assertEqual(resolved, "")
response = self.test_bot.ask_question(self.test_clientid,
"HELLO WORLD THERE")
self.assertEqual("HIYA", response)
