Add missing batch rpc calls to python coverage logs

Without this change, batch RPC calls are not included in coverage logs.
pull/11277/head
Russell Yanofsky 7 years ago
parent 505530c6cf
commit 74182f235c

@ -45,18 +45,24 @@ class AuthServiceProxyWrapper(object):
""" """
return_val = self.auth_service_proxy_instance.__call__(*args, **kwargs) return_val = self.auth_service_proxy_instance.__call__(*args, **kwargs)
self._log_call()
return return_val
def _log_call(self):
rpc_method = self.auth_service_proxy_instance._service_name rpc_method = self.auth_service_proxy_instance._service_name
if self.coverage_logfile: if self.coverage_logfile:
with open(self.coverage_logfile, 'a+', encoding='utf8') as f: with open(self.coverage_logfile, 'a+', encoding='utf8') as f:
f.write("%s\n" % rpc_method) f.write("%s\n" % rpc_method)
return return_val
def __truediv__(self, relative_uri): def __truediv__(self, relative_uri):
return AuthServiceProxyWrapper(self.auth_service_proxy_instance / relative_uri, return AuthServiceProxyWrapper(self.auth_service_proxy_instance / relative_uri,
self.coverage_logfile) self.coverage_logfile)
def get_request(self, *args, **kwargs):
self._log_call()
return self.auth_service_proxy_instance.get_request(*args, **kwargs)
def get_filename(dirname, n_node): def get_filename(dirname, n_node):
""" """
Get a filename unique to the test process ID and node. Get a filename unique to the test process ID and node.

Loading…
Cancel
Save