def delete(self) -> None:
"""Delete this workspace."""
doc = self.get_metadata()
sysdb = system_db(readonly=False)
coll = workspace_mapping_collection(readonly=False)
sysdb.delete_database(doc["internal"])
coll.delete(doc["_id"])
# Invalidate the cache for things changed by this function
workspace_mapping.cache_clear()
def save(self) -> None:
"""Save this workspace to the database."""
doc = self.get_metadata()
instance_dict = self.asdict()
doc.update(instance_dict)
coll = workspace_mapping_collection(readonly=False)
coll.update(doc)
# Invalidate the cache for things changed by this function
workspace_mapping.cache_clear()
def test_present_workspace(managed_workspace):
"""Test that workspace caching works as expected on present workspaces."""
# Assert that the cached response matches the actual response
assert workspace_mapping.__wrapped__(
managed_workspace.name) == workspace_mapping(managed_workspace.name)
workspace_mapping.cache_clear()
first_resp = workspace_mapping(managed_workspace.name)
second_resp = workspace_mapping(managed_workspace.name)
# Assert that cached response is idempotent
assert first_resp == second_resp
def rename(self, new_name: str) -> None:
"""Rename this workspace."""
if Workspace.exists(new_name):
raise AlreadyExists("Workspace", new_name)
doc = self.get_metadata()
doc["name"] = new_name
coll = workspace_mapping_collection(readonly=False)
coll.update(doc)
self.name = new_name
# Invalidate the cache for things changed by this function
workspace_mapping.cache_clear()
def test_absent_workspace():
"""Test that workspace caching works as expected on absent workspaces."""
# Test that random workspace doesn't exist
assert workspace_mapping.__wrapped__(uuid4().hex) is None
workspace_mapping.cache_clear()
workspace_name = uuid4().hex
first_resp = workspace_mapping(workspace_name)
second_resp = workspace_mapping(workspace_name)
# Test that fake workspace doesn't exist,
# and that calls are idempotent
assert first_resp is None
assert second_resp is None
def create(name: str, owner: User) -> Workspace:
"""Create a workspace, owned by `owner`."""
if Workspace.exists(name):
raise AlreadyExists("Workspace", name)
internal = util.generate_arango_workspace_name()
try:
system_db(readonly=False).create_database(internal)
except DatabaseCreateError:
# Could only happen if there's a name collision
raise InternalServerError("Error creating workspace")
workspace_dict = {
"name": name,
"internal": internal,
"permissions": WorkspacePermissions(owner=owner.sub).dict(),
}
coll = workspace_mapping_collection(readonly=False)
coll.insert(workspace_dict, sync=True)
workspace_mapping.cache_clear()
return Workspace(name)