Properly handle the first source having user/pass

pypa · Jul 31, 2018 · f78637f · f78637f
1 parent d4f710b
commit f78637f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/pipenv/utils.py b/pipenv/utils.py
@@ -200,7 +200,7 @@ def prepare_pip_source_args(sources, pip_args=None):
         # Trust the host if it's not verified.
         if not sources[0].get("verify_ssl", True):
             pip_args.extend(
-                ["--trusted-host", urlparse(sources[0]["url"]).netloc.split(":")[0]]
+                ["--trusted-host", urlparse(sources[0]["url"]).hostname]
             )
         # Add additional sources as extra indexes.
         if len(sources) > 1:

diff --git a/tests/unit/test_utils.py b/tests/unit/test_utils.py
@@ -371,6 +371,20 @@ def test_nix_normalize_drive(self, input_path, expected):
                     "https://user:[email protected]/simple",
                 ],
             ),
+            (
+                [
+                    {
+                        "url": "https://user:[email protected]/simple",
+                        "verify_ssl": False,
+                    },
+                ],
+                [
+                    "-i",
+                    "https://user:[email protected]/simple",
+                    "--trusted-host",
+                    "custom.example.com",
+                ],
+            ),
         ],
     )
     def test_prepare_pip_source_args(self, sources, expected_args):