Add @time_func annotation to sleep runner

Signed-off-by: Ganesh Ramadurai <[email protected]>

Author: Gankris96

Diff for: osbenchmark/worker_coordinator/runner.py

@@ -2000,15 +2000,13 @@ class Sleep(Runner):
     """
     Sleeps for the specified duration not issuing any request.
     """
-
+    @time_func
     async def __call__(self, opensearch, params):
         sleep_duration = mandatory(params, "duration", "sleep")
         opensearch.on_request_start()
         try:
-            request_context_holder.on_client_request_start()
             await asyncio.sleep(sleep_duration)
         finally:
-            request_context_holder.on_client_request_end()
             opensearch.on_request_end()
 
     def __repr__(self, *args, **kwargs):

Diff for: tests/worker_coordinator/runner_test.py

@@ -4715,8 +4715,8 @@ async def test_missing_parameter(self, sleep, on_client_request_start, on_client
         self.assertEqual(0, opensearch.call_count)
         self.assertEqual(0, opensearch.on_request_start.call_count)
         self.assertEqual(0, opensearch.on_request_end.call_count)
-        self.assertEqual(0, on_client_request_start.call_count)
-        self.assertEqual(0, on_client_request_end.call_count)
+        self.assertEqual(1, on_client_request_start.call_count)
+        self.assertEqual(1, on_client_request_end.call_count)
         self.assertEqual(0, sleep.call_count)
 
     @mock.patch("opensearchpy.OpenSearch")