"""Contains tests for chain module"""
import unittest

from shenzhen_solitaire.board import NumberCard, Position
from shenzhen_solitaire.solver.board_actions import MoveAction, BunkerizeAction, GoalAction, HuaKillAction
from shenzhen_solitaire.solver import board_possibilities
from .boards import TEST_BOARD


class ChainTest(unittest.TestCase):
    """Tests the chain class"""

    def test_sequence(self) -> None:
        """Tests a given sequence. Might break if I change the iterators"""

        self.assertTrue(TEST_BOARD.check_correct())
        sequence = [
            MoveAction(
                cards=[
                    NumberCard(
                        suit=NumberCard.Suit.Red,
                        number=7),
                    NumberCard(
                        suit=NumberCard.Suit.Green,
                        number=6)],
                source_id=3,
                destination_id=7),
            BunkerizeAction(
                card=NumberCard(
                    suit=NumberCard.Suit.Red,
                    number=6),
                bunker_id=0,
                field_id=2,
                to_bunker=True),
            GoalAction(
                card=NumberCard(
                    suit=NumberCard.Suit.Green,
                    number=1),
                source_id=2,
                source_position=Position.Field),
            MoveAction(
                cards=[
                    NumberCard(
                        suit=NumberCard.Suit.Red,
                        number=4)],
                source_id=2,
                destination_id=5),
            GoalAction(
                card=NumberCard(
                    suit=NumberCard.Suit.Red,
                    number=1),
                source_id=2,
                source_position=Position.Field),
            HuaKillAction(source_field_id=2),
            MoveAction(
                cards=[
                    NumberCard(
                        suit=NumberCard.Suit.Black, number=9),
                    NumberCard(suit=NumberCard.Suit.Red, number=8)],
                source_id=6,
                destination_id=2),
            GoalAction(
                card=NumberCard(
                    suit=NumberCard.Suit.Green, number=2),
                source_id=6,
                source_position=Position.Field)
        ]
        for action in sequence:
            step = list(board_possibilities.possible_actions(TEST_BOARD))
            self.assertIn(action, step)
            action.apply(TEST_BOARD)