From 3b7929daa8cd6370ae64b1294d8f0dddb18e4ede Mon Sep 17 00:00:00 2001
From: Kyle Altendorf <kyle.altendorf@epcpower.com>
Date: Wed, 16 May 2018 23:07:09 -0400
Subject: [PATCH] Add a few test cases for PackageFinder.get_extras_links()

---
 tests/unit/test_patched.py | 131 +++++++++++++++++++++++++++++++++++++
 1 file changed, 131 insertions(+)
 create mode 100644 tests/unit/test_patched.py

diff --git a/tests/unit/test_patched.py b/tests/unit/test_patched.py
new file mode 100644
index 00000000..8d396539
--- /dev/null
+++ b/tests/unit/test_patched.py
@@ -0,0 +1,131 @@
+from __future__ import absolute_import
+
+import pytest
+
+from notpip.index import PackageFinder
+
+
+get_extras_links_scenarios = {
+    'windows and not windows': (
+        [
+            'chardet',
+            '[:platform_system != "windows"]',
+            'twisted',
+            '[:platform_system == "windows"]',
+            'twisted[windows_platform]',
+        ],
+        {
+            ':platform_system != "windows"': [
+                'twisted',
+            ],
+            ':platform_system == "windows"': [
+                'twisted[windows_platform]',
+            ],
+        },
+    ),
+    'requests': (
+        [
+            'chardet<3.1.0,>=3.0.2',
+            'idna<2.7,>=2.5',
+            'urllib3<1.23,>=1.21.1',
+            'certifi>=2017.4.17',
+            '[security]',
+            'pyOpenSSL>=0.14',
+            'cryptography>=1.3.4',
+            'idna>=2.0.0',
+            '[socks]',
+            'PySocks!=1.5.7,>=1.5.6',
+            (
+                '[socks:sys_platform == "win32"'
+                ' and (python_version == "2.7" or python_version == "2.6")]'
+            ),
+            'win_inet_pton',
+        ],
+        {
+            'security': [
+                'pyOpenSSL>=0.14',
+                'cryptography>=1.3.4',
+                'idna>=2.0.0',
+            ],
+            'socks': [
+                'PySocks!=1.5.7,>=1.5.6',
+            ],
+            'socks:sys_platform == "win32" '
+            'and (python_version == "2.7" or python_version == "2.6")': [
+                'win_inet_pton',
+            ],
+        }
+    ),
+    'attrs': (
+        [
+            '[dev]',
+            'coverage',
+            'hypothesis',
+            'pympler',
+            'pytest',
+            'six',
+            'zope.interface',
+            'sphinx',
+            'zope.interface',
+            '[docs]',
+            'sphinx',
+            'zope.interface',
+            '[tests]',
+            'coverage',
+            'hypothesis',
+            'pympler',
+            'pytest',
+            'six',
+            'zope.interface',
+        ],
+        {
+            'dev': [
+                'coverage',
+                'hypothesis',
+                'pympler',
+                'pytest',
+                'six',
+                'zope.interface',
+                'sphinx',
+                'zope.interface',
+            ],
+            'docs': [
+                'sphinx',
+                'zope.interface',
+            ],
+            'tests': [
+                'coverage',
+                'hypothesis',
+                'pympler',
+                'pytest',
+                'six',
+                'zope.interface',
+            ],
+        },
+    ),
+    'misc': (
+        [
+            'chardet',
+            '[:platform_system != "windows"]',
+            'attrs',
+            '[:platform_system == "windows"]',
+            'pytz',
+        ],
+        {
+            ':platform_system != "windows"': [
+                'attrs',
+            ],
+            ':platform_system == "windows"': [
+                'pytz',
+            ],
+        },
+    ),
+}
+
+@pytest.mark.parametrize(
+    'scenarios,expected',
+    list(get_extras_links_scenarios.values()),
+    ids=list(get_extras_links_scenarios.keys()),
+)
+def test_get_extras_links(scenarios, expected):
+    assert PackageFinder.get_extras_links(scenarios) == expected