@@ -38,6 +38,7 @@ public class ExtractJobInfoTest {
3838 private static final TableId TABLE_ID = TableId .of ("dataset" , "table" );
3939 private static final String FIELD_DELIMITER = "," ;
4040 private static final String FORMAT = "CSV" ;
41+ private static final String JSON_FORMAT = "NEWLINE_DELIMITED_JSON" ;
4142 private static final Boolean PRINT_HEADER = true ;
4243 private static final String COMPRESSION = "GZIP" ;
4344 private static final JobId JOB_ID = JobId .of ("job" );
@@ -95,6 +96,14 @@ public void testOf() {
9596 job = ExtractJobInfo .of (TABLE_ID , DESTINATION_URI );
9697 assertEquals (TABLE_ID , job .sourceTable ());
9798 assertEquals (ImmutableList .of (DESTINATION_URI ), job .destinationUris ());
99+ job = ExtractJobInfo .of (TABLE_ID , JSON_FORMAT , DESTINATION_URIS );
100+ assertEquals (TABLE_ID , job .sourceTable ());
101+ assertEquals (DESTINATION_URIS , job .destinationUris ());
102+ assertEquals (JSON_FORMAT , job .format ());
103+ job = ExtractJobInfo .of (TABLE_ID , JSON_FORMAT , DESTINATION_URI );
104+ assertEquals (TABLE_ID , job .sourceTable ());
105+ assertEquals (ImmutableList .of (DESTINATION_URI ), job .destinationUris ());
106+ assertEquals (JSON_FORMAT , job .format ());
98107 job = ExtractJobInfo .of (JOB_ID , TABLE_ID , DESTINATION_URIS );
99108 assertEquals (JOB_ID , job .jobId ());
100109 assertEquals (TABLE_ID , job .sourceTable ());
@@ -103,6 +112,16 @@ public void testOf() {
103112 assertEquals (JOB_ID , job .jobId ());
104113 assertEquals (TABLE_ID , job .sourceTable ());
105114 assertEquals (ImmutableList .of (DESTINATION_URI ), job .destinationUris ());
115+ job = ExtractJobInfo .of (JOB_ID , TABLE_ID , JSON_FORMAT , DESTINATION_URIS );
116+ assertEquals (JOB_ID , job .jobId ());
117+ assertEquals (TABLE_ID , job .sourceTable ());
118+ assertEquals (DESTINATION_URIS , job .destinationUris ());
119+ assertEquals (JSON_FORMAT , job .format ());
120+ job = ExtractJobInfo .of (JOB_ID , TABLE_ID , JSON_FORMAT , DESTINATION_URI );
121+ assertEquals (JOB_ID , job .jobId ());
122+ assertEquals (TABLE_ID , job .sourceTable ());
123+ assertEquals (ImmutableList .of (DESTINATION_URI ), job .destinationUris ());
124+ assertEquals (JSON_FORMAT , job .format ());
106125 }
107126
108127 @ Test
0 commit comments