Skip to content

Commit 536294c

Browse files
Use ClientWithProject superclass for error reporting. (googleapis#3168)
1 parent 3a8ba72 commit 536294c

File tree

2 files changed

+8
-13
lines changed

2 files changed

+8
-13
lines changed

error_reporting/google/cloud/error_reporting/client.py

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
except ImportError: # pragma: NO COVER
2424
_HAVE_GAX = False
2525

26-
from google.cloud._helpers import _determine_default_project
26+
from google.cloud.client import ClientWithProject
2727
from google.cloud.error_reporting._logging import _ErrorReportingLoggingAPI
2828
from google.cloud.environment_vars import DISABLE_GRPC
2929

@@ -74,7 +74,7 @@ def __init__(self, method=None, url=None,
7474
self.remoteIp = remote_ip
7575

7676

77-
class Client(object):
77+
class Client(ClientWithProject):
7878
"""Error Reporting client. Currently Error Reporting is done by creating
7979
a Logging client.
8080
@@ -125,13 +125,8 @@ def __init__(self, project=None,
125125
service=None,
126126
version=None,
127127
use_gax=None):
128-
if project is None:
129-
self._project = _determine_default_project()
130-
else:
131-
self._project = project
132-
self._credentials = credentials
133-
self._http = http
134-
128+
super(Client, self).__init__(project=project, credentials=credentials,
129+
http=http)
135130
self._report_errors_api = None
136131

137132
self.service = service if service else self.DEFAULT_SERVICE
@@ -162,7 +157,7 @@ def report_errors_api(self):
162157
self._report_errors_api = make_report_error_api(self)
163158
else:
164159
self._report_errors_api = _ErrorReportingLoggingAPI(
165-
self._project, self._credentials, self._http)
160+
self.project, self._credentials, self._http)
166161
return self._report_errors_api
167162

168163
def _build_error_report(self,

error_reporting/unit_tests/test_client.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -52,14 +52,14 @@ def _get_report_payload(self, error_api):
5252
self.assertEqual(len(positional), 1)
5353
return positional[0]
5454

55-
@mock.patch(
56-
'google.cloud.error_reporting.client._determine_default_project')
55+
@mock.patch('google.cloud.client._determine_default_project')
5756
def test_ctor_default(self, default_mock):
5857
credentials = _make_credentials()
58+
default_mock.return_value = 'foo'
5959
client = self._make_one(credentials=credentials)
6060
self.assertEqual(client.service, client.DEFAULT_SERVICE)
6161
self.assertEqual(client.version, None)
62-
default_mock.assert_called_once_with()
62+
default_mock.assert_called_once_with(None)
6363

6464
def test_ctor_params(self):
6565
credentials = _make_credentials()

0 commit comments

Comments
 (0)