def test_get_user(self):
user = User.create(name='Teste 1')
forecast_dict = {
'address': 'Rua Marshmallow',
'notification': '07:00:00'
}
forecast_dict["period"] = {
'period_from': '08:00:00',
'period_to': '19:00:00'
}
forecast_dict["days"] = {
'sunday': True,
'monday': True,
'tuesday': True,
'wednesday': True,
'thursday': True,
'friday': False,
'saturday': False
}
forecast = Forecast.create(user=user, **forecast_dict)
Period.create(period=forecast, **forecast_dict["period"])
Days.create(days=forecast, **forecast_dict["days"])
data = {'id': user.id, 'name': 'Teste 1', 'forecast': [forecast_dict]}
url = "/users/%s" % user.id
assert data == self.app.get(url).json
def test_forecast_get(self):
user = User.create(name='Teste update')
forecast_dict = {
'user_id': user.id,
'address': 'Rua Marshmallow',
'notification': '07:00:00'
}
forecast_dict["period"] = {
'period_from': '08:00:00',
'period_to': '19:00:00'
}
forecast_dict["days"] = {
'sunday': True,
'monday': True,
'tuesday': True,
'wednesday': True,
'thursday': True,
'friday': False,
'saturday': False
}
forecast = Forecast.create(user=user, **forecast_dict)
Period.create(period=forecast, **forecast_dict["period"])
Days.create(days=forecast, **forecast_dict["days"])
forecast_dict["id"] = forecast.id
url = "users/%s/forecast/%s" % (user.id, forecast.id)
rv = self.app.get(url, content_type='application/json').json
assert forecast_dict == rv
url = "users/1/forecast/540"
rv = self.app.get(url, content_type='application/json').json
assert {"message": "Forecast could not be found"} == rv
