stages(autotailor): add small unit test

Small unit test for PR#1407.

stages/test/test_autotailor.py

@@ -0,0 +1,117 @@
+#!/usr/bin/python3
+
+import os.path
+import sys
+from unittest.mock import call, patch
+
+import pytest
+
+import osbuild.meta
+from osbuild.testutil.imports import import_module_from_path
+
+TEST_INPUT = [
+    (
+        [
+            {"var": "ssh_idle_timeout_value", "value": 60},
+        ], [
+            "--var-value", "ssh_idle_timeout_value=60",
+        ],
+    ),
+    (
+        [
+            {"var": "ssh_idle_timeout_value", "value": "60"},
+        ], [
+            "--var-value", "ssh_idle_timeout_value=60",
+        ],
+    ),
+    (
+        [
+            {"var": "var1", "value": "val1"},
+            {"var": "var2", "value": "val2"}
+        ], [
+            "--var-value", "var1=val1", "--var-value", "var2=val2",
+        ],
+    ),
+]
+
+
+@pytest.fixture(name="fake_input")
+def fake_input_fixture():
+    return {
+        "name": "org.osbuild.oscap.autotailor",
+        "options": {
+            "filepath": "/some/filepath",
+            "config": {
+                "profile_id": "some-profile-id",
+                "datastream": "some-datastream",
+                "new_profile": "some-new-profile",
+            }
+        },
+    }
+
+
+def schema_validate_stage_oscap_autotailor(fake_input, test_data):
+    name = "org.osbuild.oscap.autotailor"
+    version = "1"
+    root = os.path.join(os.path.dirname(__file__), "../..")
+    mod_info = osbuild.meta.ModuleInfo.load(root, "Stage", name)
+    schema = osbuild.meta.Schema(mod_info.get_schema(version=version), name)
+    test_input = fake_input
+    test_input["options"]["config"].update(test_data)
+    return schema.validate(test_input)
+
+
+@pytest.mark.parametrize("test_overrides,expected", TEST_INPUT)
+@patch("subprocess.run")
+def test_oscap_autotailor_overrides_smoke(mock_subprocess_run, fake_input, test_overrides, expected):
+    stage_path = os.path.join(os.path.dirname(__file__), "../org.osbuild.oscap.autotailor")
+    stage = import_module_from_path("stage", stage_path)
+
+    options = fake_input["options"]
+    options["config"]["overrides"] = test_overrides
+    stage.main("/some/sysroot", options)
+
+    assert mock_subprocess_run.call_args_list == [
+        call(["/usr/bin/autotailor", "--output", "/some/sysroot/some/filepath",
+              "--new-profile-id", "some-new-profile"] +
+             expected +
+             ["some-datastream", "some-profile-id"],
+             encoding='utf8', stdout=sys.stderr, check=True)]
+
+
+@pytest.mark.parametrize(
+    "test_data,expected_err",
+    [
+        ({
+            "overrides": [
+                {
+                    "no": "var",
+                    "value": 50,
+                },
+            ]
+        }, "'var' is a required property"),
+        ({
+            "overrides": [
+                {
+                    "no": "value",
+                    "var": "some",
+                },
+            ]
+        }, "'value' is a required property"),
+        ({
+            "overrides": [
+                {
+                    "var": "ssh_idle_timeout_value",
+                    "value": {"some": "object"},
+                },
+            ]
+        }, " is not of type 'string', 'integer'"),
+    ],
+)
+def test_schema_validation_oscap_autotailor(fake_input, test_data, expected_err):
+    res = schema_validate_stage_oscap_autotailor(fake_input, test_data)
+
+    assert res.valid is False
+    err_msgs = [e.as_dict()["message"] for e in res.errors]
+    assert len(res.errors) == 1, err_msgs
+    assert expected_err in err_msgs[0]