fixup! Add image repositories to cleanup task

snowflakedb · Nov 5, 2024 · fd9052b · fd9052b
1 parent f4b7c41
commit fd9052b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/scripts/cleanup.py b/scripts/cleanup.py
@@ -28,13 +28,12 @@ def is_stale(item):
 
 
 def remove_resource(resource_type: str, item, role: str):
-    msg = f"Deleting {resource_type} {item.name}"
+    fqn = f"{item.database_name}.{item.schema_name}.{item.name}"
+    msg = f"Deleting {resource_type} {fqn}"
     try:
         if is_stale(item):
-            session.sql(
-                f"GRANT OWNERSHIP ON {resource_type} {item.name} TO ROLE {role}"
-            )
-            session.sql(f'drop {resource_type} "{item.name}"').collect()
+            session.sql(f"GRANT OWNERSHIP ON {resource_type} {fqn} TO ROLE {role}")
+            session.sql(f'drop {resource_type} "{fqn}"').collect()
             print("SUCCESS", msg, f"created at {item.created_on}")
     except Exception as err:
         print("ERROR", msg)

diff --git a/tests_integration/spcs/test_image_repository.py b/tests_integration/spcs/test_image_repository.py
@@ -113,3 +113,4 @@ def test_create_image_repository(runner, test_database):
     assert result.json == {
         "status": f"Image Repository {repo_name.upper()} successfully created."
     }
+    runner.invoke_with_connection_json(["spcs", "image-repository", "drop", repo_name])