#!/usr/bin/python


import wordle

from wordle import LetterResult

import unittest


class TestConstraints(unittest.TestCase):
    def test_empty(self):
        constraints = wordle.Constraints()
        self.assertTrue(constraints.matches("abcde"))
        self.assertTrue(constraints.matches("bcdef"))

    def test_correct(self):
        constraints = wordle.Constraints()
        constraints.update_graded(
            "abcde",
            [
                LetterResult.CORRECT,
                LetterResult.INCORRECT,
                LetterResult.INCORRECT,
                LetterResult.INCORRECT,
                LetterResult.INCORRECT,
            ],
        )

        self.assertTrue(constraints.matches("amnop"))
        self.assertTrue(constraints.matches("aaaaa"))

        self.assertFalse(constraints.matches("abcde"))
        self.assertFalse(constraints.matches("lmnop"))
        self.assertFalse(constraints.matches("aedcb"))
        self.assertFalse(constraints.matches("amnob"))

    def test_correct_count_limit(self):
        constraints = wordle.Constraints()
        constraints.update_graded(
            "aazzz",
            [
                LetterResult.CORRECT,
                LetterResult.INCORRECT,
                LetterResult.INCORRECT,
                LetterResult.INCORRECT,
                LetterResult.INCORRECT,
            ],
        )

        self.assertTrue(constraints.matches("abcde"))
        self.assertTrue(constraints.matches("abbbb"))

        self.assertFalse(constraints.matches("aabbb"))
        self.assertFalse(constraints.matches("abbba"))
        self.assertFalse(constraints.matches("bbbbb"))
        self.assertFalse(constraints.matches("bbbba"))
        self.assertFalse(constraints.matches("babbb"))

    def test_move(self):
        constraints = wordle.Constraints()
        constraints.update_graded(
            "abcde",
            [
                LetterResult.MOVE,
                LetterResult.INCORRECT,
                LetterResult.INCORRECT,
                LetterResult.INCORRECT,
                LetterResult.INCORRECT,
            ],
        )

        self.assertTrue(constraints.matches("zzzza"))
        self.assertTrue(constraints.matches("zzzaa"))
        self.assertTrue(constraints.matches("zaaaa"))

        self.assertFalse(constraints.matches("azzzz"))
        self.assertFalse(constraints.matches("zzzzz"))
        self.assertFalse(constraints.matches("zzzba"))


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