Fix tests

Author: sage-maker

src/sagemaker/git_utils.py

@@ -52,6 +52,13 @@ def _sanitize_git_url(repo_url):
         if at_count > 1:
             raise ValueError("Invalid HTTPS URL format: multiple @ symbols detected")
 
+        # Check for invalid characters in the URL before parsing
+        # These characters should not appear in legitimate URLs
+        invalid_chars = ["<", ">", "[", "]", "{", "}", "\\", "^", "`", "|"]
+        for char in invalid_chars:
+            if char in repo_url:
+                raise ValueError("Invalid characters in hostname")
+
         try:
             parsed = urlparse(repo_url)
 

tests/unit/test_estimator.py

@@ -2794,7 +2794,7 @@ def test_git_support_bad_repo_url_format(sagemaker_session):
     )
     with pytest.raises(ValueError) as error:
         fw.fit()
-    assert "Invalid Git url provided." in str(error)
+    assert "Unsupported URL scheme" in str(error)
 
 
 @patch(