Browse Source

Add missing batch rpc calls to python coverage logs

Without this change, batch RPC calls are not included in coverage logs.

Github-Pull: #11277
Rebased-From: 74182f235c
tags/v0.15.1
Russell Yanofsky 3 years ago
parent
commit
1c8c7f8af9
1 changed files with 8 additions and 2 deletions
  1. 8
    2
      test/functional/test_framework/coverage.py

+ 8
- 2
test/functional/test_framework/coverage.py View File

@@ -45,18 +45,24 @@ class AuthServiceProxyWrapper(object):

"""
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

if self.coverage_logfile:
with open(self.coverage_logfile, 'a+', encoding='utf8') as f:
f.write("%s\n" % rpc_method)

return return_val

def __truediv__(self, relative_uri):
return AuthServiceProxyWrapper(self.auth_service_proxy_instance / relative_uri,
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):
"""
Get a filename unique to the test process ID and node.

Loading…
Cancel
Save