mirror of
https://github.com/jmagar/unraid-mcp
synced 2026-04-21 13:37:53 +00:00
74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
# tests/test_customization.py
|
|
"""Tests for customization subactions of the consolidated unraid tool."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
from conftest import make_tool_fn
|
|
|
|
|
|
@pytest.fixture
|
|
def _mock_graphql():
|
|
with patch("unraid_mcp.core.client.make_graphql_request", new_callable=AsyncMock) as m:
|
|
yield m
|
|
|
|
|
|
def _make_tool() -> Any:
|
|
return make_tool_fn("unraid_mcp.tools.unraid", "register_unraid_tool", "unraid")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_theme_returns_customization(_mock_graphql):
|
|
_mock_graphql.return_value = {
|
|
"customization": {"theme": {"name": "azure"}, "partnerInfo": None, "activationCode": None}
|
|
}
|
|
result = await _make_tool()(action="customization", subaction="theme")
|
|
assert result["customization"]["theme"]["name"] == "azure"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_public_theme(_mock_graphql):
|
|
_mock_graphql.return_value = {"publicTheme": {"name": "black"}}
|
|
result = await _make_tool()(action="customization", subaction="public_theme")
|
|
assert result["publicTheme"]["name"] == "black"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_is_initial_setup(_mock_graphql):
|
|
_mock_graphql.return_value = {"isInitialSetup": False}
|
|
result = await _make_tool()(action="customization", subaction="is_initial_setup")
|
|
assert result["isInitialSetup"] is False
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_set_theme_requires_theme(_mock_graphql):
|
|
from unraid_mcp.core.exceptions import ToolError
|
|
|
|
with pytest.raises(ToolError, match="theme_name"):
|
|
await _make_tool()(action="customization", subaction="set_theme")
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_set_theme_success(_mock_graphql):
|
|
_mock_graphql.return_value = {
|
|
"customization": {"setTheme": {"name": "azure", "showBannerImage": True}}
|
|
}
|
|
result = await _make_tool()(action="customization", subaction="set_theme", theme_name="azure")
|
|
assert result["success"] is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sso_enabled_returns_true(_mock_graphql):
|
|
_mock_graphql.return_value = {"isSSOEnabled": True}
|
|
result = await _make_tool()(action="customization", subaction="sso_enabled")
|
|
assert result["isSSOEnabled"] is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sso_enabled_returns_false(_mock_graphql):
|
|
_mock_graphql.return_value = {"isSSOEnabled": False}
|
|
result = await _make_tool()(action="customization", subaction="sso_enabled")
|
|
assert result["isSSOEnabled"] is False
|