def test_save_braintree(self):
config = FileStorageConfiguration()
config._categories_storage = FileStoreConfiguration(dirs=[os.path.dirname(__file__) + os.sep + "data" + os.sep + "categories"], extension="aiml", subdirs=False, format="xml", encoding="utf-8", delete_on_start=False)
tmpdir = os.path.dirname(__file__) + os.sep + "braintree"
tmpfile = tmpdir + os.sep + "braintree.xml"
config.braintree_storage._dirs = [tmpfile]
config.braintree_storage._has_single_file = True
engine = FileStorageEngine(config)
engine.initialise()
store = FileBraintreeStore(engine)
path = store._get_dir_from_path(store._get_storage_path())
if os.path.exists(path):
shutil.rmtree(path)
self.assertFalse(os.path.exists(path))
test_client = TestClient()
client_context = test_client.create_client_context("test1")
pattern_graph = client_context.brain.aiml_parser.pattern_parser
store.save_braintree(client_context, pattern_graph)
self.assertTrue(os.path.exists(store._get_storage_path()))
shutil.rmtree(tmpdir)
self.assertFalse(os.path.exists(tmpdir))
def test_load_all_from_text_file_diff_extension(self):
config = FileStorageConfiguration()
file = os.path.dirname(__file__) + os.sep + "data" + os.sep + "categories" + os.sep + "kinghorn.aiml"
self.assertTrue(os.path.exists(file))
config._categories_storage = FileStoreConfiguration(file=file, extension="xml", fileformat="xml", encoding="utf-8", delete_on_start=False)
engine = FileStorageEngine(config)
engine.initialise()
store = FileCategoryStore(engine)
parser = MockAIMLParser()
store.load_all(parser)
self.assertEqual(0, len(parser._parsed_files))
def test_load_all_from_test_dir_no_subdir(self):
config = FileStorageConfiguration()
dirs = os.path.dirname(__file__) + os.sep + "data" + os.sep + "categories"
self.assertTrue(os.path.exists(dirs))
config._categories_storage = FileStoreConfiguration(dirs=[dirs], extension="aiml", subdirs=False, fileformat="xml", encoding="utf-8", delete_on_start=False)
engine = FileStorageEngine(config)
engine.initialise()
store = FileCategoryStore(engine)
parser = MockAIMLParser()
store.load_all(parser)
self.assertEqual(1, len(parser._parsed_files))
self.assertTrue(parser._parsed_files[0].endswith("/kinghorn.aiml"))
