assertDoesNotThrow(() -> concreteImplementation.fixupDomain(domain));
        }

        @Test
        @DisplayName("Should strip path consumed")
        void testStripPathConsumed() {
            int consumedAmount = 5;

            // Test with mock
            doNothing().when(mockReferralData).stripPathConsumed(consumedAmount);

    // Note that this may be be an encoded "end of data" header.
    return DerHeader(tagClass, tag, constructed, length)
  }

  /**
   * Consume a header and execute [block], which should consume the entire value described by the
   * header. It is an error to not consume a full value in [block].
   */
  internal inline fun <T> read(
    name: String?,
    block: (DerHeader) -> T,
  ): T {

  /**
   * Null once the file has a complete copy of the upstream bytes. Only the [upstreamReader] thread
   * may access this source.
   */
  var upstream: Source?,
  /** The number of bytes consumed from [upstream]. Guarded by this. */
  var upstreamPos: Long,
  /** User-supplied additional data persisted with the source data. */
  private val metadata: ByteString,
  /** The maximum size of [buffer]. */

            assertEquals(4096, response.getChunkBytesWritten());
            assertEquals(12288, response.getTotalBytesWritten());
        }

        @Test
        @DisplayName("Should return correct bytes consumed")
        void testReturnBytesConsumed() throws SMBProtocolDecodingException {
            byte[] buffer = new byte[100];
            byte[] responseData = createValidCopyChunkResponse(8, 16384, 131072);

                        next = next.next();
                    } while (next != dr);

                    if (dr.getPathConsumed() > (path != null ? path.length() : 0)) {
                        log.error("Consumed more than we provided");
                    }

                    link = path != null && dr.getPathConsumed() > 0 ? path.substring(0, dr.getPathConsumed()) : "\\";
                    dr.setLink(link);

  }

  /**
   * Adds each member of {@code elements} as a candidate for the top {@code k} elements. This
   * operation takes amortized linear time in the length of {@code elements}. The iterator is
   * consumed after this operation completes.
   *
   * <p>If all input data to this {@code TopKSelector} is in a single {@code Iterator}, prefer
   * {@link Ordering#leastOf(Iterator, int)}, which provides a simpler API for that use case.

        assertEquals("smb://dfs-server/dfs-share" + l.getUNCPath().replace('\\', '/'), l.getDfsPath());
        assertEquals("dfs-share", l.getShare());

        // Path consumed negative -> coerced to 0
        when(dr.getPathConsumed()).thenReturn(-1);
        String unc2 = l.handleDFSReferral(dr, null);
        assertTrue(unc2.contains("dfs/path") || unc2.contains("dfs\\path"));
    }

        // Decode
        int bytesDecoded = smbInfoAllocation.decode(buffer, 0, buffer.length);

        // Verify we read exactly 20 bytes (4 + 4 + 4 + 4 + 4)
        // Note: bytesPerSect is read as Int2 but 4 bytes are consumed (padding)
        assertEquals(20, bytesDecoded);
    }

    @Test
    @DisplayName("Test capacity calculation overflow handling")
    void testCapacityCalculationOverflow() throws SMBProtocolDecodingException {

            assertEquals(free * sectPerAlloc * bytesPerSect, fileFsSizeInfo.getFree());
        }

        @Test
        @DisplayName("Should return correct number of bytes consumed from buffer")
        void shouldReturnCorrectBytesConsumed() throws SMBProtocolDecodingException {
            // Given - larger buffer than needed
            byte[] buffer = new byte[100];

            assertEquals(callerFree * sectPerAlloc * bytesPerSect, fileFsFullSizeInfo.getFree());
        }

        @Test
        @DisplayName("Should return correct number of bytes consumed from buffer")
        void shouldReturnCorrectBytesConsumed() throws SMBProtocolDecodingException {
            // Given - larger buffer than needed
            byte[] buffer = new byte[100];