def test_resource_caching_if_already_restored_with_different_state(
tmp_path_factory: TempPathFactory, ):
initial_storage_dir = tmp_path_factory.mktemp("initial_model_storage")
model_storage = LocalModelStorage(initial_storage_dir)
resource = Resource("my resource")
# Fill model storage
test_filename = "file.txt"
test_content = "test_resource_caching"
with model_storage.write_to(resource) as temporary_directory:
file = temporary_directory / test_filename
file.write_text(test_content)
cache_dir = tmp_path_factory.mktemp("cache_dir")
# Cache resource
resource.to_cache(cache_dir, model_storage)
# Pretend there is an additional file which is not in the restored storage.
# This makes the directories and causes the `from_cache` part to fail
# different
(temporary_directory / "another_file").touch()
new_storage_dir = tmp_path_factory.mktemp("new dir")
rasa.utils.common.copy_directory(initial_storage_dir, new_storage_dir)
with pytest.raises(ValueError):
Resource.from_cache(
resource.name,
cache_dir,
LocalModelStorage(new_storage_dir),
resource.output_fingerprint,
)
def test_resource_caching_if_already_restored(
tmp_path_factory: TempPathFactory):
initial_storage_dir = tmp_path_factory.mktemp("initial_model_storage")
model_storage = LocalModelStorage(initial_storage_dir)
resource = Resource("my resource")
# Fill model storage
test_filename = "file.txt"
test_content = "test_resource_caching"
with model_storage.write_to(resource) as temporary_directory:
file = temporary_directory / test_filename
file.write_text(test_content)
cache_dir = tmp_path_factory.mktemp("cache_dir")
# Cache resource
resource.to_cache(cache_dir, model_storage)
new_storage_dir = tmp_path_factory.mktemp("new dir")
rasa.utils.common.copy_directory(initial_storage_dir, new_storage_dir)
reinstantiated_resource = Resource.from_cache(
resource.name,
cache_dir,
LocalModelStorage(new_storage_dir),
resource.output_fingerprint,
)
assert reinstantiated_resource == resource
def test_resource_caching(tmp_path_factory: TempPathFactory):
model_storage = LocalModelStorage(
tmp_path_factory.mktemp("initial_model_storage"))
resource = Resource("my resource")
# Fill model storage
test_filename = "file.txt"
test_content = "test_resource_caching"
with model_storage.write_to(resource) as temporary_directory:
file = temporary_directory / test_filename
file.write_text(test_content)
cache_dir = tmp_path_factory.mktemp("cache_dir")
# Cache resource
resource.to_cache(cache_dir, model_storage)
# Reload resource from cache and inspect
new_model_storage = LocalModelStorage(
tmp_path_factory.mktemp("new_model_storage"))
reinstantiated_resource = Resource.from_cache(resource.name, cache_dir,
new_model_storage,
resource.output_fingerprint)
assert reinstantiated_resource == resource
assert reinstantiated_resource.fingerprint() == resource.fingerprint()
# Read written resource data from model storage to see whether all expected
# contents are there
with new_model_storage.read_from(resource) as temporary_directory:
assert (temporary_directory /
test_filename).read_text() == test_content
def test_caching_empty_resource(
default_model_storage: ModelStorage,
tmp_path: Path,
tmp_path_factory: TempPathFactory,
):
resource_name = "my resource"
resource = Resource(resource_name)
# does not raise
resource.to_cache(tmp_path, default_model_storage)
with pytest.raises(ValueError):
with default_model_storage.read_from(resource) as _:
pass
cache_dir = tmp_path_factory.mktemp("cache_dir")
# this doesn't create an empty directory in `default_model_storage`
Resource.from_cache(resource_name, cache_dir, default_model_storage,
resource.output_fingerprint)
with pytest.raises(ValueError):
with default_model_storage.read_from(resource) as _:
pass
