improve testing

Author: jvllmr

.gitignore

@@ -1,5 +1,5 @@
 __pycache__
 dist
 .venv
-tests/poetry.lock
-tests/*.toml
+tests/*.lock
+tests/pyproject.toml

poetry.lock

@@ -24,6 +24,11 @@ wheel = "0.37.1" # I still don't get why wheel is not shipped with pip or standa
 tox = "^3.25.0"
 tox-poetry = "^0.4.1"
 black = "^22.6.0"
+pip = "^22.2.2"
+
+
+
+
 
 
 

pyproject.toml

@@ -1,5 +1,6 @@
 import os
 import shutil
+import subprocess
 
 from pytest import fixture
 from tomlkit import TOMLDocument
@@ -14,12 +15,16 @@ def prepare_tests():
 @fixture(autouse=True)
 def yield_example():
     shutil.copy2("pyproject.toml.example", "pyproject.toml")
-    yield
-    os.remove("pyproject.toml")
     try:
         os.remove("poetry.lock")
     except FileNotFoundError:
         pass
+    subprocess.run(["poetry", "install"])
+    yield
+    subprocess.run(
+        ["python", "-m", "poetry", "remove", "--group", "types", "types-requests"]
+    )
+    # os.remove("pyproject.toml")
 
 
 class CustomTOMLFile(TOMLFile):

tests/conftest.py

@@ -0,0 +1,3 @@
+[virtualenvs]
+create = true
+in-project = true

tests/poetry.toml

@@ -3,11 +3,26 @@
 import subprocess
 
 import pytest
+
 from conftest import CustomTOMLFile
 
 
 @pytest.mark.parametrize("command", [["types", "update"]])
 def test_update(command: list[str], toml_file: CustomTOMLFile):
+    assert (
+        subprocess.run(
+            ["python", "-m", "poetry", "run", "pip", "show", "types-requests"]
+        ).returncode
+        != 0
+    )
+
+    assert (
+        subprocess.run(
+            ["python", "-m", "poetry", "run", "pip", "show", "types-colorama"]
+        ).returncode
+        == 0
+    )
+
     content = toml_file.poetry
     content["dependencies"].add("requests", "^2.27.1")
     del content["dependencies"]["colorama"]
@@ -16,14 +31,53 @@ def test_update(command: list[str], toml_file: CustomTOMLFile):
     assert "types-colorama" not in toml_file.poetry["group"]["types"]["dependencies"]
     assert "types-requests" in toml_file.poetry["group"]["types"]["dependencies"]
 
+    assert (
+        subprocess.run(
+            ["python", "-m", "poetry", "run", "pip", "show", "types-requests"]
+        ).returncode
+        == 0
+    )
+    assert (
+        subprocess.run(
+            ["python", "-m", "poetry", "run", "pip", "show", "types-colorama"]
+        ).returncode
+        != 0
+    )
+
 
 @pytest.mark.parametrize("command", [["types", "add", "requests"]])
 def test_add(command: list[str], toml_file: CustomTOMLFile):
+    assert (
+        subprocess.run(
+            ["python", "-m", "poetry", "run", "pip", "show", "types-requests"]
+        ).returncode
+        != 0
+    )
+
     subprocess.run(["python", "-m", "poetry", "--verbose", *command])
     assert "types-requests" in toml_file.poetry["group"]["types"]["dependencies"]
+    assert (
+        subprocess.run(
+            ["python", "-m", "poetry", "run", "pip", "show", "types-requests"]
+        ).returncode
+        == 0
+    )
 
 
 @pytest.mark.parametrize("command", [["types", "remove", "colorama"]])
 def test_remove(command: list[str], toml_file: CustomTOMLFile):
+    assert (
+        subprocess.run(
+            ["python", "-m", "poetry", "run", "pip", "show", "types-colorama"]
+        ).returncode
+        == 0
+    )
+
     subprocess.run(["python", "-m", "poetry", "--verbose", *command])
     assert "types-colorama" not in toml_file.poetry["group"]["types"]["dependencies"]
+    assert (
+        subprocess.run(
+            ["python", "-m", "poetry", "run", "pip", "show", "types-colorama"]
+        ).returncode
+        != 0
+    )