def test_send_disconnect_PGPool(self):
with Conductor() as conductor1:
conductor1.send_disconnect_PGPool()
self.assertRaises(Empty, Test._queue.get, timeout=1)
playerTracker = PlayerTracker()
playerTracker.set_accept_player(True)
playerTracker.add_player(222)
playerTracker.set_accept_player(False)
conductor1.send_disconnect_PGPool()
message = Test._queue.get(timeout=5)
self.assertEqual(Constants.ACK_COMMAND_DISCONNECT_PGPOOL, message.payload[Constants.MESSAGE_ACK_COMMAND])
def test_send_reset_players(self):
with Conductor() as conductor1:
conductor1.send_reset_players()
self.assertRaises(Empty, Test._queue.get, timeout=1)
playerTracker = PlayerTracker()
playerTracker.set_accept_player(True)
playerTracker.add_player(222)
playerTracker.set_accept_player(False)
conductor1.send_reset_players()
message = Test._queue.get(timeout=5)
self.assertEqual(Constants.ACK_COMMAND_RESET_PLAYERS, message.payload[Constants.MESSAGE_ACK_COMMAND])
def test_send_start_replication(self):
with Conductor() as conductor1:
conductor1.send_start_replication()
self.assertRaises(Empty, Test._queue.get, timeout=1)
playerTracker = PlayerTracker()
playerTracker.set_accept_player(True)
playerTracker.add_player(222)
playerTracker.set_accept_player(False)
conductor1.send_start_replication()
message = Test._queue.get(timeout=5)
self.assertEqual(Constants.ACK_COMMAND_START_REPLICATION, message.payload[Constants.MESSAGE_ACK_COMMAND])
def test_wait_replication_started(self):
tested = False
with Conductor() as conductor1:
playerTracker = PlayerTracker()
playerTracker.set_accept_player(True)
playerTracker.add_player(222)
playerTracker.set_accept_player(False)
self.assertRaises(ConductorTimeoutException, conductor1.wait_replication_started, timeout=1)
playerTracker.set_replication_started(222)
conductor1.wait_replication_started(timeout=1)
tested = True
self.assertTrue(tested)
def test_add_player(self):
playertracker1 = PlayerTracker()
playertracker2 = PlayerTracker()
players = playertracker2.get_player_ids()
self.assertEqual(1, len(players))
playertracker1.set_accept_player(True)
self.assertRaises(PlayerAlreadyRegisteredException, playertracker1.add_player, 123)
playertracker1.add_player(2)
playertracker1.set_accept_player(False)
players = playertracker2.get_player_ids()
self.assertEqual(2, len(players))
self.assertIn(123, players)
self.assertIn(2, players)
def test_set_accept_player(self):
playertracker1 = PlayerTracker()
playertracker1.clear()
ids = playertracker1.get_player_ids()
self.assertEqual(0, len(ids))
playertracker1.set_accept_player(True)
playertracker1.add_player(1)
playertracker1.add_player(2)
playertracker1.set_accept_player(False)
ids = playertracker1.get_player_ids()
self.assertEqual(2, len(ids))
self.assertRaises(PlayerDelayedRegistrationException, playertracker1.add_player, 3)
ids = playertracker1.get_player_ids()
self.assertEqual(2, len(ids))
def test_get_player_ids(self):
playertracker1 = PlayerTracker()
ids = playertracker1.get_player_ids()
self.assertEqual(1, len(ids))
playertracker1.set_accept_player(True)
playertracker1.add_player(1)
playertracker1.set_accept_player(False)
ids = playertracker1.get_player_ids('A')
self.assertEqual(0, len(ids))
playertracker1.set_player_group(123, 'A')
ids = playertracker1.get_player_ids()
self.assertEqual(2, len(ids))
ids = playertracker1.get_player_ids('A')
self.assertEqual(1, len(ids))
ids = playertracker1.get_player_ids('B')
self.assertEqual(0, len(ids))
def test_grouping_players(self):
tested = False
with Conductor() as conductor1:
playerTracker = PlayerTracker()
playerTracker.set_accept_player(True)
playerTracker.add_player(123)
playerTracker.add_player(222)
playerTracker.add_player(333)
playerTracker.add_player(444)
playerTracker.set_accept_player(False)
conductor1.grouping_players()
groupA = playerTracker.get_player_ids(Constants.PLAYER_GROUP_A)
self.assertEqual(2, len(groupA))
groupB = playerTracker.get_player_ids(Constants.PLAYER_GROUP_B)
self.assertEqual(2, len(groupB))
tested = True
self.assertTrue(tested)
def test_get_player_ids(self):
tested = False
with Conductor() as conductor1:
playerTracker = PlayerTracker()
playerTracker.set_accept_player(True)
playerTracker.add_player(123)
playerTracker.add_player(222)
playerTracker.add_player(333)
playerTracker.add_player(444)
playerTracker.set_accept_player(False)
self.assertRaises(PlayerDelayedRegistrationException, playerTracker.add_player, 123123)
ids = conductor1.get_player_ids()
self.assertEqual(4, len(ids))
self.assertIn(123, ids)
self.assertIn(222, ids)
self.assertIn(333, ids)
self.assertIn(444, ids)
ids = playerTracker.get_player_ids(Constants.PLAYER_GROUP_A)
self.assertEqual(0, len(ids))
tested = True
self.assertTrue(tested)
def setUp(self):
playertracker1 = PlayerTracker()
playertracker1.clear()
playertracker1.set_accept_player(True)
playertracker1.add_player(123)
playertracker1.set_accept_player(False)
