def test_speed(self): # get data of all speed sensors rv, json = self.client.get('/trafficSpeed') self.assertTrue(rv.status_code == 200) assert isinstance(json.pop('speedSensor'), list) #get a link by linkId rv, json = self.client.get('/trafficSpeed/137') self.assertTrue(rv.status_code == 200) self.assertTrue('dataAsOf' and 'sensorId' and 'linkId' and 'speed' and 'travelTime' in json['speedSensor'][0]) #invalid id rv, json = self.client.get('/trafficSpeed/00') self.assertTrue(rv.status_code == 404) #get sensor data by id for for date today = dt.strftime(dt.now(), "%Y-%m-%d") rv, json = self.client.get('/trafficSpeed/137/'+today) self.assertTrue(rv.status_code == 200) self.assertTrue('measures' in json['speedSensor'][0]) #invalid date rv, json = self.client.get('/trafficSpeed/137/2016-22-30') self.assertTrue(rv.status_code == 404) rv, json = self.client.get('/trafficSpeed/137/2016-05-32') self.assertTrue(rv.status_code == 404) rv, json = self.client.get('/trafficSpeed/137/2020-02-02') self.assertTrue(rv.status_code == 404) rv, json = self.client.get('/trafficSpeed/137/a30') self.assertTrue(rv.status_code == 404) #no sensor data from common.models import Sensor Sensor.drop_collection() rv, json = self.client.get('/trafficSpeed') self.assertTrue(rv.status_code == 404)
def setUp(self): app.config['TESTING'] = True self.app = app Sensor.drop_collection() Link.drop_collection()