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"))