Beispiel #1
0
class TestRepository(TestCase):
    def setUp(self):
        self.repo = Repository("test.txt")

    def test_get_scramble(self):
        self.assertEqual(self.repo.get_scramble().id, "Arthur")

    def test_IOError(self):
        with self.assertRaises(ScrambleException):
            repo = Repository("no.txt")
class Controller:
    def __init__(self, filename):
        """
        Initializing the controller
        :param filename:
        """
        self.__repo = Repository(
            filename)  # initialize the repository with the filename
        self.__validator = Validator(
        )  # initialize the validator for checking if the input is ok
        self.__scramble = self.__repo.get_scramble(
        )  # getting the randomly picked sentence from repo
        self.__undo = None

    def dead(self):
        """
        Check if it is either a win, lose or we can't decide yet
        :return:
        """
        if self.__scramble.score == 0:  # if the score is 0 it's a lose
            return "You Lose!"
        elif self.__scramble.sentance == self.__scramble.scramble:  # if the scramble sentence is solved it's a win
            return "You Win!"
        else:
            return True  # otherwise we cant decide yet

    def swap(self, word_1, letter_1, word_2, letter_2):
        """
        Swapping the letters
        :param word_1:
        :param letter_1:
        :param word_2:
        :param letter_2:
        :return:
        """
        self.__validator.validate_swap(
            word_1, letter_1, word_2, letter_2,
            self.__scramble.id)  # check if the input is correct
        self.__scramble.swap(
            word_1, letter_1, word_2, letter_2
        )  # swap the elements using the swap function from the Sentence class
        self.__scramble.dec_score(1)  # decrease the score by one
        self.__undo = [word_1, letter_1, word_2,
                       letter_2]  # add the move to undo

    def undo(self):
        """
        Undo the last operation
        :return:
        """
        if self.__undo is None:  # if we can not undo anymore we raise an error
            raise ControllerException("Error!!! Can't undo anymore!!!\n")
        else:  # otherwise we simply do the swap from the undo list once more
            self.__scramble.swap(self.__undo[0], self.__undo[1],
                                 self.__undo[2], self.__undo[3])
            # self.__scramble.inc()
            self.__undo = None  # undo becomes None because we don't want the user to do multiple undo operations

    def list_scramble(self):
        """
        Returns the str of the self.__scramble object for printing
        :return:
        """
        return str(self.__scramble)

    def _get_scramble(self):
        """
        This functions only purpose is testing for the controller
        :return:
        """
        return self.__scramble