fix tests

This commit is contained in:
sadnub 2021-02-09 19:27:37 -05:00
parent 7421138533
commit 9bb33d2afc
1 changed files with 2 additions and 2 deletions

View File

@ -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,
)