"""Unit tests for custom Marshmallow fields."""

import datetime

from analytics.schemas import fields


class TestUnifiedDateField:
    """Tests for the custom UnifiedDate field."""

    def test_serialize(self):
        """Test field serializes to unified format."""
        field = fields.UnifiedDate()

        expected_value = "2018-01-01T00:00:00Z"
        obj = {
            "string": "2018-01-01",
            "datetime": datetime.datetime(2018, 1, 1, 0, 0),
            "date": datetime.date(2018, 1, 1),
            "longstring": "2018-01-01T00:00:00Z",
        }

        for key, value in obj.items():
            assert field.serialize(key, obj) == expected_value