Add tests for the parse function #253

Signed-off-by: Li <[email protected]>
aboutcode-org · Jun 5, 2019 · 04c74ae · 04c74ae
1 parent d7ebe5a
commit 04c74ae
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tests/packagedcode/test_pypi.py b/tests/packagedcode/test_pypi.py
@@ -262,8 +262,21 @@ def test_pkginfo_parse_with_unpackaged_source(self):
         expected_loc = self.get_test_loc('pypi/unpackage_source_parser-expected.json')
         self.check_package(package, expected_loc, regen=False)
 
+    def test_pkginfo_parse_with_unpackaged_source_with_parse_function(self):
+        test_file = self.get_test_loc('pypi')
+        package = pypi.parse2(test_file)
+        expected_loc = self.get_test_loc('pypi/unpackage_source_parser-expected.json')
+        self.check_package(package, expected_loc, regen=False)
+
     def test_pkginfo_parse_with_wheelfile(self):
         test_file = self.get_test_loc('pypi/wheel/atomicwrites-1.2.1-py2.py3-none-any.whl')
         package = pypi.parse_wheel(test_file)
         expected_loc = self.get_test_loc('pypi/wheel/parse-wheel-expected.json')
         self.check_package(package, expected_loc, regen=False)
+
+
+    def test_pkginfo_parse_with_wheelfile_with_parse_function(self):
+        test_file = self.get_test_loc('pypi/wheel/atomicwrites-1.2.1-py2.py3-none-any.whl')
+        package = pypi.parse2(test_file)
+        expected_loc = self.get_test_loc('pypi/wheel/parse-wheel-expected.json')
+        self.check_package(package, expected_loc, regen=False)