def test_data_parameters_period_removes_end_date(self):
args = {'period': '1y',
'end_date': '2015-02-01'}
period = date.today() + relativedelta(years=-1)
result = {'Start time':
{'$gte': datetime.combine(period, datetime.min.time())}}
assert time_api_controller.get_data_parameters(args) == result
def test_data_parameters_end_date(self):
args = {'end_date': '2015-01-31'}
result = {'End time':
{'$lte': datetime.strptime(args.get('end_date'),
self.date_format)
}
}
assert time_api_controller.get_data_parameters(args) == result
def test_data_parameters_category(self):
args = {'category': 'test'}
result = {'Category': args.get('category')}
assert time_api_controller.get_data_parameters(args) == result
def test_data_parameters_period_years_months_days(self):
args = {'period': '1y3m2d'}
period = date.today() + relativedelta(years=-1, months=-3, days=-2)
result = {'Start time':
{'$gte': datetime.combine(period, datetime.min.time())}}
assert time_api_controller.get_data_parameters(args) == result
