Skip to content

Commit 8052824

Browse files
committed
fix tests related to new state added, fix spotbugs
1 parent 6e2fd02 commit 8052824

2 files changed

Lines changed: 18 additions & 26 deletions

File tree

runners/core-java/src/main/java/org/apache/beam/runners/core/CombinedMetadataCombiner.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -29,17 +29,15 @@ public static CombinedMetadataCombiner of() {
2929
return INSTANCE;
3030
}
3131

32-
private final CausedByDrainCombiner causedByDrainCombiner = CausedByDrainCombiner.of();
33-
3432
@Override
3533
public CombinedMetadata createAccumulator() {
36-
return CombinedMetadata.create(causedByDrainCombiner.createAccumulator());
34+
return CombinedMetadata.create(CausedByDrainCombiner.of().createAccumulator());
3735
}
3836

3937
@Override
4038
public CombinedMetadata addInput(CombinedMetadata accumulator, CombinedMetadata input) {
4139
return CombinedMetadata.create(
42-
causedByDrainCombiner.addInput(accumulator.causedByDrain(), input.causedByDrain()));
40+
CausedByDrainCombiner.of().addInput(accumulator.causedByDrain(), input.causedByDrain()));
4341
}
4442

4543
@Override

runners/google-cloud-dataflow-java/worker/src/test/java/org/apache/beam/runners/dataflow/worker/StreamingDataflowWorkerTest.java

Lines changed: 16 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -1953,20 +1953,11 @@ public void testMergeWindows() throws Exception {
19531953
assertThat(
19541954
"" + actualOutput.getValueUpdatesList(),
19551955
actualOutput.getValueUpdatesList(),
1956-
Matchers.containsInAnyOrder(
1956+
Matchers.contains(
19571957
Matchers.equalTo(
19581958
Windmill.TagValue.newBuilder()
19591959
.setTag(paneInfoTag)
19601960
.setStateFamily(stateFamily)
1961-
.setValue(
1962-
Windmill.Value.newBuilder()
1963-
.setTimestamp(Long.MAX_VALUE)
1964-
.setData(ByteString.EMPTY))
1965-
.build()),
1966-
Matchers.equalTo(
1967-
Windmill.TagValue.newBuilder()
1968-
.setTag(combinedMetadataTag)
1969-
.setStateFamily(stateFamily)
19701961
.setValue(
19711962
Windmill.Value.newBuilder()
19721963
.setTimestamp(Long.MAX_VALUE)
@@ -1976,12 +1967,18 @@ public void testMergeWindows() throws Exception {
19761967
assertThat(
19771968
"" + actualOutput.getBagUpdatesList(),
19781969
actualOutput.getBagUpdatesList(),
1979-
Matchers.contains(
1970+
Matchers.containsInAnyOrder(
19801971
Matchers.equalTo(
19811972
Windmill.TagBag.newBuilder()
19821973
.setTag(bufferTag)
19831974
.setStateFamily(stateFamily)
19841975
.setDeleteAll(true)
1976+
.build()),
1977+
Matchers.equalTo(
1978+
Windmill.TagBag.newBuilder()
1979+
.setTag(combinedMetadataTag)
1980+
.setStateFamily(stateFamily)
1981+
.setDeleteAll(true)
19851982
.build())));
19861983

19871984
verifyHolds(
@@ -2260,20 +2257,11 @@ public void testMergeWindowsCaching() throws Exception {
22602257
assertThat(
22612258
"" + actualOutput.getValueUpdatesList(),
22622259
actualOutput.getValueUpdatesList(),
2263-
Matchers.containsInAnyOrder(
2260+
Matchers.contains(
22642261
Matchers.equalTo(
22652262
Windmill.TagValue.newBuilder()
22662263
.setTag(paneInfoTag)
22672264
.setStateFamily(stateFamily)
2268-
.setValue(
2269-
Windmill.Value.newBuilder()
2270-
.setTimestamp(Long.MAX_VALUE)
2271-
.setData(ByteString.EMPTY))
2272-
.build()),
2273-
Matchers.equalTo(
2274-
Windmill.TagValue.newBuilder()
2275-
.setTag(combinedMetadataTag)
2276-
.setStateFamily(stateFamily)
22772265
.setValue(
22782266
Windmill.Value.newBuilder()
22792267
.setTimestamp(Long.MAX_VALUE)
@@ -2283,12 +2271,18 @@ public void testMergeWindowsCaching() throws Exception {
22832271
assertThat(
22842272
"" + actualOutput.getBagUpdatesList(),
22852273
actualOutput.getBagUpdatesList(),
2286-
Matchers.contains(
2274+
Matchers.containsInAnyOrder(
22872275
Matchers.equalTo(
22882276
Windmill.TagBag.newBuilder()
22892277
.setTag(bufferTag)
22902278
.setStateFamily(stateFamily)
22912279
.setDeleteAll(true)
2280+
.build()),
2281+
Matchers.equalTo(
2282+
Windmill.TagBag.newBuilder()
2283+
.setTag(combinedMetadataTag)
2284+
.setStateFamily(stateFamily)
2285+
.setDeleteAll(true)
22922286
.build())));
22932287

22942288
verifyHolds(

0 commit comments

Comments
 (0)