Update MutationsBatchError

Mariatta · Mariatta · commit 9eb4dffb6977 · 2023-02-16T10:32:45.000-08:00
diff --git a/google/cloud/bigtable/batcher.py b/google/cloud/bigtable/batcher.py
@@ -37,13 +37,9 @@
 class MutationsBatchError(Exception):
     """Error in the batch request"""
 
-    def __init__(self, status_codes):
-        self.errors = [
-            from_grpc_status(status_code.code, status_code.message)
-            for status_code in status_codes
-            if status_code.code != 0
-        ]
-        self.message = "Errors in batch mutations."
+    def __init__(self, message, exc):
+        self.exc = exc
+        self.message = message
         super().__init__(self.message)
 
 
@@ -312,7 +308,12 @@ def flush_rows(self, rows_to_flush=None):
                 responses.append(result)
 
             if has_error:
-                raise MutationsBatchError(status_codes=responses)
+                exc = [
+                    from_grpc_status(status_code.code, status_code.message)
+                    for status_code in responses
+                    if status_code.code != 0
+                ]
+                raise MutationsBatchError(message="Errors in batch mutations.", exc=exc)
 
         return responses
 
diff --git a/tests/unit/test_batcher.py b/tests/unit/test_batcher.py
@@ -243,10 +243,10 @@ def test_mutations_batcher_response_with_error_codes():
         with pytest.raises(MutationsBatchError) as exc:
             mutation_batcher.flush()
         assert exc.value.message == "Errors in batch mutations."
-        assert len(exc.value.errors) == 2
+        assert len(exc.value.exc) == 2
 
-        assert exc.value.errors[0].message == mocked_response[0].message
-        assert exc.value.errors[1].message == mocked_response[1].message
+        assert exc.value.exc[0].message == mocked_response[0].message
+        assert exc.value.exc[1].message == mocked_response[1].message
 
 
 def test_mutations_batcher_flush_async_raises_exception():
@@ -266,10 +266,10 @@ def test_mutations_batcher_flush_async_raises_exception():
         with pytest.raises(MutationsBatchError) as exc:
             mutation_batcher.flush_async()
         assert exc.value.message == "Errors in batch mutations."
-        assert len(exc.value.errors) == 2
+        assert len(exc.value.exc) == 2
 
-        assert exc.value.errors[0].message == mocked_response[0].message
-        assert exc.value.errors[1].message == mocked_response[1].message
+        assert exc.value.exc[0].message == mocked_response[0].message
+        assert exc.value.exc[1].message == mocked_response[1].message
 
 
 class _Instance(object):
