// Act
        handle.close();

        // Assert
        // Verify sendrecv was called once for open and once for close
        verify(mockDcerpcHandle, times(1)).sendrecv(any(MsrpcLsarOpenPolicy2.class));
        verify(mockDcerpcHandle, times(1)).sendrecv(any(MsrpcLsarClose.class));
    }

    @Test

            byte[] result = hmac.engineDigest();

            assertArrayEquals(finalDigest, result);
            verify(mockMd5, times(1)).digest(); // First call for inner digest
            verify(mockMd5, times(2)).update(any(byte[].class)); // Once in constructor with ipad, once in engineDigest with opad
            verify(mockMd5, times(1)).digest(innerDigest); // Second call for final digest
        }
    }

    @Test

        // Act
        handle.close();

        // Assert
        // Verify sendrecv was called for open and then for close
        verify(mockDcerpcHandle, times(1)).sendrecv(any(MsrpcSamrOpenDomain.class));
        verify(mockDcerpcHandle, times(1)).sendrecv(closeHandleCaptor.capture());
        MsrpcSamrCloseHandle capturedRpc = closeHandleCaptor.getValue();

            NdrShort ndrShort2 = new NdrShort(0);
            ndrShort2.decode(decodeBuffer);

            // Then: Values should match
            verify(encodeBuffer).enc_ndr_short(originalValue);
            verify(decodeBuffer).dec_ndr_short();
            assertEquals(ndrShort1.value, ndrShort2.value);
        }

        @Test

        assertTrue(ex.getMessage().contains("failure"));
        verify(session).getSessionKey();
    }

    // Happy path: transport is returned and verified
    @Test
    @DisplayName("getTransport returns mocked transport")
    void getTransport_happy() {
        when(session.getTransport()).thenReturn(transport);
        assertSame(transport, session.getTransport());
        verify(session).getTransport();
    }

        verify(mockDeferredNdrBuffer).enc_ndr_long(0);
        verify(mockDeferredNdrBuffer).enc_ndr_long(trustInfo.name.length / 2);
        verify(mockDeferredNdrBuffer).advance(2 * (trustInfo.name.length / 2));
        // Verify the derive call
        verify(mockDeferredNdrBuffer).derive(anyInt());
        verify(mockDeferredNdrBuffer, times(5)).enc_ndr_short(0); // Writing 5 shorts with value 0

            // Verify that user information is stored in the session
            verify(session).setAttribute("NtlmHttpAuth", ntlmAuth);
            verify(session).setAttribute("ntlmuser", "user");
            verify(session).setAttribute("ntlmdomain", "TEST_DOMAIN");

            // Verify that the chain continues
            verify(response, never()).setStatus(HttpServletResponse.SC_UNAUTHORIZED);

        // Act
        aliasHandle.close();

        // Assert
        // Verify that sendrecv was called with MsrpcSamrCloseHandle
        verify(mockDcerpcHandle).sendrecv(any(MsrpcSamrCloseHandle.class));
        // Verify that close was called only once on the mockDcerpcHandle for MsrpcSamrCloseHandle
        verify(mockDcerpcHandle, times(1)).sendrecv(any(MsrpcSamrCloseHandle.class));
    }

    @Test

        assertTrue(mockResponse.isReceived());

        // Verify the method was called
        verify(mockResponse, times(2)).isReceived();
    }

    @Test
    void testReceived() {
        // Call the method
        mockResponse.received();

        // Verify that the method was called
        verify(mockResponse, times(1)).received();
    }

    @Test
    void testClearReceived() {

        // Then - verify the body was written correctly
        int bodyOffset = headerStart + Smb2Constants.SMB2_HEADER_LENGTH;

        // Verify structure size (25)
        assertEquals(25, SMBUtil.readInt2(buffer, bodyOffset));

        // Verify session binding flag (0x1)
        assertEquals(0x1, buffer[bodyOffset + 2]);

        // Verify security mode