Python Entity.delete Examples

Example #1

File: entity_test.py Project: JesseWeinstein/geordi

    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)

Example #2

File: entity_test.py Project: reosarevok/geordi

    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