diff --git a/zerver/lib/test_helpers.py b/zerver/lib/test_helpers.py index 640787f958..c2db4fe158 100644 --- a/zerver/lib/test_helpers.py +++ b/zerver/lib/test_helpers.py @@ -199,7 +199,10 @@ class POSTRequestMock(object): INSTRUMENTING = os.environ.get('TEST_INSTRUMENT_URL_COVERAGE', '') == 'TRUE' INSTRUMENTED_CALLS = [] +UrlFuncT = Callable[..., HttpResponse] # TODO: make more specific + def instrument_url(f): + # type: (UrlFuncT) -> UrlFuncT if not INSTRUMENTING: return f else: