Fix CI Test

nbauernfeind · Jan 10, 2024 · 5966196 · 5966196
1 parent 74363be
commit 5966196
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/engine/table/src/test/java/io/deephaven/engine/table/impl/TestPartitionAwareSourceTable.java b/engine/table/src/test/java/io/deephaven/engine/table/impl/TestPartitionAwareSourceTable.java
@@ -387,7 +387,10 @@ public Object invoke(Invocation invocation) {
 
         errorNotification.reset();
         final ControlledUpdateGraph updateGraph = ExecutionContext.getContext().getUpdateGraph().cast();
-        updateGraph.runWithinUnitTestCycle(SUT::refresh);
+        updateGraph.runWithinUnitTestCycle(() -> {
+            SUT.refresh();
+            updateGraph.markSourcesRefreshedForUnitTests();
+        }, false);
         assertIsSatisfied();
         errorNotification.assertInvoked();