def test_storage_path(self):
config = FileStorageConfiguration()
engine = FileStorageEngine(config)
engine.initialise()
store = FileConversationStore(engine)
self.assertEquals('/tmp/conversations', store._get_storage_path())
self.assertIsInstance(store.get_storage(), FileStoreConfiguration)
def tests_conversation_storage(self):
config = FileStorageConfiguration()
config.conversation_storage._dirs = [self._tmpdir]
engine = FileStorageEngine(config)
engine.initialise()
store = FileConversationStore(engine)
self.assertEqual(store.storage_engine, engine)
self.assert_conversation_storage(store, can_empty=True, test_load=True)
store.empty()
def tests_debug_conversation(self):
self.tearDown()
config = FileStorageConfiguration()
config.conversation_storage._dirs = [self._tmpdir]
engine = FileStorageEngine(config)
engine.initialise()
store = FileConversationStore(engine)
self.assertEqual(store.storage_engine, engine)
self.assert_debug_conversation(store)
store.empty()
def tests_debug_conversation_no_file(self):
self.tearDown()
config = FileStorageConfiguration()
config.conversation_storage._dirs = [self._tmpdir]
engine = FileStorageEngine(config)
engine.initialise()
store = FileConversationStore(engine)
self.assertEqual(store.storage_engine, engine)
client = TestClient()
client_context = client.create_client_context("user1")
debugInfo, _ = store.debug_conversation_data(client_context)
self.assertEqual(0, len(debugInfo))
def test_initialise(self):
config = FileStorageConfiguration()
config.conversation_storage._dirs = [self._tmpdir]
engine = FileStorageEngine(config)
engine.initialise()
store = FileConversationStore(engine)
self.assertEqual(store.storage_engine, engine)
def tests_conversation_storage(self):
config = FileStorageConfiguration()
tmpdir = os.path.dirname(__file__) + os.sep + "conversations"
config.conversation_storage._dirs = [tmpdir]
engine = FileStorageEngine(config)
engine.initialise()
store = FileConversationStore(engine)
self.assertEqual(store.storage_engine, engine)
self.assert_conversation_storage(store, can_empty=True, test_load=True)
shutil.rmtree(tmpdir)
self.assertFalse(os.path.exists(tmpdir))
def test_load_conversation_with_exception(self):
config = FileStorageConfiguration()
config.conversation_storage._dirs = [self._tmpdir]
engine = FileStorageEngine(config)
engine.initialise()
store = FileConversationStore(engine)
store.empty()
client = TestClient()
client_context = client.create_client_context("user1")
conversation = Conversation(client_context)
question1 = Question.create_from_text(client_context, "Hello There")
question1.sentence(0).response = "Hi"
conversation.record_dialog(question1)
store.store_conversation(client_context, conversation)
conversation = Conversation(client_context)
store.load_conversation(client_context, conversation)
def tests_conversation_storage_no_file(self):
self.tearDown()
config = FileStorageConfiguration()
config.conversation_storage._dirs = [self._tmpdir]
engine = FileStorageEngine(config)
engine.initialise()
store = FileConversationStore(engine)
self.assertEqual(store.storage_engine, engine)
client = TestClient()
client_context = client.create_client_context("user1")
conversation = Conversation(client_context)
store.load_conversation(client_context, conversation)
self.assertEqual(0, len(conversation.questions))
store.empty()
def conversation_store(self):
return FileConversationStore(self)