diff --git a/setup.py b/setup.py index f270e6d..f7d38ce 100644 --- a/setup.py +++ b/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( diff --git a/tests/fixtures/requirements.txt b/tests/fixtures/requirements.txt new file mode 100644 index 0000000..bfed5c6 --- /dev/null +++ b/tests/fixtures/requirements.txt @@ -0,0 +1,4 @@ +django==1.0 +cookie>=1.2 +-r ../test.txt +-e .. diff --git a/tests/fixtures/setup_py.py b/tests/fixtures/setup.py similarity index 100% rename from tests/fixtures/setup_py.py rename to tests/fixtures/setup.py diff --git a/tests/fixtures/setup_py_with_extras.py b/tests/fixtures/setup_with_extras.py similarity index 100% rename from tests/fixtures/setup_py_with_extras.py rename to tests/fixtures/setup_with_extras.py diff --git a/tests/test_parse_requirements_txt.py b/tests/test_parse_requirements_txt.py new file mode 100644 index 0000000..128c5a0 --- /dev/null +++ b/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 diff --git a/tests/test_parse_setup_py.py b/tests/test_parse_setup_py.py index 5371596..c7254a4 100644 --- a/tests/test_parse_setup_py.py +++ b/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" )