byte[] rawKey = keyManager.getRawKey(sessionId);
        assertNotNull(rawKey, "Should retrieve raw key");
        assertArrayEquals(testKey, rawKey, "Raw key should match");

        // Verify we get a copy, not the original
        rawKey[0] = (byte) ~rawKey[0];
        byte[] rawKey2 = keyManager.getRawKey(sessionId);
        assertArrayEquals(testKey, rawKey2, "Should still match original");
    }

    @Test

            verify(mockReferralData, atLeastOnce()).getPath();
            verify(mockReferralData, atLeastOnce()).next();
            verify(mockReferralData, atLeastOnce()).getKey();
            verify(mockReferralData, atLeastOnce()).isResolveHashes();
            verify(mockReferralData, atLeastOnce()).isIntermediate();
            verify(mockReferralData, atLeastOnce()).combine(any());

        MsrpcLsarOpenPolicy2 msrpcLsarOpenPolicy2 = new MsrpcLsarOpenPolicy2(server, access, mockPolicyHandle);

        // Then
        // Verify that the super constructor is called with correct arguments
        // Note: Directly verifying super constructor calls is not straightforward in JUnit/Mockito.
        // We verify the effects of the super constructor by checking the fields of the created object.

        // Assert object_attributes fields

        // Unregister
        client.unregister(registration).get(5, TimeUnit.SECONDS);

        assertEquals(0, client.getActiveRegistrationCount());

        verify(mockRpc).register(any(WitnessRegisterRequest.class));
        verify(mockRpc).unregister(any(WitnessUnregisterRequest.class));

        client.close();
    }

    @Test
    void testGetWitnessServer() throws Exception {

            handler.handle(WinError.WINERR_CODES[i], WinError.WINERR_MESSAGES[i]);
        }

        // Verify the collaborator is invoked exactly N times
        verify(handler, times(WinError.WINERR_CODES.length)).handle(anyInt(), anyString());

        // Capture one known interaction to verify argument integrity
        ArgumentCaptor<Integer> codeCap = ArgumentCaptor.forClass(Integer.class);

        handle.release();

        // Assert: verify they happened in order
        InOrder inOrder = inOrder(handle);
        inOrder.verify(handle).ensureDFSResolved();
        inOrder.verify(handle).hasCapability(7);
        inOrder.verify(handle).release();
        inOrder.verifyNoMoreInteractions();
    }

        assertFalse(negotiationResponse.haveCapabilitiy(cap2));
        assertTrue(negotiationResponse.haveCapabilitiy(cap3));

        verify(negotiationResponse).haveCapabilitiy(cap1);
        verify(negotiationResponse).haveCapabilitiy(cap2);
        verify(negotiationResponse).haveCapabilitiy(cap3);
    }

    @Test
    @DisplayName("Test getSendBufferSize returns standard size")

        CIFSContext wrappedContext = context.withCredentials(newCreds);

        assertNotNull(wrappedContext);
        assertTrue(wrappedContext instanceof CIFSContextCredentialWrapper);
        // Verify that the new context uses the provided credentials
        assertEquals(newCreds, wrappedContext.getCredentials());
    }

    @Test
    void testWithAnonymousCredentials() {

        byte[] dest = new byte[5];

        int encodedLen = encodable.encode(dest, 0);

        // Verify encoded length
        assertEquals(3, encodedLen, "Encoded length should be equal to the specified length");

        // Verify content
        assertArrayEquals(new byte[] { 0x02, 0x03, 0x04, 0x00, 0x00 }, dest, "Encoded bytes should match the expected subset");
    }

    @Test

    class FilterFlagsTests {

        @Test
        @DisplayName("Verify FILE_NOTIFY_CHANGE_FILE_NAME constant value")
        void testFileNotifyChangeFileName() {
            assertEquals(0x00000001, FileNotifyInformation.FILE_NOTIFY_CHANGE_FILE_NAME);
        }

        @Test
        @DisplayName("Verify FILE_NOTIFY_CHANGE_DIR_NAME constant value")
        void testFileNotifyChangeDirName() {