From 4ca749dcb72e6f89c6fad83368fb29e9b888376b Mon Sep 17 00:00:00 2001 From: Moritz Mack Date: Wed, 3 Aug 2022 10:35:17 +0200 Subject: [PATCH] Bugfix: Fix broken assertion in PipelineTest (#22485) * Bugfix: Fix broken assertion in PipelineTest * spotless --- .../test/java/org/apache/beam/sdk/PipelineTest.java | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/PipelineTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/PipelineTest.java index 1bb4caab9742..5e19fa03315f 100644 --- a/sdks/java/core/src/test/java/org/apache/beam/sdk/PipelineTest.java +++ b/sdks/java/core/src/test/java/org/apache/beam/sdk/PipelineTest.java @@ -18,7 +18,6 @@ package org.apache.beam.sdk; import static org.hamcrest.MatcherAssert.assertThat; -import static org.hamcrest.Matchers.anyOf; import static org.hamcrest.Matchers.containsString; import static org.hamcrest.Matchers.hasItem; import static org.hamcrest.Matchers.instanceOf; @@ -407,13 +406,9 @@ public void testReplaceAll() { new PipelineVisitor.Defaults() { @Override public CompositeBehavior enterCompositeTransform(Node node) { - if (!node.isRootNode()) { - assertThat( - node.getTransform().getClass(), - not( - anyOf( - Matchers.equalTo(GenerateSequence.class), - Matchers.equalTo(Create.Values.class)))); + String fullName = node.getFullName(); + if (fullName.equals("unbounded") || fullName.equals("bounded")) { + assertThat(node.getTransform(), Matchers.instanceOf(EmptyFlatten.class)); } return CompositeBehavior.ENTER_TRANSFORM; }