diff options
Diffstat (limited to 'tests/test_libmat2.py')
| -rw-r--r-- | tests/test_libmat2.py | 24 |
1 files changed, 23 insertions, 1 deletions
diff --git a/tests/test_libmat2.py b/tests/test_libmat2.py index 4df6385..90f37a8 100644 --- a/tests/test_libmat2.py +++ b/tests/test_libmat2.py | |||
| @@ -6,7 +6,7 @@ import os | |||
| 6 | import zipfile | 6 | import zipfile |
| 7 | import tempfile | 7 | import tempfile |
| 8 | 8 | ||
| 9 | from libmat2 import pdf, images, audio, office, parser_factory, torrent | 9 | from libmat2 import pdf, images, audio, office, parser_factory, torrent, harmless |
| 10 | 10 | ||
| 11 | 11 | ||
| 12 | class TestParserFactory(unittest.TestCase): | 12 | class TestParserFactory(unittest.TestCase): |
| @@ -104,6 +104,12 @@ class TestGetMeta(unittest.TestCase): | |||
| 104 | self.assertEqual(meta['meta:creation-date'], '2011-07-26T03:27:48') | 104 | self.assertEqual(meta['meta:creation-date'], '2011-07-26T03:27:48') |
| 105 | self.assertEqual(meta['meta:generator'], 'LibreOffice/3.3$Unix LibreOffice_project/330m19$Build-202') | 105 | self.assertEqual(meta['meta:generator'], 'LibreOffice/3.3$Unix LibreOffice_project/330m19$Build-202') |
| 106 | 106 | ||
| 107 | def test_txt(self): | ||
| 108 | p, mimetype = parser_factory.get_parser('./tests/data/dirty.txt') | ||
| 109 | self.assertEqual(mimetype, 'text/plain') | ||
| 110 | meta = p.get_meta() | ||
| 111 | self.assertEqual(meta, {}) | ||
| 112 | |||
| 107 | 113 | ||
| 108 | class TestRemovingThumbnails(unittest.TestCase): | 114 | class TestRemovingThumbnails(unittest.TestCase): |
| 109 | def test_odt(self): | 115 | def test_odt(self): |
| @@ -473,3 +479,19 @@ class TestCleaning(unittest.TestCase): | |||
| 473 | 479 | ||
| 474 | os.remove('./tests/data/clean.odg') | 480 | os.remove('./tests/data/clean.odg') |
| 475 | os.remove('./tests/data/clean.cleaned.odg') | 481 | os.remove('./tests/data/clean.cleaned.odg') |
| 482 | |||
| 483 | def test_txt(self): | ||
| 484 | shutil.copy('./tests/data/dirty.txt', './tests/data/clean.txt') | ||
| 485 | p = harmless.HarmlessParser('./tests/data/clean.txt') | ||
| 486 | |||
| 487 | meta = p.get_meta() | ||
| 488 | self.assertEqual(meta, {}) | ||
| 489 | |||
| 490 | ret = p.remove_all() | ||
| 491 | self.assertTrue(ret) | ||
| 492 | |||
| 493 | p = harmless.HarmlessParser('./tests/data/clean.cleaned.txt') | ||
| 494 | self.assertEqual(p.get_meta(), {}) | ||
| 495 | |||
| 496 | os.remove('./tests/data/clean.txt') | ||
| 497 | os.remove('./tests/data/clean.cleaned.txt') | ||
