@@ -132,6 +132,7 @@ public class BigQueryImplTest {
132
132
.setField ("timestampField" );
133
133
private static final TimePartitioning TIME_PARTITIONING_NULL_TYPE =
134
134
TimePartitioning .fromPb (PB_TIMEPARTITIONING );
135
+ private static final ImmutableMap <String , String > LABELS = ImmutableMap .of ("key" , "value" );
135
136
private static final StandardTableDefinition TABLE_DEFINITION_WITH_PARTITIONING =
136
137
StandardTableDefinition .newBuilder ()
137
138
.setSchema (TABLE_SCHEMA )
@@ -155,6 +156,8 @@ public class BigQueryImplTest {
155
156
TableInfo .of (TABLE_ID , TABLE_DEFINITION_WITH_RANGE_PARTITIONING );
156
157
private static final TableInfo TABLE_INFO = TableInfo .of (TABLE_ID , TABLE_DEFINITION );
157
158
private static final TableInfo OTHER_TABLE_INFO = TableInfo .of (OTHER_TABLE_ID , TABLE_DEFINITION );
159
+ private static final TableInfo OTHER_TABLE_WITH_LABELS_INFO =
160
+ TableInfo .newBuilder (OTHER_TABLE_ID , TABLE_DEFINITION ).setLabels (LABELS ).build ();
158
161
private static final TableInfo TABLE_INFO_WITH_PROJECT =
159
162
TableInfo .of (TABLE_ID_WITH_PROJECT , TABLE_DEFINITION );
160
163
private static final TableInfo MODEL_TABLE_INFO_WITH_PROJECT =
@@ -1151,6 +1154,23 @@ public void testListTablesFromDatasetIdWithProject() throws IOException {
1151
1154
.listTablesSkipExceptionTranslation (OTHER_PROJECT , DATASET , EMPTY_RPC_OPTIONS );
1152
1155
}
1153
1156
1157
+ @ Test
1158
+ public void testListTablesWithLabels () throws IOException {
1159
+ bigquery = options .getService ();
1160
+ ImmutableList <Table > tableList =
1161
+ ImmutableList .of (
1162
+ new Table (bigquery , new TableInfo .BuilderImpl (OTHER_TABLE_WITH_LABELS_INFO )));
1163
+ Tuple <String , Iterable <com .google .api .services .bigquery .model .Table >> result =
1164
+ Tuple .of (CURSOR , Iterables .transform (tableList , TableInfo .TO_PB_FUNCTION ));
1165
+ when (bigqueryRpcMock .listTablesSkipExceptionTranslation (PROJECT , DATASET , EMPTY_RPC_OPTIONS ))
1166
+ .thenReturn (result );
1167
+ Page <Table > page = bigquery .listTables (DATASET );
1168
+ assertEquals (CURSOR , page .getNextPageToken ());
1169
+ assertArrayEquals (tableList .toArray (), Iterables .toArray (page .getValues (), Table .class ));
1170
+ verify (bigqueryRpcMock ).listTablesSkipExceptionTranslation (PROJECT , DATASET , EMPTY_RPC_OPTIONS );
1171
+ assertEquals (LABELS , page .getValues ().iterator ().next ().getLabels ());
1172
+ }
1173
+
1154
1174
@ Test
1155
1175
public void testListTablesWithOptions () throws IOException {
1156
1176
bigquery = options .getService ();
0 commit comments