import unittest2
from api.validators import validate_keys_exists, validate_territory, validate_number
from api.exceptions import ValidationError


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))