"""Unit tests for ReferenceLoader."""

from unittest.mock import Mock, patch

import pytest

from src.services.reference_loader import ReferenceLoader


class TestReferenceLoader:
    """Tests for ReferenceLoader class."""

    @pytest.fixture
    def mock_duck_conn(self):
        """Mock DuckDB connection."""
        return Mock()

    @pytest.fixture
    def mock_gateway(self):
        """Mock RemoteGateway."""
        return Mock()

    @pytest.fixture
    def loader(self, mock_duck_conn, mock_gateway):
        """Create loader instance with mocked dependencies."""
        return ReferenceLoader(duck_conn=mock_duck_conn, gateway=mock_gateway)

    def test_initialization(self, loader, mock_duck_conn, mock_gateway):
        """Test service initialization."""
        assert loader._duck_conn is mock_duck_conn
        assert loader._gateway is mock_gateway

    @patch('src.services.reference_loader.TaskGraphRunner')
    def test_load_all(self, mock_runner_cls, loader, mock_gateway):
        """Test load_all execution."""
        mock_runner = Mock()
        mock_runner_cls.return_value = mock_runner

        loader.load_all()

        # Verify runner was created and run
        mock_runner_cls.assert_called_once()
        mock_runner.run.assert_called_once()