add some tests

This commit is contained in:
sadnub
2020-11-11 11:21:02 -05:00
parent 96c46a9e12
commit 3fabae5b5f
2 changed files with 11 additions and 2 deletions

View File

@@ -415,7 +415,7 @@ class NewAgent(APIView):
""" Creates the agent """
if User.objects.filter(username=request.data["agent_id"]).exists():
if Agent.objects.filter(agent_id=request.data["agent_id"]).exists():
return notify_error(
"Agent already exists. Remove old agent first if trying to re-install"
)

View File

@@ -11,6 +11,10 @@ class TestAuditViews(TacticalTestCase):
self.setup_coresettings()
def create_audit_records(self):
# create clients for client filter
site = baker.make("clients.Site")
baker.make_recipe("agents.agent", site=site, hostname="AgentHostname1")
# user jim agent logs
baker.make_recipe(
"logs.agent_logs",
@@ -75,11 +79,13 @@ class TestAuditViews(TacticalTestCase):
_quantity=13,
)
return site
def test_get_audit_logs(self):
url = "/logs/auditlogs/"
# create data
self.create_audit_records()
site = self.create_audit_records()
# test data and result counts
data = [
@@ -111,6 +117,9 @@ class TestAuditViews(TacticalTestCase):
"count": 40,
},
{"filter": {"timeFilter": 35, "userFilter": ["james", "jim"]}, "count": 81},
{"filter": {"objectFilter": ["user"]}, "count": 26},
{"filter": {"actionFilter": ["login"]}, "count": 12},
{"filter": {"clientFilter": [site.client.id]}, "count": 23},
]
for req in data: