tests: add rmmod test

2 files changed, 58 insertions, 1 deletions

diff --git a/tests/edid.py b/tests/edid.py
index f41ac82..4a9e725 100644
--- a/tests/edid.py
+++ b/tests/edid.py
@@ -15,7 +15,7 @@ from drminfo import TestDRM
 
 class EDID(TestDRM):
     """
-    basic qemu display device tests
+    edid test
 
     :avocado: tags=x86_64
     """
diff --git a/tests/unload.py b/tests/unload.py
new file mode 100644
index 0000000..d31837e
--- /dev/null
+++ b/tests/unload.py
@@ -0,0 +1,57 @@
+#
+# edid tests
+#
+
+# stdlib
+import os
+import time
+from shutil import copyfile
+
+# avocado
+import avocado
+
+# my bits
+from drminfo import TestDRM
+
+class unload(TestDRM):
+    """
+    drm module unload test
+
+    :avocado: tags=x86_64
+    """
+
+    def run_unload_test(self, vga, module):
+
+        self.boot_gfx_vm(vga);
+        self.console_prepare();
+
+        self.console_run('for vt in /sys/class/vtconsole/vtcon*; do echo 0 > $vt/bind; done')
+        self.console_wait('---root---')
+        self.console_run('rmmod %s' % module)
+        self.console_wait('---root---', 'ERROR', 'rmmod')
+
+    @avocado.skipUnless(os.path.exists('/usr/bin/dracut'), "no dracut")
+    def setUp(self):
+        TestDRM.setUp(self);
+        if not os.path.isfile(self.initrd):
+            self.prepare_kernel_initrd()
+
+    def test_stdvga(self):
+        vga = "VGA"
+        mod = "bochs-drm"
+        self.run_unload_test(vga, mod)
+
+    def test_cirrus(self):
+        vga = "cirrus-vga"
+        mod = "cirrus"
+        self.run_unload_test(vga, mod)
+
+    def test_qxl_vga(self):
+        vga = "qxl-vga"
+        mod = "qxl"
+        self.run_unload_test(vga, mod)
+
+    def test_virtio_vga(self):
+        vga = "virtio-vga"
+        mod = "virtio-gpu"
+        self.run_unload_test(vga, mod)