def test_to_and_from_equality():
i = ObjectId()
m = FooEntity()
m.id = i
m.foo = 'giraffe'
mongo_document = utils.entity_to_mongo(m)
entity = utils.entity_from_mongo(FooEntity, mongo_document)
assert m == entity
mongo_document2 = utils.entity_to_mongo(entity)
assert mongo_document2 == mongo_document
def test_flex_entity_field():
f = FlexFooEntity()
f.foo = 'spider'
f.bar = dict(monkey='gorilla')
mongo_document = utils.entity_to_mongo(f)
assert 'foo' in mongo_document
assert 'bar' in mongo_document
assert isinstance(mongo_document['bar'], dict)
entity = utils.entity_from_mongo(FlexFooEntity, mongo_document)
assert entity.foo == 'spider'
assert hasattr(entity.bar, 'monkey')
def test_entity_from_mongo():
i = ObjectId()
m = FooEntity()
m.id = i
m.foo = 'gorilla'
mongo_document = utils.entity_to_mongo(m)
entity = utils.entity_from_mongo(FooEntity, mongo_document)
assert '_id' not in entity
assert 'id' in entity
assert entity['id'] == i
assert 'foo' in entity
assert entity['foo'] == 'gorilla'
