Avoid capitalization issues in SS hosts

scope/helpers.py

@@ -24,9 +24,8 @@ def ss_hosts_parser(hosts):
         # Raise if credentials are not present
         if not parts.username or not parts.password:
             raise ValueError("Missing credentials for SS host: %s" % host)
-        url = "%s://%s" % (parts.scheme, parts.hostname)
-        if parts.port:
-            url = "%s:%s" % (url, parts.port)
+        # Use netloc instead of hostname and port to maintain capitalization
+        url = "%s://%s" % (parts.scheme, parts.netloc.split("@")[1])
         parsed_hosts[url] = {"user": parts.username, "secret": parts.password}
     return parsed_hosts
 

scope/tests/test_helpers.py

@@ -29,11 +29,13 @@ def test_ss_hosts_parser_success(self):
         hosts = [
             "http://user:secret@localhost:62081",
             "https://user:[email protected]",
+            "https://user:[email protected]:8000",
         ]
         parsed_hosts = helpers.ss_hosts_parser(hosts)
         expected_hosts = {
             "http://localhost:62081": {"user": "user", "secret": "secret"},
             "https://192.168.1.128": {"user": "user", "secret": "secret"},
+            "https://HOST.com:8000": {"user": "user", "secret": "secret"},
         }
         self.assertEqual(parsed_hosts, expected_hosts)