def test_dealer_button(self): """ Give dealer button to first player or next of previous gamer Dealer is always awarded to the first player if the table.dealer field is empty or to the next player inthe table.dealer field """ # Arrange us1 = User(username='******') us1.save() us2 = User(username='******') us2.save() us3 = User(username='******') us3.save() us4 = User(username='******') us4.save() player1 = Player(name=us1, state='start') player1.save() player2 = Player(name=us2, state='start') player2.save() player3 = Player(name=us3, state='start') player3.save() player4 = Player(name=us4, state='start') player4.save() table = Table(player1=player1, player2=player2, player3=player3, player4=player4, game_state='start') # Act & Assertion table.dealer_button() self.assertEqual(table.dealer, player1) table.game_state = 'start' table.dealer_button() self.assertEqual(table.dealer, player2) table.game_state = 'start' table.dealer_button() self.assertEqual(table.dealer, player3) table.game_state = 'start' table.dealer_button() self.assertEqual(table.dealer, player4) table.game_state = 'start' table.dealer_button() self.assertEqual(table.dealer, player1)
def test_take_big_blind(self): """ Take small blind from the next one from 'dealer player' Big blind is always awarded if players in table > 1 - to the next player from the table.small_blind field """ # Arrange us1 = User(username='******') us1.save() us2 = User(username='******') us2.save() us3 = User(username='******') us3.save() us4 = User(username='******') us4.save() player1 = Player(name=us1, state='start') player1.save() player2 = Player(name=us2, state='start') player2.save() player3 = Player(name=us3, state='start') player3.save() player4 = Player(name=us4, state='start') player4.save() table = Table(player1=player1, player2=player2, player3=player3, player4=player4, game_state='small_blind', dealer=player1, small_blind=player2) # Act & Assertion table.take_big_blind() self.assertEqual(table.big_blind, player3) table.game_state = 'small_blind' table.dealer = player2 table.small_blind = player3 table.take_big_blind() self.assertEqual(table.big_blind, player4)