def test_TupleFilesystemStoreBackend(tmp_path_factory):
path = "dummy_str"
project_path = str(tmp_path_factory.mktemp("test_TupleFilesystemStoreBackend__dir"))
my_store = TupleFilesystemStoreBackend(
root_directory=os.path.abspath(path),
base_directory=project_path,
filepath_template="my_file_{0}",
)
with pytest.raises(InvalidKeyError):
my_store.get(("AAA",))
my_store.set(("AAA",), "aaa")
assert my_store.get(("AAA",)) == "aaa"
my_store.set(("BBB",), "bbb")
assert my_store.get(("BBB",)) == "bbb"
assert set(my_store.list_keys()) == {("AAA",), ("BBB",)}
assert (
gen_directory_tree_str(project_path)
== """\
test_TupleFilesystemStoreBackend__dir0/
my_file_AAA
my_file_BBB
"""
)
my_store.remove_key(("BBB",))
with pytest.raises(InvalidKeyError):
assert my_store.get(("BBB",)) == ""
def test_TupleFilesystemStoreBackend(tmp_path_factory):
path = "dummy_str"
project_path = str(
tmp_path_factory.mktemp("test_TupleFilesystemStoreBackend__dir"))
my_store = TupleFilesystemStoreBackend(
root_directory=os.path.abspath(path),
base_directory=project_path,
filepath_template="my_file_{0}",
)
# OPPORTUNITY: potentially standardize error instead of allowing each StoreBackend to raise its own error types
with pytest.raises(FileNotFoundError):
my_store.get(("AAA", ))
my_store.set(("AAA", ), "aaa")
assert my_store.get(("AAA", )) == "aaa"
my_store.set(("BBB", ), "bbb")
assert my_store.get(("BBB", )) == "bbb"
assert set(my_store.list_keys()) == {("AAA", ), ("BBB", )}
assert (gen_directory_tree_str(project_path) == """\
test_TupleFilesystemStoreBackend__dir0/
my_file_AAA
my_file_BBB
""")
my_store.remove_key(("BBB", ))
with pytest.raises(FileNotFoundError):
assert my_store.get(("BBB", )) == ""
def test_TupleFilesystemStoreBackend_ignores_jupyter_notebook_checkpoints(
tmp_path_factory, ):
project_path = str(tmp_path_factory.mktemp("things"))
checkpoint_dir = os.path.join(project_path, ".ipynb_checkpoints")
os.mkdir(checkpoint_dir)
assert os.path.isdir(checkpoint_dir)
nb_file = os.path.join(checkpoint_dir, "foo.json")
with open(nb_file, "w") as f:
f.write("")
assert os.path.isfile(nb_file)
my_store = TupleFilesystemStoreBackend(
root_directory=os.path.abspath("dummy_str"),
base_directory=project_path,
)
my_store.set(("AAA", ), "aaa")
assert my_store.get(("AAA", )) == "aaa"
assert (gen_directory_tree_str(project_path) == """\
things0/
.ge_store_backend_id
AAA
.ipynb_checkpoints/
foo.json
""")
assert set(my_store.list_keys()) == {(".ge_store_backend_id", ), ("AAA", )}
def test_TupleFilesystemStoreBackend(tmp_path_factory):
path = "dummy_str"
project_path = str(tmp_path_factory.mktemp("test_TupleFilesystemStoreBackend__dir"))
base_public_path = "http://www.test.com/"
my_store = TupleFilesystemStoreBackend(
root_directory=project_path,
base_directory=os.path.join(project_path, path),
filepath_template="my_file_{0}",
)
with pytest.raises(InvalidKeyError):
my_store.get(("AAA",))
my_store.set(("AAA",), "aaa")
assert my_store.get(("AAA",)) == "aaa"
my_store.set(("BBB",), "bbb")
assert my_store.get(("BBB",)) == "bbb"
assert set(my_store.list_keys()) == {(".ge_store_backend_id",), ("AAA",), ("BBB",)}
assert (
gen_directory_tree_str(project_path)
== """\
test_TupleFilesystemStoreBackend__dir0/
dummy_str/
.ge_store_backend_id
my_file_AAA
my_file_BBB
"""
)
my_store.remove_key(("BBB",))
with pytest.raises(InvalidKeyError):
assert my_store.get(("BBB",)) == ""
my_store_with_base_public_path = TupleFilesystemStoreBackend(
root_directory=project_path,
base_directory=os.path.join(project_path, path),
filepath_template="my_file_{0}",
base_public_path=base_public_path,
)
my_store_with_base_public_path.set(("CCC",), "ccc")
url = my_store_with_base_public_path.get_public_url_for_key(("CCC",))
assert url == "http://www.test.com/my_file_CCC"