Stabilize WAL search test (#5655)

GH-0000 Stabilize WAL search test

Author: hmottestad

core/sail/nativerdf/src/test/java/org/eclipse/rdf4j/sail/nativerdf/wal/ValueStoreWalSearchTest.java

@@ -8,6 +8,7 @@
  *
  * SPDX-License-Identifier: BSD-3-Clause
  *******************************************************************************/
+// Some portions generated by Codex
 
 package org.eclipse.rdf4j.sail.nativerdf.wal;
 
@@ -41,6 +42,7 @@ void findsValueByIdViaSegmentProbe() throws Exception {
 		// Configure NativeStore with small WAL segment size to ensure multiple segments possible
 		NativeStoreConfig cfg = new NativeStoreConfig("spoc,ospc,psoc");
 		cfg.setWalMaxSegmentBytes(64 * 1024); // 64 KiB
+		cfg.setWalSyncPolicy(ValueStoreWalConfig.SyncPolicy.ALWAYS.name());
 		NativeStore store = (NativeStore) new NativeStoreFactory().getSail(cfg);
 		store.setDataDir(dataDir);
 		SailRepository repo = new SailRepository(store);
@@ -58,11 +60,20 @@ void findsValueByIdViaSegmentProbe() throws Exception {
 		ValueStoreWalConfig cfgRead = ValueStoreWalConfig.builder().walDirectory(walDir).storeUuid(storeUuid).build();
 
 		// Build dictionary of minted values from WAL and pick a random entry
-		Map<Integer, ValueStoreWalRecord> dict;
-		try (ValueStoreWalReader reader = ValueStoreWalReader.open(cfgRead)) {
-			dict = new ValueStoreWalRecovery().replay(reader);
+		Map<Integer, ValueStoreWalRecord> dict = Map.of();
+		boolean complete = false;
+		for (int attempt = 0; attempt < 10 && (!complete || dict.isEmpty()); attempt++) {
+			try (ValueStoreWalReader reader = ValueStoreWalReader.open(cfgRead)) {
+				ValueStoreWalRecovery.ReplayReport report = new ValueStoreWalRecovery().replayWithReport(reader);
+				dict = report.dictionary();
+				complete = report.complete();
+			}
+			if (!complete || dict.isEmpty()) {
+				Thread.sleep(100);
+			}
 		}
 		assertThat(dict).isNotEmpty();
+		assertThat(complete).as("WAL recovery should be complete before searching").isTrue();
 		Integer[] ids = dict.keySet().toArray(Integer[]::new);
 		Integer pickId = ids[new Random().nextInt(ids.length)];