Add tests for parsing requirements.txt files

setup.py

@@ -125,6 +125,7 @@ setup(
                 ["python", "setup.py", "format_docstrings"],
                 ["python", "setup.py", "sort_imports"],
                 ["python", "setup.py", "lint_fix"],
+                ["python", "setup.py", "lint"],
             ],
         ),
         "verify": create_command(

tests/fixtures/requirements.txt

@@ -0,0 +1,4 @@
+django==1.0
+cookie>=1.2
+-r ../test.txt
+-e ..

tests/test_parse_requirements_txt.py

@@ -0,0 +1,26 @@
+import os
+
+from pippackagelist.requirements import (
+    RequirementsEditableEntry,
+    RequirementsPackageEntry,
+    RequirementsRecursiveEntry,
+)
+from pippackagelist.requirements_txt_parser import parse_requirements_txt
+
+requirements_txt_path = os.path.join(
+    os.path.dirname(__file__), "./fixtures/requirements.txt"
+)
+
+
+def test_parse_requirements_txt():
+    requirements = list(parse_requirements_txt(requirements_txt_path))
+    assert len(requirements) == 4
+
+    assert isinstance(requirements[0], RequirementsPackageEntry)
+    assert isinstance(requirements[1], RequirementsPackageEntry)
+    assert isinstance(requirements[2], RequirementsRecursiveEntry)
+    assert isinstance(requirements[3], RequirementsEditableEntry)
+
+    for index, requirement in enumerate(requirements):
+        assert requirement.source.path == requirements_txt_path
+        assert requirement.source.line_number == index + 1

tests/test_parse_setup_py.py

@@ -2,18 +2,14 @@ import os
 
 from pippackagelist.requirements import (
     RequirementsEditableEntry,
-    RequirementsEntrySource,
     RequirementsPackageEntry,
     RequirementsRecursiveEntry,
-    RequirementsVCSPackageEntry,
 )
 from pippackagelist.setup_py_parser import parse_setup_py
 
-setup_py_path = os.path.join(
-    os.path.dirname(__file__), "./fixtures/setup_py.py"
-)
+setup_py_path = os.path.join(os.path.dirname(__file__), "./fixtures/setup.py")
 setup_py_with_extras_path = os.path.join(
-    os.path.dirname(__file__), "./fixtures/setup_py_with_extras.py"
+    os.path.dirname(__file__), "./fixtures/setup_with_extras.py"
 )