"""Tests."""

import json

from neo4j.time import DateTime
from participant.utils.json_encoder import JSONNeo4jEncoder


def test_neo4j_date_encode():
    """Test that the JSONNeo4jEncoder encodes a neotime.DateTime correctly."""
    datetime = '2020-01-01T00:00:00.000000000+00:00'
    encoded = json.dumps(
        DateTime.from_iso_format(datetime),
        cls=JSONNeo4jEncoder,
    )
    assert encoded == f'"{datetime}"'