24 lines
959 B
Python
24 lines
959 B
Python
import unittest2
|
|
from api.validators import ValidationError, validate_keys_exists, validate_territory, validate_number
|
|
|
|
|
|
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')
|
|
|
|
def test_validate_number(self):
|
|
self.assertRaises(ValidationError, validate_number, 'foo')
|
|
self.assertRaises(ValidationError, validate_number, 'foo', 12)
|
|
self.assertTrue(validate_number(12))
|
|
self.assertTrue(validate_number(12, 45))
|