13 lines
474 B
Python
13 lines
474 B
Python
import unittest2
|
|
from api.validators import ValidationError, validate_keys_exists
|
|
|
|
|
|
class TestValidation(unittest2.TestCase):
|
|
|
|
def test_validate_keys_exists(self):
|
|
obj = {'a':1, 'b':2}
|
|
self.assertRaises(ValidationError, validate_keys_exists, obj, 'c')
|
|
self.assertRaises(ValidationError, validate_keys_exists, obj, 'a', 'b', 'c')
|
|
self.assertTrue(validate_keys_exists(obj, 'a'))
|
|
self.assertTrue(validate_keys_exists(obj, 'a', 'b'))
|