"""Test file for product_provided_store_artists model."""

import pytest

from product_digital.connectors import mysql
from product_digital.models import product_provided_store_artists
from tests.testutils import db


@pytest.fixture
def release_id():
    """Return a release_id for testing."""
    return 1234


@db.test_schema
def test_create_product_provided_store_artists(release_id):
    """Test that a new product_provided_store_artists record is returned."""
    response = product_provided_store_artists.create(release_id)
    response_data = response.message

    assert response.status == 201
    assert response_data['release_id'] == 1234
    assert response_data['spotify'] == 'Y'


@db.test_schema
def test_create_product_provided_store_artists_persistence(release_id):
    """Test that a new product_provided_store_artists record is created."""
    product_provided_store_artists.create(release_id)

    with mysql.db_session() as session:
        last_record = session.query(
            product_provided_store_artists.ProductProvidedStoreArtists
        ).order_by(
            product_provided_store_artists.ProductProvidedStoreArtists
            .product_provided_store_artists_id.desc()
        ).first()

        assert last_record.release_id == release_id