byte[] shortFileId = new byte[8];
        byte[] standardFileId = new byte[16];
        byte[] longFileId = new byte[32];

        new SecureRandom().nextBytes(shortFileId);
        new SecureRandom().nextBytes(standardFileId);
        new SecureRandom().nextBytes(longFileId);

        RequestWithFileId request = new Smb2CloseRequest(mockConfig, emptyFileId);

        // All sizes should be accepted without exception

        int length = 16;

        // When
        byte[] random1 = new byte[length];
        byte[] random2 = new byte[length];
        SecureRandom.getInstanceStrong().nextBytes(random1);
        SecureRandom.getInstanceStrong().nextBytes(random2);

        // Then
        assertNotNull(random1);
        assertNotNull(random2);
        assertEquals(length, random1.length);

        random.nextBytes(value);
        for (PrimitiveSink sink : sinks) {
          sink.putBytes(value);
        }
      }
    },
    PUT_BYTES_INT_INT() {
      @Override
      void performAction(Random random, Iterable<? extends PrimitiveSink> sinks) {
        byte[] value = new byte[random.nextInt(128)];
        random.nextBytes(value);

        // Initialize test keys
        sessionKey = new byte[16]; // 128-bit session key
        preauthIntegrity = new byte[64]; // Preauth integrity hash
        new SecureRandom().nextBytes(sessionKey);
        new SecureRandom().nextBytes(preauthIntegrity);
    }

    @Test
    @DisplayName("Should derive signing key for SMB 3.0.0 dialect")
    void testDeriveSigningKey_SMB300() {
        // Given

    void setUp() {
        // Initialize test keys
        testEncryptionKey = new byte[16]; // 128-bit key
        testDecryptionKey = new byte[16]; // 128-bit key
        new SecureRandom().nextBytes(testEncryptionKey);
        new SecureRandom().nextBytes(testDecryptionKey);

        // Create encryption context with required parameters
        encryptionContext = new Smb2EncryptionContext(1, DialectVersion.SMB311, testEncryptionKey, testDecryptionKey);

        String sessionId1 = "session-1";
        String sessionId2 = "session-2";

        byte[] key1 = new byte[16];
        byte[] key2 = new byte[16];
        new SecureRandom().nextBytes(key1);
        new SecureRandom().nextBytes(key2);

        keyManager.storeSessionKey(sessionId1, key1, "AES");
        keyManager.storeSessionKey(sessionId2, key2, "AES");

        byte[] retrieved1 = keyManager.getRawKey(sessionId1);

            byte[] data = new byte[1024];
            new Random().nextBytes(data);

            assertDoesNotThrow(() -> request.setData(data, 0, data.length));
        }

        @Test
        @DisplayName("Should set data with partial buffer")
        void testSetDataPartialBuffer() {
            byte[] data = new byte[1024];
            new Random().nextBytes(data);

        random.nextBytes(value);
        for (PrimitiveSink sink : sinks) {
          sink.putBytes(value);
        }
      }
    },
    PUT_BYTES_INT_INT() {
      @Override
      void performAction(Random random, Iterable<? extends PrimitiveSink> sinks) {
        byte[] value = new byte[random.nextInt(128)];
        random.nextBytes(value);

        assertTrue(preauthService.validatePreauthIntegrity(sessionId, expectedHash));

        // Validation should fail with incorrect hash
        byte[] incorrectHash = new byte[expectedHash.length];
        secureRandom.nextBytes(incorrectHash);

        assertThrows(CIFSException.class, () -> {
            preauthService.validatePreauthIntegrity(sessionId, incorrectHash);
        });
    }

    @Test

        NetbiosAddress mockHost = mock(NetbiosAddress.class);
        SecureRandom mockRandom = mock(SecureRandom.class);

        // Create a machine ID (32 bytes)
        byte[] machineId = new byte[32];
        mockRandom.nextBytes(machineId);

        lenient().when(mockConfig.getDefaultDomain()).thenReturn("TESTDOMAIN");
        lenient().when(mockConfig.isUseUnicode()).thenReturn(true);