lenient().when(parent.getLocator()).thenReturn(locator);
        lenient().when(locator.getUNCPath()).thenReturn("\\\\SERVER\\Share\\dir\\"); // ends with \\

        // Create URL with registered handler
        URL smbUrl = new URL("smb://server/share/dir/");
        lenient().when(locator.getURL()).thenReturn(smbUrl);
    }

    // Helper: reflectively set private/protected field on an object

    /**
     * Get the domain this referral is for
     *
     * @return the domain this referral is for
     */
    String getDomain();

    /**
     * Get the share this referral points to
     *
     * @return the share this referral points to
     */
    String getShare();

    /**
     * Get the number of characters from the UNC path that were consumed by this referral
     *

    @Test
    @DisplayName("handleDFSReferral handles null referral and empty path")
    void handleDfsReferralNullAndEmpty() {
        when(locator.handleDFSReferral(null, "")).thenReturn("smb://server/share/");
        assertEquals("smb://server/share/", locator.handleDFSReferral(null, ""));
        verify(locator).handleDFSReferral(null, "");
        verifyNoMoreInteractions(locator);
    }

        /** Number of delayed unassigned shards */
        public int delayedUnassignedShards;
        /** Number of unassigned shards */
        public int unassignedShards;
        /** Number of initializing shards */
        public int initializingShards;
        /** Number of relocating shards */
        public int relocatingShards;
        /** Percentage of active shards */
        public double activeShardsPercent;

                    long threadStart = System.nanoTime();
                    for (int i = 0; i < operationsPerThread; i++) {
                        try {
                            String shareName = "share" + (threadId % 5); // Use 5 different shares

                            // Test concurrent tree operations (should be thread-safe with CopyOnWriteArrayList)
                            long opStart = System.nanoTime();

        connectionState = 0;
    }

    synchronized SmbTree getSmbTree(String share, final String service) {
        SmbTree t;

        if (share == null) {
            share = "IPC$";
        }
        for (final Enumeration e = trees.elements(); e.hasMoreElements();) {
            t = (SmbTree) e.nextElement();
            if (t.matches(share, service)) {
                return t;
            }
        }

            }
            return resp.getAvailable();
        } catch (final SmbException se) {
            throw seToIoe(se);
        }
    }

    @Override
    public void close() {
        // ignore, the shared file descriptor is closed by the pipe handle
    }

        return this.sessionKey;
    }

    @Override
    public SmbTreeImpl getSmbTree(String share, String service) {
        if (share == null) {
            share = "IPC$";
        }

        synchronized (this.trees) {
            for (SmbTreeImpl t : this.trees) {
                if (t.matches(share, service)) {
                    return t.acquire();
                }
            }

     * @return the access mask hint
     */
    public int getAccessMaskHint() {
        return accessMaskHint;
    }

    /**
     * Gets the share access hint for optimizing lease handling
     * @return the share access hint
     */
    public int getShareAccessHint() {
        return shareAccessHint;
    }

    @Override

    /**
     * Test for getConnectedShare() method.
     * Verifies that the method returns the correct connected share name.
     */
    @Test
    void testGetConnectedShare() {
        when(smbTreeHandle.getConnectedShare()).thenReturn("TEST_SHARE");
        assertEquals("TEST_SHARE", smbTreeHandle.getConnectedShare(), "Connected share should be TEST_SHARE");
    }

    /**
     * Test for isSameTree() method.