T chainedResponse = null;
            if (!(request instanceof SmbComTreeDisconnect) && !(request instanceof Smb2TreeDisconnectRequest)) {
                chainedResponse = treeConnect(request, response);
            }
            if (request == null || chainedResponse != null && chainedResponse.isReceived()) {
                return chainedResponse;
            }

            }
            return null;
        }).when(session).send(any(ServerMessageBlock.class), any(ServerMessageBlock.class));

        // Test tree connect
        tree.treeConnect(null, null);

        assertEquals(2, tree.connectionState);
        assertEquals(456, tree.tid);
        assertEquals("testService", tree.service);
        assertTrue(tree.inDfs);

                          Skipping DC interrogation.""");
            }
        } else {
            final SmbSession ssn = trans.getSmbSession(NtlmPasswordAuthentication.DEFAULT);
            ssn.getSmbTree(LOGON_SHARE, null).treeConnect(null, null);
        }
        return new NtlmChallenge(trans.server.encryptionKey, dc);
    }

    /**
     * Retrieves an NTLM challenge from a domain controller for the configured domain.
     *

            t.markDomainDfs();
        }

        try {
            if (log.isTraceEnabled()) {
                log.trace("doConnect: " + addr);
            }
            t.treeConnect(null, null);
            return t.acquire();
        } catch (final SmbAuthException sae) {
            log.debug("Authentication failed", sae);
            return retryAuthentication(loc, share, trans, t, referral, sae);

        doReturn(tree).when(session).getSmbTree(eq("LOGON$"), isNull());

        // Act
        session.treeConnectLogon();

        // Assert: treeConnect invoked once with null params
        verify(tree, times(1)).treeConnect(isNull(), isNull());
        // Close of try-with-resources should call close on the mock
        verify(tree, times(1)).close();
    }

    @Test

    }

    @Test
    @DisplayName("Should verify complete protocol compliance")
    void testProtocolCompliance() throws SMBProtocolDecodingException {
        // Given - exact SMB2 TREE_CONNECT response structure
        byte[] wireData = new byte[] { 0x10, 0x00, // StructureSize (must be 16)
                0x01, // ShareType (DISK)
                0x00, // Reserved