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