tests: Add buffer map/unmap tests

1 files changed, 19 insertions, 0 deletions

diff --git a/testsuite/test_gst.py b/testsuite/test_gst.py
index d63f4e2..e788489 100644
--- a/testsuite/test_gst.py
+++ b/testsuite/test_gst.py
@@ -93,5 +93,24 @@ class TestBin(TestCase):
         Gst.init(None)
         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__":
     unittest.main()