def test_should_download_registered_mdb_embedding(
self, download_manager: MagicMock,
embedding_manager: EmbeddingManager):
embedding_manager.add_embedding_config({
'name': EMBEDDING_NAME_1,
'url': EXTERNAL_MDB_GZ_URL_1
})
assert embedding_manager.ensure_available(
EMBEDDING_NAME_1) == EMBEDDING_NAME_1
download_manager.download.assert_called_with(
EXTERNAL_MDB_GZ_URL_1,
local_file=str(
embedding_manager.get_embedding_lmdb_cache_data_path(
EMBEDDING_NAME_1)))
def test_should_unzip_mdb_embedding(
self, download_manager: MagicMock,
embedding_manager: EmbeddingManager):
embedding_manager.download_and_install_embedding(
EXTERNAL_MDB_GZ_URL_1)
download_manager.download.assert_called_with(
EXTERNAL_MDB_GZ_URL_1,
local_file=str(
embedding_manager.get_embedding_lmdb_cache_data_path(
EMBEDDING_NAME_1)))
embedding_config = embedding_manager.get_embedding_config(
EMBEDDING_NAME_1)
assert embedding_config
assert embedding_config['name'] == EMBEDDING_NAME_1