tests: Add buffer map/unmap tests

This commit is contained in:
Philipp Zabel 2019-10-17 17:31:41 +02:00
parent fecfe451a7
commit 6d53d0ae0e

View file

@ -93,5 +93,24 @@ class TestBin(TestCase):
Gst.init(None) Gst.init(None)
self.assertEqual(Gst.ElementFactory.make("bin", None).sinkpads, []) self.assertEqual(Gst.ElementFactory.make("bin", None).sinkpads, [])
class TestBufferMap(TestCase):
def test_map_unmap_manual(self):
Gst.init(None)
buf = Gst.Buffer.new_wrapped([42])
info = buf.map(Gst.MapFlags.READ | Gst.MapFlags.WRITE)
self.assertEqual(info.data[0], 42)
buf.unmap(info)
with self.assertRaises(ValueError):
info.data[0]
def test_map_unmap_context(self):
Gst.init(None)
buf = Gst.Buffer.new_wrapped([42])
with buf.map(Gst.MapFlags.READ | Gst.MapFlags.WRITE) as info:
self.assertEqual(info.data[0], 42)
with self.assertRaises(ValueError):
info.data[0]
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()