def test_link(self): # get all links rv, json = self.client.get('/trafficLink/') self.assertTrue(rv.status_code == 200) assert isinstance(json.pop('trafficLinkList'), list) #get a link by linkId rv, json = self.client.get('/trafficLink/4616337') self.assertTrue(rv.status_code == 200) self.assertTrue('borough' and 'encodedPolyLine' and 'encodedPolyLineLvls' and 'linkId' and 'linkName' and 'linkPoints' and 'owner' in json['Link'][0]) #invalid id rv, json = self.client.get('/trafficLink/123456789') self.assertTrue(rv.status_code == 404) #no link data from common.models import Link Link.drop_collection() rv, json = self.client.get('/trafficLink/') self.assertTrue(rv.status_code == 404)
def setUp(self): app.config['TESTING'] = True self.app = app Sensor.drop_collection() Link.drop_collection()