squash: extend test

teemtee · May 15, 2024 · 6fbfb08 · 6fbfb08
1 parent a3894a1
commit 6fbfb08
Showing 1 changed file with 30 additions and 6 deletions.
diff --git a/tests/unit/provision/mrack/test_hw.py b/tests/unit/provision/mrack/test_hw.py
@@ -9,6 +9,7 @@
     _parse_cpu,
     _parse_disk,
     _parse_hostname,
+    _parse_location,
     _parse_memory,
     _parse_virtualization,
     _parse_zcrypt,
@@ -231,6 +232,12 @@ def test_maximal_constraint(root_logger: Logger) -> None:
                     '_value': '%.foo.redhat.com'
                     }
                 },
+            {
+                'labcontroller': {
+                    '_op': '!=',
+                    '_value': 'lab-01.rhts.eng.pek2.redhat.com'
+                    }
+                },
             {'or': []},
             {
                 'and': [
@@ -278,12 +285,6 @@ def test_maximal_constraint(root_logger: Logger) -> None:
                     ]
                 },
 
-            {
-                'labcontroller': {
-                    '_op': '!=',
-                    '_value': 'lab-01.rhts.eng.pek2.redhat.com'
-                    }
-                },
             ]
         }
 
@@ -678,3 +679,26 @@ def test_zcrypt_mode(root_logger: Logger) -> None:
                 }
             }
         }
+
+
+def test_location_lab_controller(root_logger: Logger) -> None:
+
+    result = _CONSTRAINT_TRANSFORMERS['location.lab_controller'](
+        _parse_location({"lab-controller": "lab-01.rhts.eng.pek2.redhat.com"}), root_logger)
+
+    assert result.to_mrack() == {
+        'labcontroller': {
+            '_op': '==',
+            '_value': 'lab-01.rhts.eng.pek2.redhat.com'
+            }
+        }
+
+    result = _CONSTRAINT_TRANSFORMERS['location.lab_controller'](
+        _parse_location({"lab-controller": "!= lab-01.rhts.eng.pek2.redhat.com"}), root_logger)
+
+    assert result.to_mrack() == {
+        'labcontroller': {
+            '_op': '!=',
+            '_value': 'lab-01.rhts.eng.pek2.redhat.com'
+            }
+        }