def test_clean(self):
self.assertEqual(mongodb.scrub(1), 1)
self.assertEqual(mongodb.scrub("string"), "string")
self.assertEqual(mongodb.scrub([1]), [1])
self.assertEqual(mongodb.scrub({'1': 'A'}), {'1': 'A'})
self.assertEqual(mongodb.scrub(True), True)
self.assertEqual(mongodb.scrub(False), False)
def test_failsafe(self, mock_encoder):
mock_encoder.side_effect = Exception("Surprise!")
with self.assertRaises(mongodb.ValidationError):
mongodb.scrub('safe')
def test_invalid(self):
with self.assertRaises(mongodb.ValidationError):
mongodb.scrub(self)
def test_blanks(self):
self.assertEqual(mongodb.scrub(''), '')
self.assertEqual(mongodb.scrub({}), {})
self.assertEqual(mongodb.scrub([]), [])
self.assertEqual(mongodb.scrub(0), 0)