def test_sanitize_fields(self):
fields = {
'field1': lambda x: Item.sanitize_non_blank_value(x, ''),
'field2': lambda x: Item.sanitize_integer_value(x, ''),
'field3': lambda x: Item.sanitize_string_value(x, '', blank=True),
}
data = Item.sanitize_fields(fields, field1='foobarbaz', field2=42, field3='')
self.assertDictEqual(data, {
'field1': 'foobarbaz',
'field2': 42,
'field3': ''
})
data = Item.sanitize_fields(fields, field1='foobarbaz', field2=42, field3='', field4='another')
self.assertDictEqual(data, {
'field1': 'foobarbaz',
'field2': 42,
'field3': ''
})
def test_sanitize_non_blank_value(self):
self.assertEqual(Item.sanitize_non_blank_value(0, ''), 0)
self.assertEqual(Item.sanitize_non_blank_value('a', ''), 'a')
self.assertListEqual(Item.sanitize_non_blank_value([1], ''), [1])
self.assertDictEqual(Item.sanitize_non_blank_value({'a': 1}, ''), {'a': 1})
self.assertTupleEqual(Item.sanitize_non_blank_value((1, 2), ''), (1, 2))
with self.assertRaises(ValueError):
Item.sanitize_non_blank_value('', '')
Item.sanitize_non_blank_value([], '')
Item.sanitize_non_blank_value({}, '')
Item.sanitize_non_blank_value((), '')
Item.sanitize_non_blank_value(None, '')
