Update tests

TracecatHQ · Jan 15, 2025 · 56a242f · 56a242f
1 parent f235636
commit 56a242f
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/unit/test_registry.py b/tests/unit/test_registry.py
@@ -154,17 +154,17 @@ async def test_registry_async_function_can_be_called(mock_package):
                 host="github.com",
                 org="org",
                 repo="repo",
-                branch="main",
+                ref=None,
             ),
         ),
-        # GitHub (with branch)
+        # GitHub (with branch/sha)
         (
-            "git+ssh://[email protected]/org/repo@branch",
+            "git+ssh://[email protected]/org/repo@branchOrSHAOrTag",
             GitUrl(
                 host="github.com",
                 org="org",
                 repo="repo",
-                branch="branch",
+                ref="branchOrSHAOrTag",
             ),
         ),
         # GitLab
@@ -174,7 +174,7 @@ async def test_registry_async_function_can_be_called(mock_package):
                 host="gitlab.com",
                 org="org",
                 repo="repo",
-                branch="main",
+                ref=None,
             ),
         ),
         # GitLab (with branch)
@@ -184,7 +184,7 @@ async def test_registry_async_function_can_be_called(mock_package):
                 host="gitlab.com",
                 org="org",
                 repo="repo",
-                branch="branch",
+                ref="branch",
             ),
         ),
     ],