import unittest

from mastermind import Mastermind


class MastermindTestCase(unittest.TestCase):

    def setUp(self):
        self.game = Mastermind(10, 4, 6)

    def test_check_code_len(self):
        self.assertFalse(self.game.check_code([1, 1, 1]))
        self.assertTrue(self.game.check_code([1, 1, 1, 1]))
        self.assertFalse(self.game.check_code([1, 1, 1, 1, 1]))

    def test_check_code_range(self):
        self.assertFalse(self.game.check_code([0, 1, 1, 1]))
        self.assertTrue(self.game.check_code([1, 2, 3, 4]))
        self.assertTrue(self.game.check_code([6, 5, 4, 3]))
        self.assertFalse(self.game.check_code([7, 6, 6, 6]))


if __name__ == '__main__':
    unittest.main()