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