negotiatedField.setAccessible(true);
        negotiatedField.set(initial, negotiationResponse);

        // When: Request connection with signing enforced
        SmbTransportImpl withSigning = pool.getSmbTransport(ctx, address, 445, false, true);

        // Then: Should create new connection
        assertNotSame(initial, withSigning, "Should create new connection with different signing");
    }

    @Test

    void testDfsReferralData() throws Exception {
        String url = "smb://server/share/path/";
        DummySmbResourceLocator loc = new DummySmbResourceLocator(url);

        // Initially no DFS referral
        assertNull(loc.getDfsReferral());

        // Set and retrieve DFS referral
        loc.setDfsReferral(dfs);
        assertSame(dfs, loc.getDfsReferral());
    }

        // Given
        Smb2EncryptionContext context = new Smb2EncryptionContext(1, DialectVersion.SMB311, testEncryptionKey, testDecryptionKey);

        // When - initially should not need rotation
        assertFalse(context.needsKeyRotation(), "Should not need rotation initially");

        // Note: We can't directly test the byte limit without encrypting large amounts of data,

        @DisplayName("Individual flag values are correctly set and retrieved")
        void testIndividualFlagValues(int flag) {
            NegTokenInit init = new NegTokenInit();

            // Initially false
            assertFalse(init.getContextFlag(flag));

            // Set to true
            init.setContextFlag(flag, true);
            assertTrue(init.getContextFlag(flag));

    void isOpen_isStale_transitions() throws Exception {
        SmbPipeHandleImpl handle = newHandleWithBasicStubs(0, "\\\\pipe\\\\w");
        assertFalse(handle.isOpen(), "Initially not open");
        assertFalse(handle.isStale(), "Initially not stale");

        // After ensureOpen with valid fh
        when(pipe.ensureTreeConnected()).thenReturn(tree);
        when(tree.acquire()).thenReturn(tree);

        when(fileHandle.acquire()).thenReturn(fileHandle);
        when(fileHandle.isValid()).thenReturn(true);
        when(fileHandle.getTree()).thenReturn(tree);

        // Initially not open (isStale returns false when handle is null, not true)
        assertFalse(target.isOpen());
        assertFalse(target.isStale());

        // After ensureOpen, handle valid -> open

    }

    @Test
    @DisplayName("Test setFileInfoClass sets correct info type")
    void testSetFileInfoClassSetsCorrectInfoType() {
        request = new Smb2QueryInfoRequest(mockConfig);

        // Initially set a different info type
        request.setInfoType((byte) 0x05);

        // Call setFileInfoClass which should override info type
        request.setFileInfoClass((byte) 0x10);

        try {

        String sessionId = "rotation-test";
        keyManager.storeSessionKey(sessionId, testKey, "AES");

        // Initial version should be 0
        assertEquals(0, keyManager.getKeyVersion(sessionId), "Initial version should be 0");

        // Rotate key
        int newVersion = keyManager.rotateSessionKey(sessionId);
        assertEquals(1, newVersion, "New version should be 1");

        assertEquals(mockCredentials, wrapper.getCredentials(), "Initial credentials should be set by constructor");
    }

    @Test
    @DisplayName("Should return the credentials provided in the constructor")
    void testGetCredentials() {
        assertEquals(mockCredentials, wrapper.getCredentials(), "getCredentials should return the initially set credentials");
    }

    @Test

        suggester.createIndexIfNothing();
        Thread.sleep(100);

        // Verify initial state - should have one index
        GetAliasesResponse aliasResponse = client.admin().indices().prepareGetAliases(suggester.getIndex()).execute().actionGet();
        assertTrue("Should have at least one index initially", aliasResponse.getAliases().size() >= 1);

        // Create next index - this is the main method we're testing