import pytest

from engine.game.game_state import initialize_game


class TestVerificationBatch28:
    @pytest.fixture
    def game_state(self):
        gs = initialize_game(use_real_data=True)
        if not gs.member_db:
            pytest.skip("Card data not loaded")
        gs.active_player.tapped_energy.fill(False)
        return gs

    def get_card_id(self, gs, card_no):
        for cid, card in gs.member_db.items():
            if card.card_no == card_no:
                return cid
        for cid, card in gs.live_db.items():
            if card.card_no == card_no:
                return cid
        pytest.fail(f"Card {card_no} not found")

    def _test_vanilla(self, game_state, card_no):
        gs = game_state
        card_id = self.get_card_id(gs, card_no)
        card = gs.member_db.get(card_id) or gs.live_db.get(card_id)
        assert card is not None

    def test_pln_sd1_020_sd_vanilla(self, game_state):
        self._test_vanilla(game_state, "PL!N-sd1-020-SD")

    def test_pln_sd1_023_sd_vanilla(self, game_state):
        self._test_vanilla(game_state, "PL!N-sd1-023-SD")

    def test_pln_sd1_024_sd_vanilla(self, game_state):
        self._test_vanilla(game_state, "PL!N-sd1-024-SD")

    def test_pln_sd1_025_sd_smoke(self, game_state):
        pass

    def test_pln_sd1_027_sd_smoke(self, game_state):
        pass

    def test_pls_pr_013_pr_smoke(self, game_state):
        pass

    def test_pls_pr_014_pr_vanilla(self, game_state):
        self._test_vanilla(game_state, "PL!S-PR-014-PR")

    def test_pls_pr_015_pr_vanilla(self, game_state):
        self._test_vanilla(game_state, "PL!S-PR-015-PR")

    def test_pls_pr_017_pr_vanilla(self, game_state):
        self._test_vanilla(game_state, "PL!S-PR-017-PR")

    def test_pls_pr_018_pr_vanilla(self, game_state):
        self._test_vanilla(game_state, "PL!S-PR-018-PR")