def test_init_defaults(self):
config = FileWriterConfiguration("filename.txt")
self.assertEquals("filename.txt", config.filename)
self.assertEquals(None, config.file_format)
self.assertEquals("a", config.mode)
self.assertEquals("utf-8", config.encoding)
self.assertEquals(False, config.delete_on_start)
def test_init_no_defaults(self):
config = FileWriterConfiguration("filename2.txt", fileformat="txt", mode="r", encoding="ascii", delete_on_start=True)
self.assertEquals("filename2.txt", config.filename)
self.assertEquals("txt", config.file_format)
self.assertEquals("r", config.mode)
self.assertEquals("ascii", config.encoding)
self.assertEquals(True, config.delete_on_start)
def test_init(self):
config = FileWriterConfiguration(filename="filename.test",
file_format="txt",
mode="a",
encoding="utf-8",
delete_on_start=False)
writer = ConversationFileWriter(config)
self.assertIsNotNone(writer)
if os.path.exists("filename.test"):
os.remove("filename.test")
self.assertFalse(os.path.exists("filename.test"))
def test_init(self):
filename = get_os_specific_path() + os.sep + "conversation.txt"
config = FileWriterConfiguration(filename=filename, fileformat="txt", mode="a", encoding="utf-8",
delete_on_start=False)
writer = ConversationFileWriter(config)
self.assertIsNotNone(writer)
writer.log_question_and_answer("client1", "question", "answer")
self.assertTrue(os.path.exists(filename))
if os.path.exists(filename):
os.remove(filename)
self.assertFalse(os.path.exists(filename))
def test_init(self):
config = FileWriterConfiguration(filename="filename.test",
fileformat="txt",
mode="a",
encoding="utf-8",
delete_on_start=False)
writer = ContentFileWriter(config, content_type="txt")
self.assertIsNotNone(writer)
writer.display_debug_info()
if os.path.exists("filename.test"):
os.remove("filename.test")
self.assertFalse(os.path.exists("filename.test"))
def test_init(self):
filename = get_os_specific_path() + os.sep + "duplicates.txt"
config = FileWriterConfiguration(filename=filename,
fileformat="txt",
mode="a",
encoding="utf-8",
delete_on_start=False)
writer = DuplicatesFileWriter(config)
self.assertIsNotNone(writer)
writer.save_entry("<xml>Error with line numbers</xml>", "test.aiml",
100, 666)
writer.save_entry("<xml>Error with no line numbers</xml>",
"test2.aiml")
self.assertEquals(2, len(writer.entries))
writer.save_content()
self.assertTrue(os.path.exists(filename))
if os.path.exists(filename):
os.remove(filename)
self.assertFalse(os.path.exists(filename))