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_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