def test_check_players_have_cards(self): """Return False if at least one player have cards""" # Arrange cards1 = str(Card.new('As')) + ',' + str(Card.new('9s')) cards2 = str(Card.new('Ah')) + ',' + str(Card.new('9h')) us1 = User(username='******') us1.save() us2 = User(username='******') us2.save() us3 = User(username='******') us3.save() us4 = User(username='******') us4.save() player1 = Player(name=us1, cards=cards1) player1.save() player2 = Player(name=us2, cards=cards2) player2.save() player3 = Player(name=us3) player3.save() player4 = Player(name=us4) player4.save() table = Table(player1=player1, player2=player2, player3=player3, player4=player4) # Act result = table.check_players_have_cards() # Assertion self.assertFalse(result) table.remove_cards_from_players() result2 = table.check_players_have_cards() self.assertTrue(result2)
def test_remove_cards_from_players(self): """Remove all cards from all players""" # Arrange cards1 = str(Card.new('As')) + ',' + str(Card.new('9s')) cards2 = str(Card.new('Ah')) + ',' + str(Card.new('9h')) cards3 = str(Card.new('Ad')) + ',' + str(Card.new('9d')) cards4 = str(Card.new('Ac')) + ',' + str(Card.new('9c')) us1 = User(username='******') us1.save() us2 = User(username='******') us2.save() us3 = User(username='******') us3.save() us4 = User(username='******') us4.save() player1 = Player(name=us1, cards=cards1) player1.save() player2 = Player(name=us2, cards=cards2) player2.save() player3 = Player(name=us3, cards=cards3) player3.save() player4 = Player(name=us4, cards=cards4) player4.save() table = Table(player1=player1, player2=player2, player3=player3, player4=player4) # Act table.remove_cards_from_players() # Assertion self.assertEqual(table.player1.cards, None) self.assertEqual(table.player2.cards, None) self.assertEqual(table.player3.cards, None) self.assertEqual(table.player4.cards, None)