Improve the testsuite

2 files changed, 15 insertions, 6 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/__init__.py
diff --git a/tests/test_libmat2.py b/tests/test_libmat2.py
index 56b960e..4751aa4 100644
--- a/tests/test_libmat2.py
+++ b/tests/test_libmat2.py
@@ -10,18 +10,27 @@ from src.parsers import pdf
 class TestGetMeta(unittest.TestCase):
     def test_pdf(self):
         p = pdf.PDFParser('./tests/data/dirty.pdf')
-        meta = p.get_meta().items()
-
+        meta = p.get_meta()
+        self.assertEqual(meta['producer'], 'pdfTeX-1.40.14')
+        self.assertEqual(meta['creator'], "'Certified by IEEE PDFeXpress at 03/19/2016 2:56:07 AM'")
 
 class TestCleaning(unittest.TestCase):
     def setUp(self):
         shutil.copy('./tests/data/dirty.pdf', './tests/data/clean.pdf')
 
     def tearDown(self):
-        #os.remove('./tests/data/clean.pdf')
-        pass
+        os.remove('./tests/data/clean.pdf')
 
     def test_pdf(self):
         p = pdf.PDFParser('./tests/data/clean.pdf')
-        p.remove_all()
-        #self.assertEqual(p.get_meta(), {})
+
+        meta = p.get_meta()
+        self.assertEqual(meta['producer'], 'pdfTeX-1.40.14')
+
+        ret = p.remove_all()
+        self.assertTrue(ret)
+
+        p = pdf.PDFParser('./tests/data/clean.pdf.cleaned')
+        remaining_meta = {'creator': 'cairo 1.14.10 (http://cairographics.org)',
+                'producer': 'cairo 1.14.10 (http://cairographics.org)'}
+        self.assertEqual(p.get_meta(), remaining_meta)