import httpx
from owsclient.test import OwsClientMock

from app.adapters.ows_dmp import AudienceExportStatus, OwsDmpClient


def test_update_audience_export(
    ows_dmp_client: OwsDmpClient, ows_client_mock: OwsClientMock
) -> None:
    export_id = "abc"
    audience_id = "def"
    audience_name = "test audience"
    status = AudienceExportStatus.COMPLETED
    zip_key = "test"
    identity_id = "ab11-cd22-ef33"

    ows_client_mock.patch(
        "ows-dmp",
        f"/audience-exports/{export_id}",
        json={
            "status": status,
            "zipKey": zip_key,
        },
    ).mock(
        return_value=httpx.Response(
            json={
                "id": export_id,
                "audienceId": audience_id,
                "audienceName": audience_name,
                "createdBy": identity_id,
            },
            status_code=200,
        )
    )

    export = ows_dmp_client.update_audience_export(
        export_id=export_id,
        status=AudienceExportStatus.COMPLETED,
        zip_key=zip_key,
    )

    assert export.id == export_id
    assert export.audience_id == audience_id
    assert export.audience_name == audience_name
    assert export.created_by == identity_id