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))
