laraschmidt commented on a change in pull request #15540:
URL: https://github.com/apache/beam/pull/15540#discussion_r746171825
##########
File path:
sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/ParDoTest.java
##########
@@ -1791,6 +1791,119 @@ public void finishBundle(FinishBundleContext c) {
/** Tests to validate output timestamps. */
@RunWith(JUnit4.class)
public static class TimestampTests extends SharedTestBase implements
Serializable {
+ /**
+ * A {@link DoFn} that outputs elements with timestamp equal to the input
timestamp minus the
+ * input element.
+ */
+ private static class SkewingDoFn extends DoFn<Duration, String> {
+ private final Duration allowedSkew;
+ static final String CORRECT_ELEMENT = "element";
+
+ private SkewingDoFn(Duration allowedSkew) {
+ this.allowedSkew = allowedSkew;
+ }
+
+ @ProcessElement
+ public void processElement(ProcessContext context) {
+ Instant outputTimestamp = context.timestamp().minus(context.element());
+ try {
+ context.outputWithTimestamp(CORRECT_ELEMENT, outputTimestamp);
+ } catch (IllegalArgumentException e) {
+ // Errors differ between runners but at least check that the output
timestamp is printed.
+ if (e.getMessage().contains("timestamp " + outputTimestamp)) {
+ context.output(outputTimestamp.toString());
+ }
+ }
+ }
+
+ @Override
+ public Duration getAllowedTimestampSkew() {
+ return allowedSkew;
+ }
+ }
+
+ /**
+ * A {@link DoFn} that creates/sets a timer with an output timestamp equal
to the input
+ * timestamp minus the input element's value. Keys are ignored but
required for timers.
+ */
+ private static class TimerSkewingDoFn extends DoFn<KV<String, Duration>,
String> {
+ static final String TIMER_ID = "testTimerId";
+ private final Duration allowedSkew;
+ static final String CORRECT_ELEMENT = "element";
+
+ @TimerId(TIMER_ID)
+ private static final TimerSpec timer =
TimerSpecs.timer(TimeDomain.PROCESSING_TIME);
+
+ private TimerSkewingDoFn(Duration allowedSkew) {
+ this.allowedSkew = allowedSkew;
+ }
+
+ @ProcessElement
+ public void processElement(ProcessContext context, @TimerId(TIMER_ID)
Timer timer) {
+
+ Instant outputTimestamp =
context.timestamp().minus(context.element().getValue());
+ try {
+ timer.withOutputTimestamp(outputTimestamp).set(new Instant(0));
+ context.output(CORRECT_ELEMENT);
+ } catch (IllegalArgumentException e) {
+ // Errors differ between runners but at least check that the output
timestamp is printed.
+ if (e.getMessage().contains("timestamp " + outputTimestamp)) {
+ context.output(outputTimestamp.toString());
+ }
+ }
+ }
+
+ @OnTimer(TIMER_ID)
+ public void onTimer(OnTimerContext context) {}
+
+ @Override
+ public Duration getAllowedTimestampSkew() {
+ return allowedSkew;
+ }
+ }
+
+ /**
+ * A {@link DoFn} that creates/sets a timer with an output timestamp equal
to the input
+ * timestamp minus the input element's value. Keys are ignored but
required for timers.
+ */
+ private static class TimerOutputSkewingDoFn extends DoFn<KV<String,
String>, String> {
+ static final String TIMER_ID = "testTimerId";
+ static final String CORRECT_ELEMENT = "element";
+ private final Duration allowedSkew;
+ private final Duration outputTimestampSkew;
+
+ @TimerId(TIMER_ID)
+ private static final TimerSpec timer =
TimerSpecs.timer(TimeDomain.EVENT_TIME);
+
+ private TimerOutputSkewingDoFn(Duration allowedSkew, Duration
outputTimestampSkew) {
+ this.allowedSkew = allowedSkew;
+ this.outputTimestampSkew = outputTimestampSkew;
+ }
+
+ @ProcessElement
+ public void processElement(ProcessContext context, @TimerId(TIMER_ID)
Timer timer) {
+ timer.set(context.timestamp());
+ }
+
+ @OnTimer(TIMER_ID)
+ public void onTimer(OnTimerContext context) {
+ Instant outputTimestamp =
context.timestamp().minus(outputTimestampSkew);
Review comment:
Done.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
