. (#253)

Author: yirutang

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1alpha2/StreamWriterTest.java

@@ -432,8 +432,13 @@ public void testFlowControlBehaviorException() throws Exception {
         appendFuture2.get();
         Assert.fail("This should fail");
       } catch (Exception e) {
-        LOG.info("ControlFlow test exception: " + e.toString());
-        assertEquals("The maximum number of batch elements: 1 have been reached.", e.getMessage());
+        if (!e.getMessage().equals("The maximum number of batch elements: 1 have been reached.")) {
+          LOG.info("More error info:");
+          e.printStackTrace();
+        }
+        assertEquals(
+            "java.util.concurrent.ExecutionException: The maximum number of batch elements: 1 have been reached.",
+            e.toString());
       }
       assertEquals(1L, appendFuture1.get().getOffset());
     }