From 31bd300f1798347866a1adbbe9400d092dbd3920 Mon Sep 17 00:00:00 2001
From: Tres Seaver <tseaver@palladion.com>
Date: Thu, 1 Nov 2018 16:10:50 -0400
Subject: [PATCH] Add explicit coverage for
 'row_data._retry_read_rows_exception'.

Fixes broken coverage on 'master'.
---
 bigtable/tests/unit/test_row_data.py | 65 ++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)

diff --git a/bigtable/tests/unit/test_row_data.py b/bigtable/tests/unit/test_row_data.py
index 5044dd7c5942..b5a3a797f634 100644
--- a/bigtable/tests/unit/test_row_data.py
+++ b/bigtable/tests/unit/test_row_data.py
@@ -320,6 +320,71 @@ class _Client(object):
     data_stub = None
 
 
+class Test_retry_read_rows_exception(unittest.TestCase):
+
+    @staticmethod
+    def _call_fut(exc):
+        from google.cloud.bigtable.row_data import _retry_read_rows_exception
+
+        return _retry_read_rows_exception(exc)
+
+    @staticmethod
+    def _make_grpc_call_error(exception):
+        from grpc import Call
+        from grpc import RpcError
+
+        class TestingException(Call, RpcError):
+            def __init__(self, exception):
+                self.exception = exception
+
+            def code(self):
+                return self.exception.grpc_status_code
+
+            def details(self):
+                return 'Testing'
+
+        return TestingException(exception)
+
+    def test_w_miss(self):
+        from google.api_core.exceptions import Conflict
+
+        exception = Conflict('testing')
+        self.assertFalse(self._call_fut(exception))
+
+    def test_w_service_unavailable(self):
+        from google.api_core.exceptions import ServiceUnavailable
+
+        exception = ServiceUnavailable('testing')
+        self.assertTrue(self._call_fut(exception))
+
+    def test_w_deadline_exceeded(self):
+        from google.api_core.exceptions import DeadlineExceeded
+
+        exception = DeadlineExceeded('testing')
+        self.assertTrue(self._call_fut(exception))
+
+    def test_w_miss_wrapped_in_grpc(self):
+        from google.api_core.exceptions import Conflict
+
+        wrapped = Conflict('testing')
+        exception = self._make_grpc_call_error(wrapped)
+        self.assertFalse(self._call_fut(exception))
+
+    def test_w_service_unavailable_wrapped_in_grpc(self):
+        from google.api_core.exceptions import ServiceUnavailable
+
+        wrapped = ServiceUnavailable('testing')
+        exception = self._make_grpc_call_error(wrapped)
+        self.assertTrue(self._call_fut(exception))
+
+    def test_w_deadline_exceeded_wrapped_in_grpc(self):
+        from google.api_core.exceptions import DeadlineExceeded
+
+        wrapped = DeadlineExceeded('testing')
+        exception = self._make_grpc_call_error(wrapped)
+        self.assertTrue(self._call_fut(exception))
+
+
 class TestPartialRowsData(unittest.TestCase):
     ROW_KEY = b'row-key'
     FAMILY_NAME = u'family'