@@ -370,65 +370,17 @@ public void testExtractDataUris() throws Exception {
370370 }
371371
372372 @ Test
373- public void testGetFromId () throws Exception {
374- initializeExpectedTable (3 );
375- expect (bigquery .getTable (TABLE_INFO .tableId ())).andReturn (expectedTable );
376- replay (bigquery );
377- Table loadedTable = Table .get (bigquery , TABLE_INFO .tableId ());
378- compareTable (expectedTable , loadedTable );
379- }
380-
381- @ Test
382- public void testGetFromStrings () throws Exception {
383- initializeExpectedTable (3 );
384- expect (bigquery .getTable (TABLE_INFO .tableId ())).andReturn (expectedTable );
385- replay (bigquery );
386- Table loadedTable = Table .get (bigquery , TABLE_ID1 .dataset (), TABLE_ID1 .table ());
387- compareTable (expectedTable , loadedTable );
388- }
389-
390- @ Test
391- public void testGetFromIdNull () throws Exception {
392- initializeExpectedTable (1 );
393- expect (bigquery .getTable (TABLE_INFO .tableId ())).andReturn (null );
394- replay (bigquery );
395- assertNull (Table .get (bigquery , TABLE_ID1 ));
396- }
397-
398- @ Test
399- public void testGetFromStringsNull () throws Exception {
373+ public void testBigquery () {
400374 initializeExpectedTable (1 );
401- expect (bigquery .getTable (TABLE_INFO .tableId ())).andReturn (null );
402375 replay (bigquery );
403- assertNull (Table .get (bigquery , TABLE_ID1 .dataset (), TABLE_ID1 .table ()));
404- }
405-
406- @ Test
407- public void testGetFromIdWithOptions () throws Exception {
408- initializeExpectedTable (3 );
409- expect (bigquery .getTable (TABLE_INFO .tableId (), BigQuery .TableOption .fields ()))
410- .andReturn (expectedTable );
411- replay (bigquery );
412- Table loadedTable = Table .get (bigquery , TABLE_INFO .tableId (), BigQuery .TableOption .fields ());
413- compareTable (expectedTable , loadedTable );
414- }
415-
416- @ Test
417- public void testGetFromStringsWithOptions () throws Exception {
418- initializeExpectedTable (3 );
419- expect (bigquery .getTable (TABLE_INFO .tableId (), BigQuery .TableOption .fields ()))
420- .andReturn (expectedTable );
421- replay (bigquery );
422- Table loadedTable =
423- Table .get (bigquery , TABLE_ID1 .dataset (), TABLE_ID1 .table (), BigQuery .TableOption .fields ());
424- compareTable (expectedTable , loadedTable );
376+ assertSame (serviceMockReturnsOptions , expectedTable .bigquery ());
425377 }
426378
427379 @ Test
428- public void testBigquery () {
429- initializeExpectedTable (1 );
380+ public void testToAndFromPb () {
381+ initializeExpectedTable (4 );
430382 replay (bigquery );
431- assertSame ( serviceMockReturnsOptions , expectedTable .bigquery ( ));
383+ compareTable ( expectedTable , Table . fromPb ( serviceMockReturnsOptions , expectedTable .toPb () ));
432384 }
433385
434386 private void compareTable (Table expected , Table value ) {
0 commit comments