Esempio n. 1
0
    def testRemoveLocationByName(self):
        test_location_name = 'test_location'
        db.session.add(Location(name=test_location_name))
        db.session.commit()

        assert Location.getLocationByName(test_location_name) is not None
        Location.removeLocationByName(test_location_name)
        db.session.commit()
        assert Location.getLocationByName(test_location_name) is None
Esempio n. 2
0
 def iterLocations(self):
     for l in LOCATIONS:
         loc = Location.getLocationByName(l['name'])
         yield l, loc
         db.session.add(loc)
     db.session.commit()