"""Unit tests for AssetFinal model."""

from asset_transcoder.connectors import mysql
from asset_transcoder.constants import asset_types as asset_type_const
from asset_transcoder.models import asset_final


def test_create_asset_final_success(db_fixture):
    """Test asset final is created."""
    payload = dict(
        asset_upload_id=1,
        asset_type=asset_type_const.TYPE_FILE_JPG,
        asset_subtype=asset_type_const.SUBTYPE_LARGE_COVER,
        filename='new_filename_output',
        duration=0
    )
    with mysql.transcoder_db_session() as session:
        payload['session'] = session
        result = asset_final.create_asset_final(**payload)
        assert result
        assert result['filename'] == payload['filename']
        assert isinstance(result['id'], int)


def test_get_asset_finals(db_fixture):
    """Test get asset finals by asset_upload_id."""
    result = asset_final.get_asset_finals(2)
    assert result
    assert result['items'] == [
        {
            'id': 2, 'asset_upload_id': 2, 'asset_type': 'JPG', 'asset_subtype': 'large_cover',
            'filename': 'test_bucket_filename_2', 'duration': 900
        }
    ]