Update test_pod_worker.py

runpod · Dec 19, 2022 · 783b773 · 783b773
1 parent 90e860f
commit 783b773
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/test_serverless/test_pod_worker.py b/tests/test_serverless/test_pod_worker.py
@@ -9,12 +9,10 @@
 class TestPodWorker(unittest.TestCase):
     ''' Tests for pod_worker '''
 
-    @patch('runpod.serverless.modules.lifecycle.LifecycleManager.heartbeat_ping')
     @patch('runpod.serverless.modules.lifecycle.LifecycleManager')
-    def test_start_worker(self, mock_lifecycle, mock_heartbeat_ping):
+    def test_start_worker(self, mock_lifecycle):
         '''
         Tests start_worker
         '''
         pod_worker.start_worker()
         mock_lifecycle.assert_called_once_with()
-        mock_heartbeat_ping.assert_called_once_with()