}

    @Test
    @DisplayName("Should get share")
    void testGetShare() {
        // Given
        String share = "testshare";
        when(mockReferralData.getShare()).thenReturn(share);

        // When
        String result = mockReferralData.getShare();

        // Then
        assertEquals(share, result);
        verify(mockReferralData).getShare();
    }

    @Test

        CompletableFuture<WitnessRegistration> future = client.registerForNotifications("\\\\server\\share", serverAddress, listener);

        WitnessRegistration registration = future.get(5, TimeUnit.SECONDS);

        assertNotNull(registration);
        assertEquals(WitnessRegistrationState.REGISTERED, registration.getState());
        assertEquals("\\\\server\\share", registration.getShareName());

    class HappyPath {
        @Test
        void testBasicProperties() throws Exception {
            String url = "smb://server/share/path/file.txt";
            DummySmbResourceLocator loc = new DummySmbResourceLocator(url);
            assertEquals("file.txt", loc.getName());
            assertEquals("smb://server/share/path/", loc.getParent());
            assertEquals(url, loc.getPath());
            // canonicalisation simply removes '.' and '..'

    }

    @Test
    void testGetReferral_Success() throws IOException, SmbAuthException {
        String domain = "domain.com";
        String root = "share";
        String path = "\\folder";
        String fullPath = "\\" + domain + "\\" + root + path; // Should be "\domain.com\share\folder"

        lenient().when(smbTransport.getDfsReferrals(auth, fullPath, 0)).thenReturn(dfsReferral);

        t1.url = "smb://server1/share/file.txt";

        t2.numThreads = 5;
        t2.url = "smb://server2/share/file.txt";

        // Verify independence
        assertEquals(2, t1.numThreads, "t1 should have its own numThreads value");
        assertEquals(5, t2.numThreads, "t2 should have its own numThreads value");
        assertEquals("smb://server1/share/file.txt", t1.url, "t1 should have its own URL");

                            SmbTreeImpl uct = smbSession.getSmbTree(share, null).unwrap(SmbTreeImpl.class);
                            SmbTreeImpl ct = connectTree(loc, host, share, trans, uct, dr)) {

                        if (dr != null) {
                            ct.setTreeReferral(dr);
                            if (dr != start) {

        serverAddress = InetAddress.getByName("192.168.1.100");
        registration = new WitnessRegistration("\\\\server\\share", serverAddress, WitnessServiceType.FILE_SERVER_WITNESS);
    }

    @Test
    void testRegistrationCreation() {
        assertNotNull(registration.getRegistrationId());
        assertEquals("\\\\server\\share", registration.getShareName());
        assertEquals(serverAddress, registration.getServerAddress());

     * @throws CIFSException if an error occurs retrieving the domain name
     */
    String getOEMDomainName() throws CIFSException;

    /**
     * Gets the name of the share this tree is connected to
     * @return the share we are connected to
     */
    String getConnectedShare();

    /**
     * Checks if this tree handle refers to the same tree as another
     * @param th the tree handle to compare with

    synchronized void insert(final String path, final DfsReferral dr) {
        int s1, s2;
        String server, share, key;

        if (DISABLED) {
            return;
        }

        s1 = path.indexOf('\\', 1);
        s2 = path.indexOf('\\', s1 + 1);
        server = path.substring(1, s1);
        share = path.substring(s1 + 1, s2);

        key = path.substring(0, dr.pathConsumed).toLowerCase();

            throw new SmbException("Invalid server name in UNC path");
        }

        // Validate share name
        String share = parts[1];
        if (share.isEmpty() || share.endsWith("$") && share.length() == 1) {
            throw new SmbException("Invalid share name in UNC path");
        }
    }

    /**
     * Validate individual path components
     */