From 7645cc270758a026c2c31d8a7818472abecf3cf6 Mon Sep 17 00:00:00 2001 From: Alan Rominger Date: Wed, 27 Jul 2022 11:53:09 -0400 Subject: [PATCH] Remove mocks for reap method that was removed --- awx/main/tests/functional/test_dispatch.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/awx/main/tests/functional/test_dispatch.py b/awx/main/tests/functional/test_dispatch.py index 4b65726b84..f3c9afe58b 100644 --- a/awx/main/tests/functional/test_dispatch.py +++ b/awx/main/tests/functional/test_dispatch.py @@ -199,10 +199,7 @@ class TestAutoScaling: assert len(self.pool) == 10 # cleanup should scale down to 8 workers - with mock.patch('awx.main.dispatch.reaper.reap') as reap: - with mock.patch('awx.main.dispatch.reaper.reap_waiting') as reap: - self.pool.cleanup() - reap.assert_called() + self.pool.cleanup() assert len(self.pool) == 2 def test_max_scale_up(self): @@ -250,10 +247,7 @@ class TestAutoScaling: time.sleep(1) # wait a moment for sigterm # clean up and the dead worker - with mock.patch('awx.main.dispatch.reaper.reap') as reap: - with mock.patch('awx.main.dispatch.reaper.reap_waiting') as reap: - self.pool.cleanup() - reap.assert_called() + self.pool.cleanup() assert len(self.pool) == 1 assert self.pool.workers[0].pid == alive_pid