diff --git a/api/tacticalrmm/agents/tests/__init__.py b/api/tacticalrmm/agents/tests/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/api/tacticalrmm/agents/tests.py b/api/tacticalrmm/agents/tests/test_agents.py similarity index 99% rename from api/tacticalrmm/agents/tests.py rename to api/tacticalrmm/agents/tests/test_agents.py index a63f3519..d01a7f11 100644 --- a/api/tacticalrmm/agents/tests.py +++ b/api/tacticalrmm/agents/tests/test_agents.py @@ -17,14 +17,14 @@ from tacticalrmm.test import TacticalTestCase from winupdate.models import WinUpdatePolicy from winupdate.serializers import WinUpdatePolicySerializer -from .models import Agent, AgentCustomField, AgentHistory, Note -from .serializers import ( +from agents.models import Agent, AgentCustomField, AgentHistory, Note +from agents.serializers import ( AgentHistorySerializer, AgentHostnameSerializer, AgentNoteSerializer, AgentSerializer, ) -from .tasks import auto_self_agent_update_task +from agents.tasks import auto_self_agent_update_task if TYPE_CHECKING: from clients.models import Client, Site @@ -583,7 +583,7 @@ class TestAgentViews(TacticalTestCase): def test_run_script(self, run_script, email_task): from clients.models import ClientCustomField, SiteCustomField - from .models import AgentCustomField, AgentHistory, Note + from agents.models import AgentCustomField, AgentHistory, Note run_script.return_value = "ok" url = f"/agents/{self.agent.agent_id}/runscript/" @@ -1552,7 +1552,7 @@ class TestAgentTasks(TacticalTestCase): self.assertEqual(agent_update.call_count, 33) def test_agent_history_prune_task(self): - from .tasks import prune_agent_history + from agents.tasks import prune_agent_history # setup data agent = baker.make_recipe("agents.agent")