fix: update test import _ALLOWED_SUBSCRIPTION_NAMES → _ALLOWED_SUBSCRIPTION_FIELDS

This commit is contained in:
Jacob Magar
2026-03-13 10:51:53 -04:00
parent 6fe6e30f1e
commit e24ef5e85d

View File

@@ -8,7 +8,7 @@ import pytest
from unraid_mcp.core.exceptions import ToolError from unraid_mcp.core.exceptions import ToolError
from unraid_mcp.subscriptions.diagnostics import ( from unraid_mcp.subscriptions.diagnostics import (
_ALLOWED_SUBSCRIPTION_NAMES, _ALLOWED_SUBSCRIPTION_FIELDS,
_validate_subscription_query, _validate_subscription_query,
) )
@@ -16,7 +16,7 @@ from unraid_mcp.subscriptions.diagnostics import (
class TestValidateSubscriptionQueryAllowed: class TestValidateSubscriptionQueryAllowed:
"""All whitelisted subscription names must be accepted.""" """All whitelisted subscription names must be accepted."""
@pytest.mark.parametrize("sub_name", sorted(_ALLOWED_SUBSCRIPTION_NAMES)) @pytest.mark.parametrize("sub_name", sorted(_ALLOWED_SUBSCRIPTION_FIELDS))
def test_all_allowed_names_accepted(self, sub_name: str) -> None: def test_all_allowed_names_accepted(self, sub_name: str) -> None:
query = f"subscription {{ {sub_name} {{ data }} }}" query = f"subscription {{ {sub_name} {{ data }} }}"
result = _validate_subscription_query(query) result = _validate_subscription_query(query)