From 37cce37e05defa839a326adcbb2ac1bc1590f4f0 Mon Sep 17 00:00:00 2001
From: Ethan Paul <24588726+enpaul@users.noreply.github.com>
Date: Fri, 16 Apr 2021 20:53:35 -0400
Subject: [PATCH] Add tests for the transient dependency identification
 function

---
 tests/test_transients.py | 72 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 tests/test_transients.py

diff --git a/tests/test_transients.py b/tests/test_transients.py
new file mode 100644
index 0000000..53f7039
--- /dev/null
+++ b/tests/test_transients.py
@@ -0,0 +1,72 @@
+import poetry.factory
+import poetry.utils.env
+import pytest
+from poetry.puzzle.provider import Provider
+
+from .fixtures import mock_poetry_factory
+from .fixtures import mock_venv
+from tox_poetry_installer import constants
+from tox_poetry_installer import datatypes
+from tox_poetry_installer import exceptions
+from tox_poetry_installer import utilities
+
+
+def test_exclude_unsafe():
+    assert Provider.UNSAFE_PACKAGES == constants.UNSAFE_PACKAGES
+
+    for dep in constants.UNSAFE_PACKAGES:
+        assert utilities.identify_transients(dep, dict(), None) == []
+
+
+def test_allow_missing():
+    with pytest.raises(exceptions.LockedDepNotFoundError):
+        utilities.identify_transients("luke-skywalker", dict(), None)
+
+    assert (
+        utilities.identify_transients(
+            "darth-vader", dict(), None, allow_missing=["darth-vader"]
+        )
+        == []
+    )
+
+
+def test_exclude_pep508():
+    for version in [
+        "foo==1.0",
+        "foo==1",
+        "foo>2.0.0",
+        "foo<=9.3.4.7.8",
+        "foo>999,<=4.6",
+        "foo>1234 || foo<2021.01.01",
+        "foo!=7",
+        "foo~=0.8",
+        "foo!=9,==7",
+        "=>foo",
+    ]:
+        with pytest.raises(exceptions.LockedDepVersionConflictError):
+            utilities.identify_transients(version, dict(), None)
+
+
+def test_functional(mock_poetry_factory, mock_venv):
+    pypoetry = poetry.factory.Factory().create_poetry(None)
+    packages: datatypes.PackageMap = {
+        item.name: item for item in pypoetry.locker.locked_repository(False).packages
+    }
+    venv = poetry.utils.env.VirtualEnv()
+
+    requests_requires = [
+        packages["certifi"],
+        packages["chardet"],
+        packages["idna"],
+        packages["urllib3"],
+        packages["requests"],
+    ]
+
+    transients = utilities.identify_transients("requests", packages, venv)
+
+    assert all((item in requests_requires) for item in transients)
+    assert all((item in transients) for item in requests_requires)
+
+    for package in [packages["requests"], packages["tox"], packages["flask"]]:
+        transients = utilities.identify_transients(package, packages, venv)
+        assert transients[-1] == package