From 9bb33d2afc371b02de839f299ee81e191b9361c9 Mon Sep 17 00:00:00 2001 From: sadnub Date: Tue, 9 Feb 2021 19:27:37 -0500 Subject: [PATCH] fix tests --- api/tacticalrmm/automation/tests.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/tacticalrmm/automation/tests.py b/api/tacticalrmm/automation/tests.py index 4a00dad5..3080cbb4 100644 --- a/api/tacticalrmm/automation/tests.py +++ b/api/tacticalrmm/automation/tests.py @@ -982,7 +982,7 @@ class TestPolicyTasks(TacticalTestCase): # should trigger task in save method on core generate_agent_checks_by_location_task.assert_called_with( - location={"site__client_id": server_agent.site.pk}, + location={"site__client_id": server_agent.client.pk}, mon_type="server", create_tasks=True, ) @@ -1004,7 +1004,7 @@ class TestPolicyTasks(TacticalTestCase): # should trigger task in save method on core generate_agent_checks_by_location_task.assert_called_with( - location={"site__client_id": server_agent.site.pk}, + location={"site__client_id": server_agent.client.pk}, mon_type="server", create_tasks=True, )