diff --git a/google-cloud-clients/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/JobStatistics.java b/google-cloud-clients/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/JobStatistics.java index 633a0bdbfdd3..ac6e4a85750a 100644 --- a/google-cloud-clients/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/JobStatistics.java +++ b/google-cloud-clients/google-cloud-bigquery/src/main/java/com/google/cloud/bigquery/JobStatistics.java @@ -175,6 +175,7 @@ public static class LoadStatistics extends JobStatistics { private final Long inputFiles; private final Long outputBytes; private final Long outputRows; + private final Long badRecords; static final class Builder extends JobStatistics.Builder { @@ -182,6 +183,7 @@ static final class Builder extends JobStatistics.Builder REFERENCED_TABLES = ImmutableList.of(TableId.of("foo", "bar", "table1"), TableId.of("foo", "bar", "table2")); private static final List FILE_COUNT = ImmutableList.of(1L, 2L, 3L); @@ -74,6 +75,7 @@ public class JobStatisticsTest { .setInputFiles(INPUT_FILES) .setOutputBytes(OUTPUT_BYTES) .setOutputRows(OUTPUT_ROWS) + .setBadRecords(BAD_RECORDS) .build(); private static final LoadStatistics LOAD_STATISTICS_INCOMPLETE = LoadStatistics.newBuilder() @@ -82,6 +84,7 @@ public class JobStatisticsTest { .setStartTime(START_TIME) .setInputBytes(INPUT_BYTES) .setInputFiles(INPUT_FILES) + .setBadRecords(BAD_RECORDS) .build(); private static final List SUBSTEPS1 = ImmutableList.of("substep1", "substep2"); private static final List SUBSTEPS2 = ImmutableList.of("substep3", "substep4"); @@ -168,6 +171,7 @@ public void testBuilder() { assertEquals(INPUT_FILES, LOAD_STATISTICS.getInputFiles()); assertEquals(OUTPUT_BYTES, LOAD_STATISTICS.getOutputBytes()); assertEquals(OUTPUT_ROWS, LOAD_STATISTICS.getOutputRows()); + assertEquals(BAD_RECORDS, LOAD_STATISTICS.getBadRecords()); assertEquals(CREATION_TIME, QUERY_STATISTICS.getCreationTime()); assertEquals(START_TIME, QUERY_STATISTICS.getStartTime()); @@ -192,6 +196,7 @@ public void testBuilder() { assertEquals(END_TIME, LOAD_STATISTICS_INCOMPLETE.getEndTime()); assertEquals(INPUT_BYTES, LOAD_STATISTICS_INCOMPLETE.getInputBytes()); assertEquals(INPUT_FILES, LOAD_STATISTICS_INCOMPLETE.getInputFiles()); + assertEquals(BAD_RECORDS, LOAD_STATISTICS_INCOMPLETE.getBadRecords()); assertEquals(null, LOAD_STATISTICS_INCOMPLETE.getOutputBytes()); assertEquals(null, LOAD_STATISTICS_INCOMPLETE.getOutputRows());