"""Test datetime encoder."""

import datetime
import json

from video.utils.date_time_encoder import DateTimeEncoder


def test_date_time_encoder() -> None:
    """Test date time encoder of a datetime."""
    to_convert = {"a": datetime.datetime(2018, 1, 1, 12)}
    expected = '{"a": "2018-01-01T12:00:00"}'
    assert json.dumps(to_convert, cls=DateTimeEncoder) == expected


def test_date_encoder() -> None:
    """Test date encoder of a datetime."""
    to_convert = {"a": datetime.date(2018, 1, 1)}
    expected = '{"a": "2018-01-01"}'
    assert json.dumps(to_convert, cls=DateTimeEncoder) == expected


def test_date_time_encoder_of_non_datetime() -> None:
    """Test date time encoder of something else."""
    assert json.dumps({"a": 1}, cls=DateTimeEncoder) == '{"a": 1}'