From 4007517e45a589e10a6111d5066f0a96694e2d6d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Thayn=C3=A3=20Moretti?= <moretty@gmail.com>
Date: Tue, 12 Oct 2021 11:24:43 -0300
Subject: [PATCH] chore: create libvirtvms tests

---
 tests/modules/contrib/test_libvirtvms.py | 43 +++++++++++++++++++++++-
 1 file changed, 42 insertions(+), 1 deletion(-)

diff --git a/tests/modules/contrib/test_libvirtvms.py b/tests/modules/contrib/test_libvirtvms.py
index efa5880..48ba72c 100644
--- a/tests/modules/contrib/test_libvirtvms.py
+++ b/tests/modules/contrib/test_libvirtvms.py
@@ -1,7 +1,48 @@
+import sys
 import pytest
+from unittest.mock import Mock
 
-pytest.importorskip("libvirt")
+import core.config
+
+sys.modules['libvirt'] = Mock()
+
+import modules.contrib.libvirtvms
+
+def build_module():
+    return modules.contrib.libvirtvms.Module(
+        config=core.config.Config([]),
+        theme=None
+    )
 
 def test_load_module():
     __import__("modules.contrib.libvirtvms")
 
+def test_input_registration(mocker):
+   input_register = mocker.patch('core.input.register')
+
+   module = build_module()
+
+   input_register.assert_called_with(
+       module,
+       button=core.input.LEFT_MOUSE,
+       cmd="virt-manager"
+   )
+
+def test_status_failed(mocker):
+    mocker.patch('libvirt.openReadOnly', return_value=None)
+
+    module = build_module()
+    status = module.status(None)
+
+    assert status == "Failed to open connection to the hypervisor"
+
+def test_status(mocker):
+    virtMock = mocker.Mock()
+    virtMock.numOfDomains = mocker.Mock(return_value=10)
+
+    mocker.patch('libvirt.openReadOnly', return_value=virtMock)
+
+    module = build_module()
+    status = module.status(None)
+
+    assert status == "VMs 10"