from src.engine.chengdu_mahjong_state import ChengduMahjongState
from src.engine.hand import Hand
from src.engine.mahjong_tile import MahjongTile


def test_set_missing_suit():
    """测试设置缺门功能"""
    state = ChengduMahjongState()
    state.set_missing_suit(0, "条")
    assert state.missing_suits[0] == "条", "测试失败：缺门设置为 '条' 后未正确更新"
    state.set_missing_suit(1, "筒")
    assert state.missing_suits[1] == "筒", "测试失败：缺门设置为 '筒' 后未正确更新"
    state.set_missing_suit(2, "万")
    assert state.missing_suits[2] == "万", "测试失败：缺门设置为 '万' 后未正确更新"

    try:
        state.set_missing_suit(0, "花")
    except ValueError:
        print("测试通过：设置无效缺门 '花' 抛出异常")
    else:
        raise AssertionError("测试失败：设置无效缺门 '花' 未抛出异常")


def test_can_win_with_sequence_and_pair():
    hand = Hand()
    hand.add_tile(MahjongTile("条", 1))
    hand.add_tile(MahjongTile("条", 2))
    hand.add_tile(MahjongTile("条", 3))
    hand.add_tile(MahjongTile("条", 4))
    hand.add_tile(MahjongTile("条", 4))
    hand.add_tile(MahjongTile("筒", 5))
    hand.add_tile(MahjongTile("筒", 6))
    hand.add_tile(MahjongTile("筒", 7))
    hand.add_tile(MahjongTile("万", 1))
    hand.add_tile(MahjongTile("万", 2))
    hand.add_tile(MahjongTile("万", 3))
    hand.add_tile(MahjongTile("筒", 8))
    hand.add_tile(MahjongTile("筒", 8))
    state = ChengduMahjongState()
    state.hands[0] = hand
    assert state.can_win(hand), "测试失败：顺子和对子应该可以胡牌"
    print("测试通过：顺子和对子可以胡牌")


def test_can_win_with_triplets_and_pair():
    """测试刻子和对子胡牌"""
    hand = Hand()
    hand.add_tile(MahjongTile("条", 1))
    hand.add_tile(MahjongTile("条", 2))
    hand.add_tile(MahjongTile("条", 3))
    hand.add_tile(MahjongTile("条", 1))
    hand.add_tile(MahjongTile("条", 2))
    hand.add_tile(MahjongTile("条", 3))
    hand.add_tile(MahjongTile("条", 1))
    hand.add_tile(MahjongTile("条", 2))
    hand.add_tile(MahjongTile("条", 3))
    hand.add_tile(MahjongTile("筒", 3))
    hand.add_tile(MahjongTile("筒", 3))
    hand.add_tile(MahjongTile("筒", 3))
    hand.add_tile(MahjongTile("筒", 4))
    hand.add_tile(MahjongTile("筒", 4))
    state = ChengduMahjongState()
    state.hands[0] = hand
    print(f"\n,state.hand[0]: {state.hands[0]}")
    assert state.can_win(state.hands[0]) == True, "测试失败：刻子和对子应该可以胡牌"


def test_can_win_with_mixed_groups():
    """测试顺子、刻子和对子混合胡牌"""
    hand = Hand()
    hand.add_tile(MahjongTile("条", 1))
    hand.add_tile(MahjongTile("条", 2))
    hand.add_tile(MahjongTile("条", 3))
    hand.add_tile(MahjongTile("筒", 4))
    hand.add_tile(MahjongTile("筒", 4))
    hand.add_tile(MahjongTile("筒", 4))
    hand.add_tile(MahjongTile("万", 7))
    hand.add_tile(MahjongTile("万", 8))
    hand.add_tile(MahjongTile("万", 9))
    hand.add_tile(MahjongTile("筒", 6))
    hand.add_tile(MahjongTile("筒", 6))
    state = ChengduMahjongState()
    assert state.can_win(hand), "测试失败：顺子、刻子和对子混合应该可以胡牌"


def test_cannot_win():
    """测试不能胡牌"""
    hand = Hand()
    hand.add_tile(MahjongTile("条", 1))
    hand.add_tile(MahjongTile("条", 2))
    hand.add_tile(MahjongTile("条", 3))
    hand.add_tile(MahjongTile("筒", 4))
    hand.add_tile(MahjongTile("筒", 5))
    hand.add_tile(MahjongTile("筒", 6))
    state = ChengduMahjongState()
    assert not state.can_win(hand), "测试失败：当前手牌不应可以胡牌"