@@ -155,6 +155,9 @@ public class BigQueryImplTest {
155155 TableInfo .of (TABLE_ID , TABLE_DEFINITION_WITH_RANGE_PARTITIONING );
156156 private static final TableInfo TABLE_INFO = TableInfo .of (TABLE_ID , TABLE_DEFINITION );
157157 private static final TableInfo OTHER_TABLE_INFO = TableInfo .of (OTHER_TABLE_ID , TABLE_DEFINITION );
158+ private static final TableInfo OTHER_TABLE_WITH_LABELS_INFO = TableInfo .newBuilder (OTHER_TABLE_ID , TABLE_DEFINITION )
159+ .setLabels (ImmutableMap .of ("key" , "value" ))
160+ .build ();
158161 private static final TableInfo TABLE_INFO_WITH_PROJECT =
159162 TableInfo .of (TABLE_ID_WITH_PROJECT , TABLE_DEFINITION );
160163 private static final TableInfo MODEL_TABLE_INFO_WITH_PROJECT =
@@ -1151,6 +1154,23 @@ public void testListTablesFromDatasetIdWithProject() throws IOException {
11511154 .listTablesSkipExceptionTranslation (OTHER_PROJECT , DATASET , EMPTY_RPC_OPTIONS );
11521155 }
11531156
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 (ImmutableMap .of ("key" , "value" ), page .getValues ().iterator ().next ().getLabels ());
1172+ }
1173+
11541174 @ Test
11551175 public void testListTablesWithOptions () throws IOException {
11561176 bigquery = options .getService ();
0 commit comments