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