Add unittests for PluginRequirement

Author: mdellweg

pulp-glue/tests/conftest.py

@@ -1,10 +1,19 @@
+import json
 import os
 import typing as t
 
 import pytest
 
 from pulp_glue.common.context import PulpContext
-from pulp_glue.common.openapi import BasicAuthProvider
+from pulp_glue.common.openapi import BasicAuthProvider, OpenAPI
+
+FAKE_OPENAPI_SPEC = json.dumps(
+    {
+        "openapi": "3.0.3",
+        "info": {"x-pulp-app-versions": {"core": "3.75.0"}},
+        "paths": {},
+    }
+)
 
 
 @pytest.fixture
@@ -22,6 +31,26 @@ def pulp_ctx(
     return PulpContext.from_config(settings)
 
 
+@pytest.fixture
+def mock_pulp_ctx(
+    request: pytest.FixtureRequest,
+    monkeypatch: pytest.MonkeyPatch,
+) -> PulpContext:
+    monkeypatch.setattr(
+        OpenAPI, "load_api", lambda self, refresh_cache: self._parse_api(FAKE_OPENAPI_SPEC)
+    )
+    monkeypatch.setattr(
+        OpenAPI,
+        "_send_request",
+        lambda *args, **kwags: pytest.fail("Invalid use of pulp_ctx fixture in a non live test."),
+    )
+
+    verbose = request.config.getoption("verbose")
+    settings: t.Dict[str, t.Any] = {"base_url": "nowhere"}
+    settings["debug_callback"] = lambda i, s: i <= verbose and print(s)
+    return PulpContext.from_config(settings)
+
+
 @pytest.fixture
 def fake_pulp_ctx(
     request: pytest.FixtureRequest, pulp_cli_settings: t.Dict[str, t.Dict[str, t.Any]]

pulp-glue/tests/test_plugin_requirement.py

@@ -0,0 +1,29 @@
+import pytest
+
+from pulp_glue.common.context import PluginRequirement, PulpContext
+from pulp_glue.common.exceptions import PulpException
+
+pytestmark = pytest.mark.glue
+
+
+def test_has_plugin(mock_pulp_ctx: PulpContext) -> None:
+    assert mock_pulp_ctx.has_plugin(PluginRequirement("core"))
+    assert mock_pulp_ctx.has_plugin(PluginRequirement("core", specifier=">=3.50"))
+    assert not mock_pulp_ctx.has_plugin(PluginRequirement("core", specifier=">=3.50,<3.60"))
+    assert not mock_pulp_ctx.has_plugin(PluginRequirement("core", specifier=">=3.50,<3.60"))
+    assert mock_pulp_ctx.has_plugin(
+        PluginRequirement("core", specifier=">=3.50,<3.60", inverted=True)
+    )
+
+    assert not mock_pulp_ctx.has_plugin(PluginRequirement("elephant"))
+    assert not mock_pulp_ctx.has_plugin(PluginRequirement("elephant", specifier=">=0.0.0"))
+    assert mock_pulp_ctx.has_plugin(
+        PluginRequirement("elephant", specifier=">=0.0.0", inverted=True)
+    )
+
+
+def test_needs_plugin(mock_pulp_ctx: PulpContext) -> None:
+    mock_pulp_ctx.needs_plugin(PluginRequirement("core"))
+    mock_pulp_ctx.needs_plugin(PluginRequirement("elephant", feature="zoo"))
+    with pytest.raises(PulpException, match=r"elephant.*zoo"):
+        mock_pulp_ctx.has_plugin(PluginRequirement("core"))