def test_register_unregister(self):
class Foo(object):
pass
assert isinstance(MyEntity.parent, MongoInstrumentedAttribute)
with pytest.raises(ValueError):
MongoClassRegistry.unregister(Foo)
assert MongoClassRegistry.is_registered(MyEntity)
# Registering a registered class raises a ValueError.
with pytest.raises(ValueError):
MongoClassRegistry.register(MyEntity, None)
MongoClassRegistry.unregister(MyEntity)
assert not MongoClassRegistry.is_registered(MyEntity)
