"""Tests for reassign_artist_info in mysql __init__."""

import pytest
from pytest_mock import MockerFixture

from contributor.queries.mysql import (
    reassign_artist_info,
    release_artist,
    track_artist,
    track_writer,
)


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


class TestReassignArtistInfo:
    def test_delegates_to_all_three_reassign_methods(
        self, mocker: MockerFixture, session
    ) -> None:
        mock_release_artist = mocker.patch.object(
            release_artist, "reassign_artist_info"
        )
        mock_track_artist = mocker.patch.object(track_artist, "reassign_artist_info")
        mock_track_writer = mocker.patch.object(track_writer, "reassign_artist_info")

        reassign_artist_info(
            session=session,
            artist_name="DRUG CHURCH",
            artist_info_id=123,
            duplicate_artist_info_id=456,
        )

        expected_kwargs = dict(
            session=session,
            artist_name="DRUG CHURCH",
            artist_info_id=123,
            duplicate_artist_info_id=456,
        )
        mock_release_artist.assert_called_once_with(**expected_kwargs)
        mock_track_artist.assert_called_once_with(**expected_kwargs)
        mock_track_writer.assert_called_once_with(**expected_kwargs)