import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import org.jspecify.annotations.NullMarked;

/**
 * Generators of different types of map and related collections, such as keys, entries and values.
 *
 * @author Hayward Chan
 */
@GwtCompatible
@NullMarked
public class MapGenerators {

    // TODO(kevinb): check more preconditions (as bufferSize >= chunkSize) if this is ever public
    checkArgument(bufferSize % chunkSize == 0);

    // TODO(user): benchmark performance difference with longer buffer
    // always space for a single primitive
    this.buffer = ByteBuffer.allocate(bufferSize + 7).order(ByteOrder.LITTLE_ENDIAN);
    this.bufferSize = bufferSize;
    this.chunkSize = chunkSize;
  }

        // 1. GIVEN
        KerberosAuthData mockAuthData1 = mock(KerberosAuthData.class);
        KerberosAuthData mockAuthData2 = mock(KerberosAuthData.class);

        // Mock the parse method to return two different objects based on input
        // Must use matchers for all arguments when using any matcher
        mockedStaticAuthData.when(() -> KerberosAuthData.parse(eq(1), eq(new byte[] { 0x01 }), any(Map.class)))

        assertArrayEquals(secondData, secondResult);
    }

    @Test
    void testReadDataWireFormatWithOffset() throws SMBProtocolDecodingException {
        // Test reading data from different buffer offsets
        byte[] testData = "Test data with offset".getBytes();
        byte[] buffer = new byte[200];
        int offset = 50;
        System.arraycopy(testData, 0, buffer, offset, testData.length);

set +e
MAX_RETRIES=2
for ((i=1; i <= $MAX_RETRIES; i++)); do
  tfrun bazel --version
done
set -e

# Single handler for all cleanup actions, triggered on an EXIT trap.
# TODO(angerson) Making this use different scripts may be overkill.
cleanup() {
  if [[ "$TFCI_DOCKER_ENABLE" == 1 ]]; then
    ./ci/official/utilities/cleanup_docker.sh
  fi
  ./ci/official/utilities/cleanup_summary.sh
}

            lastResponseData = null;
            returnedResultData = null;
            throwException = false;
        }
    }

    /**
     * Another concrete implementation for testing different scenarios
     */
    public static class NameTrackingTransformer extends AbstractTransformer {
        private final List<String> nameHistory = new ArrayList<>();

        @Override

        assertNull("Should return null in test environment", objectResult);
    }

    public void test_setObject_withDifferentTypes() {
        // Test setObject with different object types
        try {
            WebApiUtil.setObject("stringValue", "test");
            WebApiUtil.setObject("intValue", 123);
            WebApiUtil.setObject("booleanValue", true);

        int actualInfoLevel = SMBUtil.readInt2(buffer, 0);
        assertEquals(0x0103, actualInfoLevel);
    }

    @DisplayName("Test writeParametersWireFormat with different information levels")
    @ParameterizedTest
    @CsvSource({ "-1, 1", // SMB_INFO_ALLOCATION -> 0x0001
            "3, 259" // FS_SIZE_INFO -> 0x0103
    })

            assertTrue(utf16Bytes.length > 0, "UTF-16LE bytes should not be empty");
            assertNotEquals(utf8Bytes.length, utf16Bytes.length, "Different encodings should produce different lengths");
        }

        @ParameterizedTest
        @DisplayName("getBytes should handle various string inputs")
        @NullAndEmptySource

            @Override
            public RescorerBuilder<?> evaluate(Map<String, Object> params) {
                if (params == null) {
                    return null;
                }

                // Test different parameter types
                if (params.containsKey("string") && params.get("string") instanceof String) {
                    return createMockRescorerBuilder();
                }