def load_storage(self):
super(BinariesTestClient, self).load_storage()
self.add_default_stores()
bot_config = self.configuration.client_configuration.configurations[0]
brain_config = bot_config.configurations[0]
brain_config.binaries._load_binary = self._load_bin
brain_config.binaries._save_binary = self._save_bin
aimlfile = os.path.dirname(__file__) + os.sep + self.TEST_AIML_FILE
self._file_store_config._categories_storage = FileStoreConfiguration(
dirs=aimlfile,
format="xml",
extension="aiml",
encoding="utf-8",
delete_on_start=False)
self.storage_factory._storage_engines[
StorageFactory.CATEGORIES] = self._storage_engine
self.storage_factory._store_to_engine_map[
StorageFactory.CATEGORIES] = self._storage_engine
config = FileStoreConfiguration()
config._dirs = [self._filename]
config._has_single_file = True
self._file_store_config._binaries_storage = config
self.storage_factory._storage_engines[
StorageFactory.BINARIES] = self._storage_engine
self.storage_factory._store_to_engine_map[
StorageFactory.BINARIES] = self._storage_engine
def load_storage(self):
super(BrainTreeTestClient, self).load_storage()
self.add_default_stores()
aimlfile = os.path.dirname(__file__) + os.sep + self.TEST_AIML_FILE
self._file_store_config._categories_storage = FileStoreConfiguration(
dirs=aimlfile,
format="xml",
extension="aiml",
encoding="utf-8",
delete_on_start=False)
self.storage_factory._storage_engines[
StorageFactory.CATEGORIES] = self._storage_engine
self.storage_factory._store_to_engine_map[
StorageFactory.CATEGORIES] = self._storage_engine
config = FileStoreConfiguration()
config._format = self._tree_type
config._dirs = [self._filename]
config._has_single_file = True
self._file_store_config._braintree_storage = config
self.storage_factory._storage_engines[
StorageFactory.BRAINTREE] = self._storage_engine
self.storage_factory._store_to_engine_map[
StorageFactory.BRAINTREE] = self._storage_engine
def add_duplicates_store(self):
config = FileStoreConfiguration()
tmpdir = os.path.dirname(__file__) + os.sep + self.TEST_DIRECTORY
tmpfile = tmpdir + os.sep + self.DUPLICATES_FILE
config._dirs = [tmpfile]
config._has_single_file = True
self._file_store_config._duplicates_storage = config
self.storage_factory._storage_engines[
StorageFactory.DUPLICATES] = self._storage_engine
self.storage_factory._store_to_engine_map[
StorageFactory.DUPLICATES] = self._storage_engine
