from bulkperformancerights.logic import async
from bulkperformancerights.logic import export_status


def test_export_tracks(monkeypatch, db_fixture):
    def mock_tracks_json_to_s3(a, b):
        return True

    def mock_tracks_json_to_xlsx(a, b):
        return True

    json_method = 'bulkperformancerights.logic.exporter.tracks_json_to_s3'
    monkeypatch.setattr(json_method, mock_tracks_json_to_s3)
    xlsx_method = 'bulkperformancerights.logic.exporter.tracks_json_to_xlsx'
    monkeypatch.setattr(xlsx_method, mock_tracks_json_to_xlsx)

    status_obj = export_status.create('vendor', 7123)
    async.export_tracks(status_obj)
    status_list = export_status.get('vendor', 7123, None)
    assert len(status_list) > 0
    assert status_list[0].get('job_status') == 'exported'