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