Fixed 2 Flaky Tests (#15376)

This commit is contained in:
Krishna Anandan 2023-11-15 07:10:09 -06:00 committed by GitHub
parent bedf246ed2
commit 4ca5acdc33
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 4 deletions

View File

@ -24,6 +24,7 @@ import com.fasterxml.jackson.databind.Module;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Lists;
import com.google.common.io.Files;
import org.apache.druid.common.config.NullHandling;
@ -57,6 +58,7 @@ import org.junit.rules.ExpectedException;
import java.io.File;
import java.io.IOException;
import java.util.Collections;
import java.util.HashSet;
public class ProtobufInputFormatTest
{
@ -231,7 +233,7 @@ public class ProtobufInputFormatTest
).read().next();
Assert.assertEquals(
ImmutableList.builder()
ImmutableSet.builder()
.add("eventType")
.add("foobar")
.add("bar0")
@ -244,7 +246,7 @@ public class ProtobufInputFormatTest
.add("id")
.add("someBytesColumn")
.build(),
row.getDimensions()
new HashSet<>(row.getDimensions())
);
ProtobufInputRowParserTest.verifyNestedData(row, dateTime);
@ -368,7 +370,7 @@ public class ProtobufInputFormatTest
InputRow row = transformingReader.read().next();
Assert.assertEquals(
ImmutableList.of(
ImmutableSet.of(
"someOtherId",
"someIntColumn",
"isValid",
@ -381,7 +383,7 @@ public class ProtobufInputFormatTest
"id",
"someBytesColumn"
),
row.getDimensions()
new HashSet<>(row.getDimensions())
);
Assert.assertEquals(ImmutableMap.of("bar", "baz"), row.getRaw("foo"));