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()
