1 files changed, 22 insertions, 0 deletions
diff --git a/tests/test_helpers.py b/tests/test_helpers.py
new file mode 100755
index 0000000..b3979f3
--- /dev/null
+++ b/tests/test_helpers.py
@@ -0,0 +1,22 @@
+import unittest
+
+from AptUrl.Helpers import parse_pkg
+
+
+class Candidate:
+    def __init__(self, descr):
+        self.description = descr
+        self.homepage = None
+
+
+class MockPkg:
+    def __init__(self, descr):
+        self.candidate = Candidate(descr)
+
+
+class HelpersTest(unittest.TestCase):
+    def test_parse_pkg(self):
+        pkgobj = MockPkg("summary\ndescr\n")
+        self.assertEqual(parse_pkg(pkgobj), ('summary','descr\n',None))
+        pkgobj = MockPkg("summary only")
+        self.assertEqual(parse_pkg(pkgobj), ('summary only','',None))