From 04bb8c8ccfe1fb789ead340e88b18d6ac9b9e908 Mon Sep 17 00:00:00 2001
From: jvoisin
Date: Sun, 28 Oct 2018 07:41:04 -0700
Subject: Add mp4 support

---
 tests/test_libmat2.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'tests/test_libmat2.py')

diff --git a/tests/test_libmat2.py b/tests/test_libmat2.py
index 1602480..e3072a8 100644
--- a/tests/test_libmat2.py
+++ b/tests/test_libmat2.py
@@ -521,3 +521,26 @@ class TestCleaning(unittest.TestCase):
         os.remove('./tests/data/dirty.cleaned.zip')
         os.remove('./tests/data/dirty.cleaned.cleaned.zip')
 
+
+    def test_mp4(self):
+        try:
+            video._get_ffmpeg_path()
+        except RuntimeError:
+            raise unittest.SkipTest
+
+        shutil.copy('./tests/data/dirty.mp4', './tests/data/clean.mp4')
+        p = video.MP4Parser('./tests/data/clean.mp4')
+
+        meta = p.get_meta()
+        self.assertEqual(meta['Encoder'], 'HandBrake 0.9.4 2009112300')
+
+        ret = p.remove_all()
+        self.assertTrue(ret)
+
+        p = video.MP4Parser('./tests/data/clean.cleaned.mp4')
+        self.assertNotIn('Encoder', p.get_meta())
+        self.assertTrue(p.remove_all())
+
+        os.remove('./tests/data/clean.mp4')
+        os.remove('./tests/data/clean.cleaned.mp4')
+        os.remove('./tests/data/clean.cleaned.cleaned.mp4')
-- 
cgit v1.3