From df3c27d79dec231809deb4e617070a16858c306d Mon Sep 17 00:00:00 2001
From: jvoisin
Date: Sun, 18 Mar 2018 21:42:12 +0100
Subject: Improve the testsuite

---
 tests/__init__.py     |  0
 tests/test_libmat2.py | 21 +++++++++++++++------
 2 files changed, 15 insertions(+), 6 deletions(-)
 create mode 100644 tests/__init__.py

(limited to 'tests')

diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
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)
-- 
cgit v1.3