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)