"""Tests for reassign_contributor_content."""

import pytest
from pytest_mock import MockerFixture

from contributor.queries.mysql import (
    project,
    reassign_contributor_content,
    release,
)


@pytest.fixture
def session(mocker: MockerFixture):
    return mocker.Mock()


class TestReassignContributorContent:
    def test_calls_both_reassign_methods(self, mocker: MockerFixture, session) -> None:
        reassign_releases = mocker.patch.object(
            release, "reassign_contributor_releases"
        )
        reassign_projects = mocker.patch.object(
            project, "reassign_contributor_projects"
        )

        reassign_contributor_content(
            session=session,
            contributor_artist_info_id=11,
            duplicate_artist_info_id=22,
        )

        expected_kwargs = dict(
            session=session,
            contributor_artist_info_id=11,
            duplicate_artist_info_id=22,
        )
        reassign_releases.assert_called_once_with(**expected_kwargs)
        reassign_projects.assert_called_once_with(**expected_kwargs)