diff --git a/test/unit/video_test.py b/test/unit/video_test.py index ec65b8d..58c211b 100644 --- a/test/unit/video_test.py +++ b/test/unit/video_test.py @@ -94,15 +94,39 @@ class TestVideo(unittest.TestCase): self.assertEqual(avi.tags['container'], 'ogg') def test_capture(self): - """test capture with some small movie""" + """test capture with some small movie and play a little with tags""" avi = Video("mocks/m.avi") filename = avi.capture() self.assertTrue(filename != None) self.assertTrue(os.path.exists(filename)) file_size = os.stat(filename)[6] - self.assertEqual(file_size, 9077) + self.assertEqual(file_size, 9067) os.unlink(filename) + for length in (480, 380, 4): + avi.tags['length'] = length + filename = avi.capture() + self.assertTrue(filename is not None) + os.unlink(filename) + + avi.tags['length'] = 3 + self.assertTrue(avi.capture() is None) + + avi.tags['length'] = 4 + + avi.tags['width'] = 0 + self.assertTrue(avi.capture() is None) + + avi.tags['width'] = 1025 + filename = avi.capture() + self.assertTrue(filename is not None) + os.unlink(filename) + + del(avi.tags['length']) + self.assertTrue(avi.capture() is None) + + self.assertTrue(len(str(avi)) > 0) + if __name__ == "__main__": os.chdir(os.path.join(os.path.abspath(os.path.dirname(__file__)), "../"))