assertEquals(0, buffer[SmbConstants.CMD_OFFSET]);
        }

        @Test
        @DisplayName("Test encode with signature")
        void testEncodeWithSignature() {
            testBlock.setDigest(mockDigest);
            byte[] buffer = new byte[1024];

            int length = testBlock.encode(buffer, 0);

            assertTrue(length > 0);

        assertTrue(length > 0);
        assertEquals(1, andxBlock.batchLevel); // batchLevel should be incremented
    }

    @Test
    @DisplayName("Test encode with signature")
    void testEncodeWithSignature() {
        testBlock = new TestAndXServerMessageBlock(mockConfig, (byte) 0x25);
        testBlock.setDigest(mockDigest);
        byte[] buffer = new byte[1024];

        int length = testBlock.encode(buffer, 0);

            assertEquals((byte) 'M', buffer[2]);
            assertEquals((byte) 'B', buffer[3]);
        }

        @Test
        @DisplayName("Should encode with signature when digest is set")
        void testEncodeWithSignature() {
            byte[] buffer = new byte[1024];
            testMessage.setBytesWritten(10);
            testMessage.setDigest(mockDigest);

            int len = testMessage.encode(buffer, 0);