from unittest.mock import MagicMock, patch

import pytest

from tmt.package_managers._rpm import RpmVersion


@pytest.mark.parametrize(
    ("raw_id", "error"),
    [
        ("copr.repository:invalid-id", "Invalid Copr repository format"),
        ("copr.repository:@/stable", "Invalid Copr repository format"),
        ("copr.repository:", "Missing Copr repository name"),
    ],
)
def test_invalid_repository_id(raw_id, error, artifact_provider):
    with pytest.raises(ValueError, match=error):
        artifact_provider(raw_id)


def test_fetch_contents_enables_repository(
    mock_copr_build_session, mock_package_manager, artifact_provider, tmppath
):
    mock_repo = MagicMock()
    mock_guest = MagicMock()
    mock_guest.package_manager = mock_package_manager

    provider = artifact_provider("copr.repository:@teemtee/stable")
    download_path = tmppath / "artifacts"

    # @teemtee/stable: is_group=True -> owner="group_teemtee"
    expected_repo_filename = "_copr:copr.fedorainfracloud.org:group_teemtee:stable.repo"

    with patch(
        'tmt.steps.prepare.artifact.providers.copr_repository.Repository.from_content',
        return_value=mock_repo,
    ) as mock_from_content:
        result = provider.fetch_contents(mock_guest, download_path)

    mock_package_manager.enable_copr.assert_called_once_with('@teemtee/stable')
    mock_guest.execute.assert_called_once()
    assert expected_repo_filename in str(mock_guest.execute.call_args[0][0])
    mock_from_content.assert_called_once_with(
        mock_guest.execute.return_value.stdout or '',
        '@teemtee/stable',
        provider.logger,
    )
    assert result == []
    assert provider.repository is mock_repo


def test_enumerate_artifacts(
    mock_copr_build_session, mock_package_manager, artifact_provider, tmppath
):
    mock_repo = MagicMock()
    mock_repo.repo_ids = ['group_teemtee-stable-fedora-42-x86_64']
    mock_guest = MagicMock()
    mock_guest.package_manager = mock_package_manager

    mock_package_manager.list_packages.return_value = [
        RpmVersion.from_nevra(
            'tmt-1.69.0-1.fc42.noarch', repo_id='group_teemtee-stable-fedora-42-x86_64'
        ),
        RpmVersion.from_nevra(
            'tmt-all-0:1.69.0-1.fc42.noarch', repo_id='group_teemtee-stable-fedora-42-x86_64'
        ),
    ]

    provider = artifact_provider("copr.repository:@teemtee/stable")

    with patch(
        'tmt.steps.prepare.artifact.providers.copr_repository.Repository.from_content',
        return_value=mock_repo,
    ):
        provider.fetch_contents(mock_guest, tmppath / "artifacts")

    provider.enumerate_artifacts(mock_guest)

    assert len(provider.artifacts) == 2
    assert provider.artifacts[0].version.name == 'tmt'
    assert provider.artifacts[0].version.version == '1.69.0'
    assert provider.artifacts[1].version.name == 'tmt-all'
    assert provider.artifacts[1].version.epoch == 0