def test_set_pgpool_connected(self):
playertracker1 = PlayerTracker()
self.assertFalse(playertracker1.is_pgpool_connected(123))
self.assertFalse(playertracker1.is_pgpool_disconnected(123))
playertracker1.set_pgpool_disconnected(123)
self.assertFalse(playertracker1.is_pgpool_connected(123))
self.assertTrue(playertracker1.is_pgpool_disconnected(123))
def test_ACK_COMMAND_DISCONNECT_PGPOOL(self):
reply_to_conductor.register_player(112, self.__connection, conductor.exchange, Constants.CONDUCTOR_ROUTING_KEY)
reply_to_conductor.register_player(115, self.__connection, conductor.exchange, Constants.CONDUCTOR_ROUTING_KEY)
reply_to_conductor.acknowledgement_pgpool_disconnected(112, self.__connection, conductor.exchange, Constants.CONDUCTOR_ROUTING_KEY)
time.sleep(2)
player_tracker = PlayerTracker()
self.assertTrue(player_tracker.is_pgpool_disconnected(112))
self.assertFalse(player_tracker.is_pgpool_disconnected(115))
def test_ACK_COMMAND_DISCONNECT_PGPOOL(self):
reply_to_conductor.register_player(112, self.__connection,
conductor.exchange,
Constants.CONDUCTOR_ROUTING_KEY)
reply_to_conductor.register_player(115, self.__connection,
conductor.exchange,
Constants.CONDUCTOR_ROUTING_KEY)
reply_to_conductor.acknowledgement_pgpool_disconnected(
112, self.__connection, conductor.exchange,
Constants.CONDUCTOR_ROUTING_KEY)
time.sleep(2)
player_tracker = PlayerTracker()
self.assertTrue(player_tracker.is_pgpool_disconnected(112))
self.assertFalse(player_tracker.is_pgpool_disconnected(115))
def test_set_master_disconnected(self):
playertracker1 = PlayerTracker()
playertracker1.set_replication_stopped(123)
self.assertFalse(playertracker1.is_replication_started(123))
self.assertTrue(playertracker1.is_replication_stopped(123))
playertracker1.set_replication_started(123)
self.assertTrue(playertracker1.is_replication_started(123))
self.assertFalse(playertracker1.is_pgpool_disconnected(123))
def test__set_player_status(self):
playertracker1 = PlayerTracker()
playertracker1._set_player_status(123, Constants.PLAYER_PGPOOL_CONNECTION_STATUS, Constants.PLAYER_CONNECTION_STATUS_DISCONNECTED)
self.assertTrue(playertracker1.is_pgpool_disconnected(123))
self.assertRaises(PlayerNotRegisteredException, playertracker1._set_player_status, 1, Constants.PLAYER_PGPOOL_CONNECTION_STATUS, Constants.PLAYER_CONNECTION_STATUS_DISCONNECTED)
