import base64

import orjson

from app.schemas import KafkaRecord

from tests.unit.factories import EmailPresaveEventFactory


def test_parse_kafka_records() -> None:
    email_presave_event = EmailPresaveEventFactory.build()
    event_value = {
        "taskId": email_presave_event.task_id,
        "groupId": email_presave_event.group_id,
        "albumId": email_presave_event.album_id,
        "country": email_presave_event.country,
        "productUpc": email_presave_event.product_upc,
        "storeName": email_presave_event.store_name,
        "storeId": email_presave_event.store_id,
        "clientId": email_presave_event.client_id,
        "clientIp": email_presave_event.client_ip,
        "email": email_presave_event.email,
        "createdAt": email_presave_event.created_at.isoformat(),
    }

    record = KafkaRecord.model_validate(
        {
            "value": base64.b64encode(orjson.dumps(event_value)),
        }
    )

    assert record.value == event_value