Encdec.enc_uint16le((short) 50, buffer, 8); // Fragment length
                return 30; // Initial bytes received
            });

            // Mock second receive to complete fragment
            when(mockSmbPipeHandleInternal.recv(buf, 30, 20)).thenReturn(20);

            int result = handle.doReceiveFragment(buf);
            assertEquals(50, result);

        assertEquals("0 MB", format.format(49L, ScaleUnit.MEGABYTE));
        assertEquals("0 GB", format.format(49L, ScaleUnit.GIGABYTE));

        assertEquals("50 B", format.format(50L));
        assertEquals("50 B", format.format(50L, ScaleUnit.BYTE));
        assertEquals("0.1 kB", format.format(50L, ScaleUnit.KILOBYTE));
        assertEquals("0 MB", format.format(50L, ScaleUnit.MEGABYTE));

      "FINE: Q10000 run again after  50 µs: task",
      "FINE: Q10000 starting              : task",
      "FINE: Q10000 finished run in   0 µs: task",
      "FINE: Q10000 run again after 150 µs: task",
      "FINE: Q10000 starting              : task",
      "FINE: Q10000 finished run in   0 µs: task",
    )
  }

  /** Repeat with a delay of 200 but schedule with a delay of 50. The schedule wins. */

                Arguments.of(49L, ScaleUnit.MEGABYTE, "0 MB"),
                Arguments.of(49L, ScaleUnit.GIGABYTE, "0 GB"),

                // 50 bytes
                Arguments.of(50L, null, "50 B"),
                Arguments.of(50L, ScaleUnit.BYTE, "50 B"),
                Arguments.of(50L, ScaleUnit.KILOBYTE, "0.1 kB"),
                Arguments.of(50L, ScaleUnit.MEGABYTE, "0 MB"),

            @Override
            public String getLtrModelName() {
                return "";
            }

            @Override
            public int getLtrWindowSize() {
                return 50;
            }
        });

        Map<String, Object> params = new HashMap<>();
        params.put("param1", "value1");

        RescorerBuilder<?> result = ltrQueryRescorer.evaluate(params);

        String seed = "very_long_seed_" + "x".repeat(100);
        String[] tags = { "long_tag_" + "y".repeat(50), "another_long_tag_" + "z".repeat(50) };
        String[] roles = { "long_role_" + "a".repeat(50) };
        String[] fields = { "long_field_" + "b".repeat(50) };
        String[] excludes = { "long_exclude_" + "c".repeat(50) };

        String result = popularWordHelper.getCacheKey(seed, tags, roles, fields, excludes);

    void testSizeWithOutputData() {
        setupMockConfig();
        Smb2IoctlRequest request = new Smb2IoctlRequest(mockConfig, TEST_CONTROL_CODE);
        when(mockOutputData.size()).thenReturn(50);
        request.setOutputData(mockOutputData);

        int size = request.size();

        // Base size 120 + 50 = 170, aligned to 8 = 176
        assertEquals(176, size);

    assertThat(cache.getIfPresent(10)).isEqualTo(50);
    assertThat(asMap.get(10)).isEqualTo(50);
    assertThat(cache.getIfPresent(2)).isEqualTo(79);
    assertThat(asMap.get(2)).isEqualTo(79);

    asMap.remove(10, 100);
    asMap.remove(2, 79);

    assertThat(cache.getIfPresent(10)).isEqualTo(50);
    assertThat(asMap.get(10)).isEqualTo(50);
    assertThat(cache.getIfPresent(2)).isNull();

    void testDecodeWithDifferentOffset() throws SMBProtocolDecodingException {
        // Prepare test data with different offset
        byte[] buffer = new byte[100];
        int bufferIndex = 50;

        // Set test values
        int testCapabilities = 0xABCDEF00;
        byte[] testGuid = new byte[16];
        Arrays.fill(testGuid, (byte) 0xFF);
        int testSecurityMode = 0x0001;

            // When
            smbComTreeDisconnect.writeParameterWordsWireFormat(originalBuffer, 50);
            smbComTreeDisconnect.writeBytesWireFormat(originalBuffer, 50);
            smbComTreeDisconnect.readParameterWordsWireFormat(originalBuffer, 50);
            smbComTreeDisconnect.readBytesWireFormat(originalBuffer, 50);

            // Then - buffer should remain unchanged since all methods return 0