def test_get_created_on_date_multiple(self): def query_args_func(key): args = {"created_on": ["20141211", "20141110"]} return args.get(key, []) expected = datetime.date(2014, 11, 10) self.assertEqual(expected, get_created_on_date(query_args_func))
def test_get_created_on_date_valid_format_2(self): def query_args_func(key): args = {"created_on": "20141212"} return args.get(key, []) expected = datetime.date(2014, 12, 12) self.assertEqual(expected, get_created_on_date(query_args_func))
def test_get_created_on_date_wrong_format(self): def query_args_func(key): args = { "created_on": "201412121243" } return args.get(key, []) self.assertIsNone(get_created_on_date(query_args_func))
def test_get_created_on_date_wrong_type(self): def query_args_func(key): args = { "created_on": 2014 } return args.get(key, []) self.assertIsNone(get_created_on_date(query_args_func))
def test_get_created_on_raise_attribute_error_not_valid(self): patcher = mock.patch("datetime.datetime", spec=True) patched_datetime = patcher.start() self.addCleanup(patcher.stop) patched_datetime.strptime.side_effect = AttributeError( "Attribute error") def query_args_func(key): args = {"created_on": "2015-07-02"} return args.get(key, []) self.assertIsNone(get_created_on_date(query_args_func))
def test_get_created_on_raise_attribute_error_valid(self): a_date = datetime.datetime.strptime("2015-07-02", "%Y-%m-%d") expected = datetime.date(2015, 07, 02) patcher = mock.patch("datetime.datetime", spec=True) patched_datetime = patcher.start() self.addCleanup(patcher.stop) patched_datetime.strptime.side_effect = [ AttributeError("Attribute error"), a_date ] def query_args_func(key): args = {"created_on": "2015-07-02"} return args.get(key, []) self.assertEqual(expected, get_created_on_date(query_args_func))
def test_get_created_on_date_missing(self): def query_args_func(key): args = {} return args.get(key, []) self.assertIsNone(get_created_on_date(query_args_func))