from tests import started_game, set_events, current_player, next_player, current_player_with_cards

from bang.expansions.wild_west_show.card_events import *
from bang.cards import Card, Suit


# test Camposanto
def test_camposanto():
    g = started_game(['wild_west_show'], 4)
    set_events(g, [Camposanto()])
    current_player_with_cards(g, []).end_turn()
    p = current_player_with_cards(g, [])
    p.lives = 0
    p.notify_self()
    p1 = current_player_with_cards(g, [])
    p1.lives = 0
    p1.notify_self()
    current_player_with_cards(g, []).end_turn()
    current_player_with_cards(g, []).end_turn()
    assert p.is_my_turn
    assert p.lives == 1
    current_player_with_cards(g, []).end_turn()
    assert p1.is_my_turn
    assert p1.lives == 1


# test DarlingValentine
def test_darling_valentine():
    g = started_game(['wild_west_show'], 4)
    set_events(g, [DarlingValentine()])
    p = next_player(g)
    hand = p.hand.copy()
    current_player_with_cards(g, []).end_turn()
    assert hand != current_player(g).hand


# test DorothyRage

# test HelenaZontero
def test_helena_zontero():
    g = started_game(['wild_west_show'], 6)
    set_events(g, [None, HelenaZontero()])
    roles = [p.role.name for p in g.players]
    for i in range(5):
        current_player_with_cards(g, []).end_turn()
    g.deck.cards = [Card(Suit.HEARTS, 'card', 0)]*5
    current_player_with_cards(g, []).end_turn()
    roles2 = [p.role.name for p in g.players]
    assert roles != roles2

# test LadyRosaDelTexas

# test MissSusanna

# test RegolamentoDiConti

# test Sacagaway

# test WildWestShow