From 0d02a8801507fd6e45592ab78b36109db8006de0 Mon Sep 17 00:00:00 2001 From: Sam R Date: Fri, 10 Sep 2021 08:33:22 -0700 Subject: [PATCH] [BEAM-12842] Add timestamp to test work item to deflake --- .../dataflow/worker/StreamingDataflowWorkerTest.java | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/StreamingDataflowWorkerTest.java b/runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/StreamingDataflowWorkerTest.java index 12b7df20c538..17b59ecf3737 100644 --- a/runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/StreamingDataflowWorkerTest.java +++ b/runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/StreamingDataflowWorkerTest.java @@ -794,7 +794,8 @@ public void testHotKeyLogging() throws Exception { final int numIters = 2000; for (int i = 0; i < numIters; ++i) { - server.addWorkToOffer(makeInput(i, 0, "key", DEFAULT_SHARDING_KEY)); + server.addWorkToOffer( + makeInput(i, TimeUnit.MILLISECONDS.toMicros(i), "key", DEFAULT_SHARDING_KEY)); } Map result = server.waitForAndGetCommits(numIters); @@ -829,7 +830,8 @@ public void testHotKeyLoggingNotEnabled() throws Exception { final int numIters = 2000; for (int i = 0; i < numIters; ++i) { - server.addWorkToOffer(makeInput(i, 0, "key", DEFAULT_SHARDING_KEY)); + server.addWorkToOffer( + makeInput(i, TimeUnit.MILLISECONDS.toMicros(i), "key", DEFAULT_SHARDING_KEY)); } Map result = server.waitForAndGetCommits(numIters);