fix: update resumable errors to match string from test failure

tswast · tswast · commit 50ecc98466c1 · 2020-01-08T14:52:48.000-06:00
diff --git a/bigquery_storage/google/cloud/bigquery_storage_v1beta1/reader.py b/bigquery_storage/google/cloud/bigquery_storage_v1beta1/reader.py
@@ -49,8 +49,8 @@
 # for error message that we know are caused by problems that are safe to
 # reconnect.
 _STREAM_RESUMPTION_INTERNAL_ERROR_MESSAGES = (
-    # See: https://issuetracker.google.com/143292803
-    "unexpected EOS on DATA frame",
+    # See: https://github.com/googleapis/google-cloud-python/pull/9994
+    "RST_STREAM",
 )
 
 _FASTAVRO_REQUIRED = (
diff --git a/bigquery_storage/tests/unit/test_reader.py b/bigquery_storage/tests/unit/test_reader.py
@@ -185,7 +185,7 @@ def _avro_blocks_w_resumable_internal_error(avro_blocks):
     for block in avro_blocks:
         yield block
     raise google.api_core.exceptions.InternalServerError(
-        "INTERNAL: Received unexpected EOS on DATA frame from server."
+        "INTERNAL: Received RST_STREAM with error code 2."
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -185,7 +185,7 @@ def _avro_blocks_w_resumable_internal_error(avro_blocks):`
`185`	`185`	`for block in avro_blocks:`
`186`	`186`	`yield block`
`187`	`187`	`raise google.api_core.exceptions.InternalServerError(`
`188`		`- "INTERNAL: Received unexpected EOS on DATA frame from server."`
	`188`	`+ "INTERNAL: Received RST_STREAM with error code 2."`
`189`	`189`	`)`
`190`	`190`
`191`	`191`