class GCSContentsManagerTestCase_prefix(TestContentsManager):
def setUp(self):
"""
This setup is a hardcoded to run on my laptop and GCP account :)
"""
self.contents_manager = GCSContentsManager(
project="continuum-compute",
token="~/.config/gcloud/application_default_credentials.json",
bucket="gcsfs-test",
prefix="this/is/the/prefix")
self.tearDown()
def tearDown(self):
for item in self.contents_manager.fs.ls(""):
self.contents_manager.fs.rm(item)
self.contents_manager.fs.init()
# Overwrites from TestContentsManager
def make_dir(self, api_path):
self.contents_manager.new(
model={"type": "directory"},
path=api_path,
)
class GcsContentsManagerLargeFileTestCase(TestLargeFileManager):
def setUp(self):
"""
This setup is a hardcoded to run on my laptop and GCP account :)
"""
self.contents_manager = GCSContentsManager(
project="continuum-compute",
token="~/.config/gcloud/application_default_credentials.json",
bucket="gcsfs-test",
)
self.tearDown()
def tearDown(self):
for item in self.contents_manager.fs.ls(""):
self.contents_manager.fs.rm(item)
self.contents_manager.fs.init()
# Overwrites from TestContentsManager
def make_dir(self, api_path):
self.contents_manager.new(
model={"type": "directory"}, path=api_path,
)
def test_save(self):
current_value = content_chunks.get()
current_value["stale_file.txt"] = {
"started_at": time.time() - 4000,
"chunks": [],
}
super().test_save()
self.assertNotIn("stale_file.txt", current_value)
def setUp(self):
"""
This setup is a hardcoded to run on my laptop and GCP account :)
"""
self.contents_manager = GCSContentsManager(
project="continuum-compute",
token="~/.config/gcloud/application_default_credentials.json",
bucket="gcsfs-test")
self.tearDown()