import orjson as json


async def test_generate_export_file(mocker, client, auth):
    request = {
        "content-type": "text/csv;+charset=utf-8",
        "data": [
            {
                "Date": "11/17/21",
                "Country": "All+markets",
                "All+Streams": 5770460,
                "Source+Album+Page+Streams": 205483,
                "Source+Artist+Page+Streams": 388000,
                "Source+Chart+Streams": 469497,
                "Source+Collection+Streams": 1769280,
                "Source+Daily+Mix+Streams": 68792,
                "Source+Discover+Weekly+Streams": 247,
                "Source+Other+Streams": 172193,
                "Source+Playlists+Streams": 1968420,
                "Source+Play+Queue+Streams": 166517,
                "Source+Radio+Streams": 257361,
                "Source+Release+Radar+Streams": 65,
                "Source+Search+Streams": 304605,
                "Sony+Playlist+Streams": 0,
                "Spotify+Playlist+Streams": 0,
                "Other+Playlist+Streams": 0,
                "Shuffle+Streams": 0
            },
            {
                "Date": "11/18/21",
                "Country": "All+markets",
                "All+Streams": 6024561,
                "Source+Album+Page+Streams": 347706,
                "Source+Artist+Page+Streams": 423122,
                "Source+Chart+Streams": 494800,
                "Source+Collection+Streams": 1789430,
                "Source+Daily+Mix+Streams": 73646,
                "Source+Discover+Weekly+Streams": 239,
                "Source+Other+Streams": 173322,
                "Source+Playlists+Streams": 2004367,
                "Source+Play+Queue+Streams": 169158,
                "Source+Radio+Streams": 233187,
                "Source+Release+Radar+Streams": 87,
                "Source+Search+Streams": 315497,
                "Sony+Playlist+Streams": 0,
                "Spotify+Playlist+Streams": 0,
                "Other+Playlist+Streams": 0,
                "Shuffle+Streams": 0
            }
        ],
        "filename": "Easy+On+Me+-+All+markets+-+Spotify.csv"
    }
    response = await client.post("/api/services/export/", data=json.dumps(request), headers=auth)
    assert response.headers["Content-Type"] == "text/csv;+charset=utf-8"
    assert response.headers["Content-Disposition"] == "attachment; filename*=Easy+On+Me+-+All+markets+-+Spotify.csv"