"""Test file for track model."""

from unittest.mock import MagicMock

from product_digital.models import ows_track
from product_digital.models import track
from tests.testutils import db


@db.test_schema
def test__get_tracks(product_id):
    """Test get tracks data for given product."""
    new_track_data = track.Track(
        release_id=product_id, meta_language='english'
    )
    db.seed_models(new_track_data)
    result = track._get_tracks(product_id)
    assert str(result[0].release_id) == product_id


@db.test_schema
def test_delete_all_tracks_success(product_id, mocker):
    """Test delete all tracks of the product."""
    mock_validation_response = MagicMock(status_code=200)
    mocker.patch.object(
        ows_track, 'delete_tracks_by_product_id',
        return_value=mock_validation_response
    )
    new_track_data = track.Track(
        release_id=product_id, meta_language='english'
    )
    db.seed_models(new_track_data)
    result = track.delete_all_tracks(product_id)
    assert ows_track.delete_tracks_by_product_id.called
    assert result.message == {'status': 'ok'}