def test_delete(self):
self.assertEqual(Entity.query.count(), 0)
entity = Entity(mbid='f27ec8db-af05-4f36-916e-3d57f91ecf5e', type='test')
db.session.add(entity)
db.session.flush()
self.assertEqual(Entity.query.count(), 1)
entity.delete()
self.assertEqual(Entity.query.count(), 0)
def test_get(self):
mbid = 'f27ec8db-af05-4f36-916e-3d57f91ecf5e'
entity = Entity(mbid=mbid, type='test')
db.session.add(entity)
db.session.flush()
same_entity = Entity.get(mbid)
self.assertEqual(entity, same_entity)
missing_csrf = Entity.get('85d9c621-e30f-4788-a962-a089c0d34182')
self.assertIsNone(missing_csrf)
def entity_data(mbid):
use_cache = not request.args.get('no_cache', False)
type_hint = request.args.get('type_hint', None)
entity = None
if use_cache:
entity = Entity.get(mbid=mbid)
if not entity:
entity = Entity.get_remote(mbid=mbid, type_hint=type_hint)
db.session.commit()
if not entity:
return Response(json.dumps({'entity': None}), status=404, mimetype='application/json')
return jsonify(entity=entity.to_dict())
def test_to_dict(self):
entity_1 = Entity(mbid='f27ec8db-af05-4f36-916e-3d57f91ecf5e', type='test')
db.session.add(entity_1)
db.session.flush()
self.assertDictEqual(entity_1.to_dict(),
{'mbid': entity_1.mbid, 'type': entity_1.type, 'data': {}})
# Trying with data
entity_2 = Entity(mbid='85d9c621-e30f-4788-a962-a089c0d34182', type='another_test', data='{"test": true}')
db.session.add(entity_2)
db.session.flush()
self.assertDictEqual(entity_2.to_dict(),
{'mbid': entity_2.mbid, 'type': entity_2.type, 'data': {'test': True}})
def test_delete(self):
entities = Entity.query.all()
assert len(entities) == 0
entity = Entity(mbid='f27ec8db-af05-4f36-916e-3d57f91ecf5e', type='test')
db.session.add(entity)
db.session.flush()
entities = Entity.query.all()
assert len(entities) == 1
assert entities[0] == entity
entity.delete()
entities = Entity.query.all()
assert len(entities) == 0
