def test_RotateRoles_1(self):
e = UDPGameEngine()
s = GameState()
s.player_size = 1
s.roles = [GameState.ROLE_LEFT_PADDLE]
s.players = [-1, 0]
e.RotateRoles(s)
self.assertTrue(s.roles == [GameState.ROLE_LEFT_PADDLE])
self.assertTrue(s.players == [-1, 0])
def test_RotateRoles_2(self):
e = UDPGameEngine()
s = GameState()
s.player_size = 2
s.roles = [GameState.ROLE_LEFT_PADDLE, GameState.ROLE_RIGHT_PADDLE]
s.players = [-1]*(s.player_size + 1)
s.players[GameState.ROLE_LEFT_PADDLE] = 0
s.players[GameState.ROLE_RIGHT_PADDLE] = 1
e.RotateRoles(s)
self.assertTrue(s.roles == [GameState.ROLE_RIGHT_PADDLE,
GameState.ROLE_LEFT_PADDLE])
self.assertTrue(s.players == [-1, 1, 0])
def test_RotateRoles_3(self):
e = UDPGameEngine()
s = GameState()
s.roles = [GameState.ROLE_RIGHT_PADDLE, GameState.ROLE_LEFT_PADDLE,
GameState.ROLE_BALL]
s.player_size = len(s.roles)
s.players = [-1]*(len(s.roles)+ 1)
s.players[GameState.ROLE_LEFT_PADDLE] = 1
s.players[GameState.ROLE_RIGHT_PADDLE] = 0
s.players[GameState.ROLE_BALL] = 2
e.RotateRoles(s)
self.assertTrue(s.roles == [GameState.ROLE_LEFT_PADDLE,
GameState.ROLE_BALL, GameState.ROLE_RIGHT_PADDLE])
players = [-1]*(len(s.roles)+1)
players[GameState.ROLE_LEFT_PADDLE] = 0
players[GameState.ROLE_RIGHT_PADDLE] = 2
players[GameState.ROLE_BALL] = 1
self.assertTrue(s.players == players)
