SNOW-1011766: Updating integration tests for repository url to cover …

…full flow of create repo, show repos, get repo url

src/snowflake/cli/api/project/util.py

@@ -143,3 +143,8 @@ def validate_version(version: str):
         raise ValueError(
             f"Project definition version {version} is not supported by this version of Snowflake CLI. Supported versions: {SUPPORTED_VERSIONS}"
         )
+
+
+def escape_like_pattern(pattern: str) -> str:
+    pattern = pattern.replace("%", r"\\}%").replace("_", r"\\_")
+    return pattern

src/snowflake/cli/plugins/spcs/image_repository/manager.py

@@ -1,6 +1,7 @@
 from urllib.parse import urlparse
 
 from click import ClickException
+from snowflake.cli.api.project.util import escape_like_pattern
 from snowflake.cli.api.sql_execution import SqlExecutionMixin
 from snowflake.connector.cursor import SnowflakeCursor
 
@@ -16,18 +17,10 @@ def get_role(self):
         return self._conn.role
 
     def get_repository_url_list(self, repo_name: str) -> SnowflakeCursor:
-        role = self.get_role()
-        database = self.get_database()
-        schema = self.get_schema()
-
-        registry_query = f"""
-            use role {role};
-            use database {database};
-            use schema {schema};
-            show image repositories like '{repo_name}';
-            """
-
-        return self._execute_query(registry_query)
+        repository_query = (
+            f"show image repositories like '{escape_like_pattern(repo_name)}'"
+        )
+        return self._execute_query(repository_query)
 
     def get_repository_url(self, repo_name):
         database = self.get_database()

tests_integration/spcs/test_image_repository.py

@@ -1,6 +1,8 @@
 import pytest
+from snowflake.cli.api.project.util import escape_like_pattern
 
 from tests_integration.test_utils import contains_row_with, row_from_snowflake_session
+from tests_integration.testing_utils.naming_utils import ObjectNameProvider
 
 INTEGRATION_DATABASE = "SNOWCLI_DB"
 INTEGRATION_SCHEMA = "PUBLIC"
@@ -61,15 +63,20 @@ def _list_tags(runner):
 
 
 @pytest.mark.integration
-def test_get_repo_url(runner, snowflake_session):
-    # requires at least one repository to exist in the schema of snowflake_session
-    expect = snowflake_session.execute_string(f"show image repositories")
-    expect_row = row_from_snowflake_session(expect)[0]
-    expect_name = expect_row["name"]
-    expect_url = expect_row["repository_url"]
+def test_get_repo_url(runner, snowflake_session, test_database):
+    repo_name = ObjectNameProvider("Test_Repo").create_and_get_next_object_name()
+    snowflake_session.execute_string(f"create image repository {repo_name}")
 
+    created_repo = snowflake_session.execute_string(
+        f"show image repositories like '{escape_like_pattern(repo_name)}'"
+    )
+    created_row = row_from_snowflake_session(created_repo)[0]
+    created_name = created_row["name"]
+    assert created_name.lower() == repo_name.lower()
+
+    expect_url = created_row["repository_url"]
     result = runner.invoke_with_connection(
-        ["spcs", "image-repository", "url", expect_name]
+        ["spcs", "image-repository", "url", created_name]
     )
     assert isinstance(result.output, str), result.output
     assert result.output.strip() == expect_url