laraschmidt commented on a change in pull request #15540:
URL: https://github.com/apache/beam/pull/15540#discussion_r746171750
##########
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.
Review comment:
Yeah though I still feel a bit odd checking specific error messages in
multiple runners. I updated it to check the relevant pieces but not the
specific wording.
--
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]
