new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };

        Smb2LeaseKey key = new Smb2LeaseKey(originalBytes);
        byte[] retrievedKey = key.getKey();

        // Modify the retrieved key
        retrievedKey[0] = (byte) 0xFF;

        // Original key should remain unchanged
        assertEquals(0x01, key.getKey()[0]);
    }

    private Smb2LeaseKey testKey;
    private int testState;

    @BeforeEach
    void setUp() {
        byte[] keyBytes = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };
        testKey = new Smb2LeaseKey(keyBytes);
        testState = Smb2LeaseState.SMB2_LEASE_READ_WRITE;
        leaseContext = new LeaseV1CreateContextRequest(testKey, testState);

        void testDcerpcFirstFrag() {
            assertEquals(0x01, DcerpcConstants.DCERPC_FIRST_FRAG, "DCERPC_FIRST_FRAG should be 0x01");
        }

        @Test
        @DisplayName("DCERPC_LAST_FRAG should have correct value")
        void testDcerpcLastFrag() {
            assertEquals(0x02, DcerpcConstants.DCERPC_LAST_FRAG, "DCERPC_LAST_FRAG should be 0x02");
        }

        @Test

            context = new PreauthIntegrityNegotiateContext();
            byte[] buffer = new byte[] { 0x02, 0x00, // 2 hash algos
                    0x04, 0x00, // 4 bytes salt
                    0x01, 0x00, // SHA512
                    0x02, 0x00, // Another hash algo
                    0x0A, 0x0B, 0x0C, 0x0D // Salt
            };

            // Act

     */
    int RPC_PT_REQUEST = 0x00;
    /**
     * Ping packet type
     */
    int RPC_PT_PING = 0x01;
    /**
     * Response packet type
     */
    int RPC_PT_RESPONSE = 0x02;
    /**
     * Fault packet type - indicates an error
     */
    int RPC_PT_FAULT = 0x03;
    /**
     * Bind packet type - establishes context
     */
    int RPC_PT_BIND = 0x0B;
    /**

    private byte[] testMacSigningKey;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        testMacSigningKey = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };
    }

    @Test
    @DisplayName("Test constructor with MAC signing key and bypass flag")
    void testConstructorWithBypass() {

    class ToStringTests {

        @ParameterizedTest
        @DisplayName("Should format command name correctly")
        @ValueSource(shorts = { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10, 0x11,
                0x12 })
        void testToStringCommandNames(short command) {
            testMessage.setCommand(command);

        smb.tid = 789;
        smb.uid = 101;
        smb.flags2 = ServerMessageBlock.FLAGS2_UNICODE;
        smb.useUnicode = true;

        byte[] params = { 0x01, 0x02, 0x03, 0x04 };
        byte[] bytes = { 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F };
        smb.setParamWords(params);
        smb.setBytes(bytes);

        byte[] buffer = new byte[1024];
        int length = smb.encode(buffer, 0);

        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        PrintStream ps = new PrintStream(baos);

        // Test with control characters and printable characters
        byte[] data = { 0x00, 0x01, 0x09, 0x0A, 0x0D, 0x1F, // Control characters
                0x20, 0x41, 0x42, 0x7E, // Printable characters
                (byte) 0x7F, (byte) 0x80, (byte) 0xFF // Extended/control
        };

        void testSizeWithMultipleHashAlgosAndSalt() {
            int[] hashAlgos = { 0x01, 0x02, 0x03 };
            byte[] salt = { 0x01, 0x02, 0x03, 0x04, 0x05 };
            PreauthIntegrityNegotiateContext context = new PreauthIntegrityNegotiateContext(mockConfig, hashAlgos, salt);

            assertEquals(15, context.size());
        }

        @Test