18 lines
673 B
Python
18 lines
673 B
Python
import unittest2
|
|
from api.validators import ValidationError, validate_keys_exists, validate_territory
|
|
|
|
|
|
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'))
|
|
|
|
def test_validate_territory(self):
|
|
self.assertTrue(validate_territory('SE'))
|
|
self.assertRaises(ValidationError, validate_territory, 'unknown')
|