assertEquals(channel2, selected); // Should select less loaded channel
    }

    @Test
    void testWeightedRandomStrategy() {
        loadBalancer.setStrategy(LoadBalancingStrategy.WEIGHTED_RANDOM);
        when(mockChannelManager.getHealthyChannels()).thenReturn(Arrays.asList(channel1, channel2));

        // Test multiple selections to ensure it works
        for (int i = 0; i < 10; i++) {

    ROUND_ROBIN,

    /**
     * Select the least busy channel based on pending operations
     */
    LEAST_LOADED,

    /**
     * Weighted random selection based on channel scores
     */
    WEIGHTED_RANDOM,

    /**
     * Affinity-based selection for related operations
     */
    AFFINITY_BASED,

    /**
     * Adaptive selection based on operation type and performance
     */
    ADAPTIVE

    }

    @Test
    void testLoadBalancingStrategyProperty() throws CIFSException {
        Properties props = new Properties();

        String[] strategies = { "round_robin", "least_loaded", "weighted_random", "affinity_based", "adaptive" };

        for (String strategy : strategies) {
            props.setProperty("jcifs.smb.client.loadBalancingStrategy", strategy);

        switch (strategy) {
        case ROUND_ROBIN:
            return selectRoundRobin(availableChannels);

        case LEAST_LOADED:
            return selectLeastLoaded(availableChannels);

        case WEIGHTED_RANDOM:
            return selectWeightedRandom(availableChannels);

        case AFFINITY_BASED:
            return selectWithAffinity(message, availableChannels);

        case ADAPTIVE:
        default:

        // Test setting different strategies
        LoadBalancingStrategy[] strategies = { LoadBalancingStrategy.ROUND_ROBIN, LoadBalancingStrategy.LEAST_LOADED,
                LoadBalancingStrategy.WEIGHTED_RANDOM, LoadBalancingStrategy.AFFINITY_BASED, LoadBalancingStrategy.ADAPTIVE };

        for (LoadBalancingStrategy strategy : strategies) {
            loadBalancer.setStrategy(strategy);