Fix unit tests

ansible-collections · Jan 4, 2024 · 4c5e8f8 · 4c5e8f8
1 parent 6eec740
commit 4c5e8f8
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/tests/unit/plugins/modules/test_mysql_info.py b/tests/unit/plugins/modules/test_mysql_info.py
@@ -36,16 +36,32 @@ def __cursor_return_value(input_parameter):
 
     assert info.get_info([], [], False)['version']['suffix'] == suffix
 
+
+@pytest.mark.parametrize(
+    'server_implementation',
+    [
+        ('mysql'),
+        ('mariadb'),
+    ]
+)
 def test_is_mariadb(server_implementation):
     cursor = MagicMock()
 
     info = MySQL_Info(MagicMock(), cursor, server_implementation)
 
-    assert info.is_mariadb() == ( server_implementation == "mariadb" )
+    assert info.is_mariadb() == (server_implementation == "mariadb")
 
+
+@pytest.mark.parametrize(
+    'server_implementation',
+    [
+        ('mysql'),
+        ('mariadb'),
+    ]
+)
 def test_is_mysql(server_implementation):
     cursor = MagicMock()
 
     info = MySQL_Info(MagicMock(), cursor, server_implementation)
 
-    assert info.is_mysql() == ( server_implementation == "mysql" )
+    assert info.is_mysql() == (server_implementation == "mysql")