move tests

This commit is contained in:
wh1te909 2022-05-10 16:29:09 +00:00
parent 1505fa547e
commit d0c6e3a158
2 changed files with 5 additions and 5 deletions

View File

View File

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