def test_wrong_params_fetch_zero_results(self):
params = dict(freq='biba',
name='boba',
start_date=date(year=2005, month=1, day=1),
end_date=date(year=2006, month=1, day=1))
number_of_results = select_datapoints(**params).count()
self.assertEqual(number_of_results, 0)
def test_upsert_updates_value_for_existing_row(self):
upsert(self.dp1_dict)
dp1_updated_value = self.dp1_dict['value'] + 4.56
dp1_dict_with_new_value = {
k: v if k != "value" else dp1_updated_value
for k, v in self.dp1_dict.items()
}
upsert(dp1_dict_with_new_value)
datapoint = select_datapoints(**self.dp1_search_param).first()
self.assertEqual(datapoint.value, dp1_updated_value)
def test_data_is_fetching(self):
params = dict(freq='m',
name='CPI_ALCOHOL_rog',
start_date=date(year=2016, month=6, day=1),
end_date=date(year=2016, month=7, day=1))
datapoint = select_datapoints(**params).first()
self.assertEqual(
datapoint.serialized, {
"date": "2016-06-30",
"freq": "m",
"name": "CPI_ALCOHOL_rog",
"value": 100.6
})
def test_after_upsert_datapoint_found(self):
upsert(self.dp1_dict)
datapoint = select_datapoints(**self.dp1_search_param).first()
self.assertEqual(datapoint.serialized, self.dp1_dict)
def test_before_upsert_datapoint_not_found(self):
datapoints_count = select_datapoints(**self.dp1_search_param).count()
self.assertEqual(datapoints_count, 0)
def get_datapoints(self):
datapoint_params = DatapointParameters(self.params).get()
return select_datapoints(**datapoint_params)
def get_datapoints():
params = utils.DatapointParameters(request.args).get()
data = queries.select_datapoints(**params)
fmt = request.args.get('format')
return get_datapoints_response(data, fmt)