def test_localization(self):
class ModelMock(Model):
pass
key = Key(ModelMock, 1)
field = KeyField()
field._set_attr_name('n')
self.assertEqual(1, field.localize(key))
def test_validation_with_kind(self):
class ModelMock(Model):
pass
key = Key(ModelMock, 1)
field = KeyField(ModelMock)
self.assertIsNone(field.validate(None))
self.assertIsNone(field.validate(''))
self.assertIsNone(field.validate(key))
self.assertIsNone(field.validate(key.urlsafe()))
self.assertIsNone(field.validate('1'))
self.assertEqual('Invalid key', field.validate('abcd'))
def test_normalization_wit_kind(self):
class ModelMock(Model):
pass
key = Key(ModelMock, 1)
field = KeyField(ModelMock)
self.assertIsNone(field.normalize(None))
self.assertIsNone(field.normalize(''))
self.assertEqual(key, field.normalize(key))
self.assertEqual(key, field.normalize(key.urlsafe()))
self.assertEqual(key, field.normalize('1'))
self.assertRaises(Exception, field.normalize, 'abcd')
