"""Test for release_asset_version model."""
from product.models import release_asset_version as release_asset_version_model
from tests.factories.release_asset_version import ReleaseAssetVersionFactory
from tests.testutils import db


@db.test_schema
def test_get_artist_asset_version(test_product):
    """Test get_artist_asset_version with valid product id."""
    db.seed_models(ReleaseAssetVersionFactory.build(release_id=test_product))
    db.seed_models(ReleaseAssetVersionFactory.build(
        release_id=test_product, api_version='legacy'))
    response = release_asset_version_model.get(
        test_product)

    assert response.status == 200
    assert response.message.get('release_id') == test_product
    assert response.message.get('api_version') == 'legacy'


@db.test_schema
def test_get_artist_asset_version_not_found():
    """Test get_artist_asset_version with product id not found."""
    db.seed_models(ReleaseAssetVersionFactory.build(release_id=1234))
    db.seed_models(ReleaseAssetVersionFactory.build(
        release_id=1234, api_version='legacy'))

    response = release_asset_version_model.get(5678)

    assert response.status == 200
    assert response.message.get('api_version') == 'unknown'


@db.test_schema
def test_update_product_asset_version(test_product):
    """Test update_product_asset_version with valid product id."""
    updated_api_version = 'v2'
    db.seed_models(ReleaseAssetVersionFactory.build(release_id=test_product))
    db.seed_models(ReleaseAssetVersionFactory.build(
        release_id=test_product, api_version='legacy'))
    update_response = release_asset_version_model.update_api_version(
        test_product, api_version=updated_api_version)
    updated_product_response = release_asset_version_model.get(test_product)

    assert update_response.status == 200
    assert updated_product_response.message.get(
        'release_id') == test_product
    assert updated_product_response.message.get(
        'api_version') == updated_api_version


@db.test_schema
def test_update_product_asset_version_not_found():
    """Test update_product_asset_version with non-existent product id."""
    updated_api_version = 'v2'
    db.seed_models(ReleaseAssetVersionFactory.build(release_id=1234))
    db.seed_models(ReleaseAssetVersionFactory.build(
        release_id=1234, api_version='legacy'))
    response = release_asset_version_model.update_api_version(
        product_id=5678, api_version=updated_api_version)

    assert response.status == 404