public AbilityStatus getConnectionAbility(AbilityKey abilityKey) {
        if (currentConnection != null) {
            return currentConnection.getConnectionAbility(abilityKey);
        }
        // return null if connection is not ready
        return null;
    }
@Test
    void testGetConnectionAbilityWithReadyConnection() {
        when(connection.getConnectionAbility(AbilityKey.SERVER_TEST_1)).thenReturn(AbilityStatus.SUPPORTED);
        rpcClient.currentConnection = connection;
        AbilityStatus abilityStatus = rpcClient.getConnectionAbility(AbilityKey.SERVER_TEST_1);
        assertNotNull(abilityStatus);
        assertEquals(AbilityStatus.SUPPORTED, abilityStatus);
    }