def test_code_deregistration():
marshall = Marshall()
fcn = "eventz.marshall.DatetimeCodec"
marshall.register_codec(fcn=fcn, codec=DatetimeCodec())
assert marshall.has_codec(fcn) is True
marshall.deregister_codec(fcn)
assert marshall.has_codec(fcn) is False
def test_entity_with_custom_datetime_codec_serialised_to_json():
entity_name = "Entity One"
dt1 = datetime(2020, 1, 2, 3, 4, 5, 123456)
iso_dt1 = "2020-01-02T03:04:05.123456"
entity1 = CustomTypeEntity(name=entity_name, timestamp=dt1)
marshall = Marshall()
marshall.register_codec(fcn="eventz.marshall.DatetimeCodec",
codec=DatetimeCodec())
assert marshall.to_json(entity1) == json.dumps(
_make_with_datetime_dict(name=entity_name, iso_dt=iso_dt1))
