Skip to content

Commit 74182f2

Browse files
committed
Add missing batch rpc calls to python coverage logs
Without this change, batch RPC calls are not included in coverage logs.
1 parent 505530c commit 74182f2

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

test/functional/test_framework/coverage.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,18 +45,24 @@ def __call__(self, *args, **kwargs):
4545
4646
"""
4747
return_val = self.auth_service_proxy_instance.__call__(*args, **kwargs)
48+
self._log_call()
49+
return return_val
50+
51+
def _log_call(self):
4852
rpc_method = self.auth_service_proxy_instance._service_name
4953

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

54-
return return_val
55-
5658
def __truediv__(self, relative_uri):
5759
return AuthServiceProxyWrapper(self.auth_service_proxy_instance / relative_uri,
5860
self.coverage_logfile)
5961

62+
def get_request(self, *args, **kwargs):
63+
self._log_call()
64+
return self.auth_service_proxy_instance.get_request(*args, **kwargs)
65+
6066
def get_filename(dirname, n_node):
6167
"""
6268
Get a filename unique to the test process ID and node.

0 commit comments

Comments
 (0)