initial commit

Author: hmottestad

core/common/iterator/src/test/java/org/eclipse/rdf4j/common/iteration/DistinctIterationTest.java

@@ -0,0 +1,43 @@
+/*******************************************************************************
+ * Copyright (c) 2026 Eclipse RDF4J contributors.
+ *
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Distribution License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/org/documents/edl-v10.php.
+ *
+ * SPDX-License-Identifier: BSD-3-Clause
+ *******************************************************************************/
+// Some portions generated by Codex
+package org.eclipse.rdf4j.common.iteration;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+import java.util.List;
+
+import org.eclipse.rdf4j.http.client.QueryExecutionContext;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.Test;
+
+class DistinctIterationTest {
+
+	@AfterEach
+	void tearDown() {
+		QueryExecutionContext.reset();
+	}
+
+	@Test
+	void shouldConsultQueryExecutionContextWhenFilteringDistinctValues() {
+		QueryExecutionContext.failOnCheckpoint(new RuntimeException("breaker checkpoint"));
+
+		DistinctIteration<Integer> iteration = new DistinctIteration<>(
+				new CloseableIteratorIteration<>(List.of(1, 2, 3).iterator()), java.util.HashSet::new);
+
+		RuntimeException exception = assertThrows(RuntimeException.class, iteration::hasNext);
+		assertEquals("breaker checkpoint", exception.getMessage());
+		assertTrue(QueryExecutionContext.getMarkHeavyCalls() > 0);
+		assertTrue(QueryExecutionContext.getCheckpointCalls() > 0);
+	}
+}