1 files changed, 65 insertions, 0 deletions
diff --git a/tests/kms-test-modes.py b/tests/kms-test-modes.py
new file mode 100755
index 0000000..31128a7
--- /dev/null
+++ b/tests/kms-test-modes.py
@@ -0,0 +1,65 @@
+#!/usr/bin/python3
+
+import kmstest
+import pykms
+
+class ModesTest(kmstest.KMSTest):
+    """Test all available modes on all available connectors."""
+
+    def handle_page_flip(self, frame, time):
+        self.logger.log("Page flip complete")
+
+    def test_mode(self, connector, crtc, mode):
+        self.logger.log("Testing connector %s on CRTC %u with mode %s" % \
+              (connector.fullname, crtc.id, mode.name))
+
+        # Create a frame buffer
+        fb = pykms.DumbFramebuffer(self.card, mode.hdisplay, mode.vdisplay, "XR24")
+        pykms.draw_test_pattern(fb)
+
+        # Perform the mode set
+        ret = self.atomic_crtc_mode_set(crtc, connector, mode, fb)
+        if ret < 0:
+            raise RuntimeError("atomic mode set failed with %d" % ret)
+
+        self.logger.log("Atomic mode set complete")
+        self.run(4)
+
+        if self.flips == 0:
+            raise RuntimeError("Page flip not registered")
+
+    def main(self):
+        for connector in self.card.connectors:
+            self.start("modes on connector %s" % connector.fullname)
+
+            # Skip disconnected connectors
+            if not connector.connected():
+                self.skip("unconnected connector")
+                continue
+
+            # Find a CRTC suitable for the connector
+            crtc = connector.get_current_crtc()
+            if not crtc:
+                crtcs = connector.get_possible_crtcs()
+                if len(crtcs) == 0:
+                    pass
+
+                crtc = crtcs[0]
+
+            # Test all available modes
+            modes = connector.get_modes()
+            if len(modes) == 0:
+                self.skip("no mode available")
+                continue
+
+            for i in range(len(modes)):
+                try:
+                    self.progress(i+1, len(modes))
+                    self.test_mode(connector, crtc, modes[i])
+                except RuntimeError as e:
+                    self.fail(e.message)
+                    break
+            else:
+                self.success()
+
+ModesTest().execute()