Exemple #1
0
class Zoo(object):
    def __init__(self, name, capacity, budget):
        self.__validator = Validator()

        self.name = name
        self.capacity = capacity
        self.budget = budget

    @property
    def name(self):
        return self.__name

    @name.setter
    def name(self, name):
        self.__validator.validate_name(2, 'zoos', name)
        self.__name = name

    @property
    def capacity(self):
        return self.__capacity

    @capacity.setter
    def capacity(self, capacity):
        self.__validator.validate_number(1, capacity, 'capacity')
        self.__capacity = capacity

    @property
    def budget(self):
        return self.__budget

    @budget.setter
    def budget(self, budget):
        self.__validator.validate_number(0, budget, 'budget')
        self.__budget = budget
class ValidatorTest(unittest.TestCase):

    def setUp(self):
        self.v = Validator('../../database/zoos_test_database.db')

    def test_validate_name_returns_none_on_valid_name(self):
        self.assertEqual(None, self.v.validate_name(2, 'zoos', 'aa'))
        self.assertEqual(None, self.v.validate_name(0, 'zoos', 'aa'))
        self.assertEqual(None, self.v.validate_name(1, 'zoos', 'a'))

    def test_validate_name_raises_exception_on_invalid_name(self):
        self.assertRaises(Exception, self.v.validate_name, 2, 'zoos', 'a')
        self.assertRaises(Exception, self.v.validate_name, 1, 'zoos', '')

    def test_validate_name_raises_exception_on_used_name(self):
        self.assertRaises(
            Exception, self.v.validate_name, 2, 'zoos', 'Zoo Sofia'
        )

        self.assertRaises(
            Exception, self.v.validate_name, 2, 'animals', 'Nikky'
        )

        self.assertRaises(
            Exception, self.v.validate_name, 2, 'species', 'lion'
        )

    def test_validate_number_returns_none_on_valid_number(self):
        self.assertEqual(None, self.v.validate_number(1, 1, ''))
        self.assertEqual(None, self.v.validate_number(0, 0, ''))
        self.assertEqual(None, self.v.validate_number(0, 10, ''))
        self.assertEqual(None, self.v.validate_number(0, 3.56, ''))
        self.assertEqual(None, self.v.validate_number(0, 0.01, ''))

    def test_validate_number_raises_exception_on_invalid_number(self):
        self.assertRaises(Exception, self.v.validate_number, 2, 1, '')
        self.assertRaises(Exception, self.v.validate_number, 0, -1, '')
        self.assertRaises(Exception, self.v.validate_number, 1, 0, '')

    def test_validate_number_works_with_strings(self):
        self.assertEqual(None, self.v.validate_number(1, '1', ''))
        self.assertEqual(None, self.v.validate_number(0, '0', ''))
        self.assertEqual(None, self.v.validate_number(0, '10', ''))
        self.assertEqual(None, self.v.validate_number(0, '0.001', ''))

        self.assertRaises(Exception, self.v.validate_number, 2, '1', '')
        self.assertRaises(Exception, self.v.validate_number, 0, '-1', '')
        self.assertRaises(Exception, self.v.validate_number, 1, '0', '')
        self.assertRaises(Exception, self.v.validate_number, 1, '0.76', '')

    def test_validate_returns_none_on_valid_parameter(self):
        lambda_function = lambda x: x < 10
        self.assertEqual(None, self.v.validate(lambda_function, 2, ''))

        lambda_function = lambda x: x ** 3 == 8
        self.assertEqual(None, self.v.validate(lambda_function, 2, ''))

    def test_validate_raises_exception_on_invalid_parameter(self):
        lf = lambda x: x < 10
        self.assertRaises(Exception, self.v.validate, lf, 20, '')

        lf = lambda x: x ** 3 == 8
        self.assertRaises(Exception, self.v.validate, lf, 20, '')