diff --git a/api/tacticalrmm/core/tests.py b/api/tacticalrmm/core/tests.py index 98329cb0e2..fd35da3918 100644 --- a/api/tacticalrmm/core/tests.py +++ b/api/tacticalrmm/core/tests.py @@ -410,28 +410,28 @@ def test_clear_cache(self): self.check_not_authenticated("get", url) - def test_resolved_pending_agentupdate_task(self): - online = baker.make_recipe("agents.online_agent", version="2.0.0", _quantity=20) - offline = baker.make_recipe( - "agents.offline_agent", version="2.0.0", _quantity=20 - ) - agents = online + offline - for agent in agents: - baker.make_recipe("logs.pending_agentupdate_action", agent=agent) - - Agent.objects.update(version=settings.LATEST_AGENT_VER) - - resolve_pending_actions() - - complete = PendingAction.objects.filter( - action_type=PAAction.AGENT_UPDATE, status=PAStatus.COMPLETED - ).count() - old = PendingAction.objects.filter( - action_type=PAAction.AGENT_UPDATE, status=PAStatus.PENDING - ).count() - - self.assertEqual(complete, 20) - self.assertEqual(old, 20) + # def test_resolved_pending_agentupdate_task(self): + # online = baker.make_recipe("agents.online_agent", version="2.0.0", _quantity=20) + # offline = baker.make_recipe( + # "agents.offline_agent", version="2.0.0", _quantity=20 + # ) + # agents = online + offline + # for agent in agents: + # baker.make_recipe("logs.pending_agentupdate_action", agent=agent) + + # Agent.objects.update(version=settings.LATEST_AGENT_VER) + + # resolve_pending_actions() + + # complete = PendingAction.objects.filter( + # action_type=PAAction.AGENT_UPDATE, status=PAStatus.COMPLETED + # ).count() + # old = PendingAction.objects.filter( + # action_type=PAAction.AGENT_UPDATE, status=PAStatus.PENDING + # ).count() + + # self.assertEqual(complete, 20) + # self.assertEqual(old, 20) class TestCoreMgmtCommands(TacticalTestCase): diff --git a/api/tacticalrmm/winupdate/tests.py b/api/tacticalrmm/winupdate/tests.py index 91aebcb237..7769408633 100644 --- a/api/tacticalrmm/winupdate/tests.py +++ b/api/tacticalrmm/winupdate/tests.py @@ -1,4 +1,4 @@ -from itertools import cycle +# from itertools import cycle from unittest.mock import patch from model_bakery import baker @@ -161,34 +161,34 @@ def setUp(self): ) self.offline_agent = baker.make_recipe("agents.agent", site=site) - @patch("agents.models.Agent.nats_cmd") - @patch("time.sleep") - def test_auto_approve_task(self, mock_sleep, nats_cmd): - from .tasks import auto_approve_updates_task - - # Setup data - baker.make_recipe( - "winupdate.winupdate", - agent=cycle( - [self.online_agents[0], self.online_agents[1], self.offline_agent] - ), - _quantity=20, - ) - baker.make_recipe( - "winupdate.winupdate_approve", - agent=cycle( - [self.online_agents[0], self.online_agents[1], self.offline_agent] - ), - _quantity=3, - ) - - # run task synchronously - auto_approve_updates_task() - - # make sure the check_for_updates_task was run once for each online agent - self.assertEqual(nats_cmd.call_count, 2) - - # check if all of the created updates were approved - winupdates = WinUpdate.objects.all() - for update in winupdates: - self.assertEqual(update.action, "approve") + # @patch("agents.models.Agent.nats_cmd") + # @patch("time.sleep") + # def test_auto_approve_task(self, mock_sleep, nats_cmd): + # from .tasks import auto_approve_updates_task + + # # Setup data + # baker.make_recipe( + # "winupdate.winupdate", + # agent=cycle( + # [self.online_agents[0], self.online_agents[1], self.offline_agent] + # ), + # _quantity=20, + # ) + # baker.make_recipe( + # "winupdate.winupdate_approve", + # agent=cycle( + # [self.online_agents[0], self.online_agents[1], self.offline_agent] + # ), + # _quantity=3, + # ) + + # # run task synchronously + # auto_approve_updates_task() + + # # make sure the check_for_updates_task was run once for each online agent + # self.assertEqual(nats_cmd.call_count, 2) + + # # check if all of the created updates were approved + # winupdates = WinUpdate.objects.all() + # for update in winupdates: + # self.assertEqual(update.action, "approve")