def bench_test(self): User.objects.create_user("dillen", "*****@*****.**", "letmein") team1 = Team(team_name="Winners",owner=User.objects.get(username__exact="dillen")) p1 = Player(full_name="abc1",position="WR",team=team1,benched=False) p2 = Player(full_name="abc2",position="WR",team=team1,benched=False) team1.wr_count = 2 p3 = Player(full_name="abc3",position="WR",team=team1) p4 = Player(full_name="abc4",position="WR") team1.save() p1.save() p2.save() p3.save() p4.save() #Bench someone already benched self.assertEqual(bench(p3),False) self.assertEqual(team1.wr_count,2) self.assertEqual(p3.benched,True) #Bench someone not on a team self.assertEqual(bench(p4),False) self.assertEqual(team1.wr_count,2) self.assertEqual(p4.benched,True) #Bench a WR self.assertEqual(bench(p2),False) self.assertEqual(team1.wr_count,1) self.assertEqual(p1.benched,False) self.assertEqual(p2.benched,True)
def test_play(self): User.objects.create_user("dillen", "*****@*****.**", "letmein") User.objects.create_user("eli", "*****@*****.**", "letmein") team1 = Team(team_name="Winners",owner=User.objects.get(username__exact="dillen")) p1 = Player(full_name="abc1",position="WR",team=team1,benched=False) team1.wr_count = 1 p2 = Player(full_name="abc2",position="WR",team=team1) p3 = Player(full_name="abc3",position="WR") p4 = Player(full_name="abc4",position="QB",team=team1,benched=False) team1.qb_count = 1 p5 = Player(full_name="abc5",position="QB",team=team1) team1.save() p1.save() p2.save() p3.save() p4.save() p5.save() #Wrong position self.assertEqual(play(p2,"QB"),False) self.assertEqual(p2.benched,True) self.assertEqual(team1.qb_count,1) #Already playing self.assertEqual(play(p1,"WR"),False) self.assertEqual(p1.benched,False) self.assertEqual(team1.wr_count,1) #Not on a team self.assertEqual(play(p3,"WR"),False) self.assertEqual(p3.team, None) self.assertEqual(p3.benched, True) self.assertEqual(team1.wr_count,1) #2 QB's at the same time self.assertEqual(play(p5,"QB"),False) self.assertEqual(p4.benched, False) self.assertEqual(p5.benched, True) self.assertEqual(team1.qb_count,1) #2 WR's at the same time (valid) self.assertEqual(play(p2,"WR"),True) self.assertEqual(p2.benched, False) self.assertEqual(team1.wr_count,2)