From 1c8c7f8af926175896cfd0d0e8b8b8663f863505 Mon Sep 17 00:00:00 2001 From: Russell Yanofsky Date: Wed, 4 Oct 2017 03:07:01 -0400 Subject: 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: 74182f235cd04dcac7a8b3e763bc9add549745e1 --- test/functional/test_framework/coverage.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/test/functional/test_framework/coverage.py b/test/functional/test_framework/coverage.py index 8101775ce1..84049e76bc 100644 --- a/test/functional/test_framework/coverage.py +++ b/test/functional/test_framework/coverage.py @@ -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. -- cgit v1.2.3