bang/backend/tests/__init__.py
2024-06-13 15:32:52 +03:00

44 lines
1.1 KiB
Python

from typing import Any, List
import pytest
from bang.characters import Character
from bang.game import Game
from bang.players import Player
from tests.dummy_socket import DummySocket
from globals import G
G.sio = DummySocket()
def started_game(expansions=[], players=4, character=Character("test_char", 4)) -> Game:
g = Game("test")
g.expansions = expansions
ps = [Player(f"p{i}", f"p{i}") for i in range(players)]
for p in ps:
g.add_player(p)
g.start_game()
for p in ps:
p.available_characters = [character]
if "high_noon" in expansions:
p.available_characters.append(Character("test_char2", 4))
p.set_character(p.available_characters[0].name)
return g
def set_events(g: Game, event_cards) -> None:
g.deck.event_cards = event_cards
def current_player(g: Game) -> Player:
return g.players[g.turn]
def next_player(g: Game) -> Player:
return g.players[(g.turn + 1) % len(g.players)]
def current_player_with_cards(g: Game, cards: List[Any]) -> Player:
p = current_player(g)
p.draw("")
p.hand = cards
return p