feat: Create a function that can enumerate Oracle classes

tests/test_oracle_type_class_lister.py

@@ -0,0 +1,53 @@
+# SPDX-FileCopyrightText: 2025 2025
+# SPDX-FileContributor: Gergely Polonkai
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+"""Tests for the type class lister"""
+
+import pytest
+
+from gm_assistant.oracle import generate_type_classes
+from gm_assistant.oracle.base import Oracle
+from gm_assistant.oracle.object_generator import ObjectGeneratorOracle
+
+
+class _TestOracle(Oracle):
+    """Test oracle class that has the same marker as ObjectGeneratorOracle"""
+
+    TYPE_MARKER = ObjectGeneratorOracle.TYPE_MARKER
+
+    def generate(self) -> str:  # pragma: no cover
+        return ""
+
+
+def test_generate_empty() -> None:
+    """Test generating the type class list from an empty dictionary"""
+
+    assert generate_type_classes({}) == {}  # pylint: disable=use-implicit-booleaness-not-comparison
+
+
+def test_nontype_not_present() -> None:
+    """Test that non-types don’t get included in the results"""
+
+    assert generate_type_classes({"test": True}) == {}  # pylint: disable=use-implicit-booleaness-not-comparison
+
+
+def test_non_oracle_not_present() -> None:
+    """Test that non-oracle types don’t get included in the results"""
+
+    assert generate_type_classes({"test": dict}) == {}  # pylint: disable=use-implicit-booleaness-not-comparison
+
+
+def test_oracle_not_present() -> None:
+    """Test that the ``Oracle`` class doesn’t get included in the results"""
+
+    assert generate_type_classes({"oracle": Oracle}) == {}  # pylint: disable=use-implicit-booleaness-not-comparison
+
+
+def test_duplace_type_marker() -> None:
+    """Test if ``generate_type_classes`` raises an error if a type marker appears twice"""
+
+    with pytest.raises(KeyError) as ctx:
+        generate_type_classes({"ObjectGeneratorOracle": ObjectGeneratorOracle, "TestOracle": _TestOracle})
+
+    assert str(ctx.value) == "'ObjectGeneratorOracle is already registered as a handler for object-generator'"