Skip to content

Commit c45c84a

Browse files
committed
updated unittest
1 parent 08b2fc0 commit c45c84a

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/api_app/pivots_manager/test_models.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from django.core.exceptions import ValidationError
22
from django.db import transaction
33

4+
from api_app.analyzers_manager.constants import AllTypes
45
from api_app.analyzers_manager.models import AnalyzerConfig
56
from api_app.connectors_manager.models import ConnectorConfig
67
from api_app.models import Job, PythonModule
@@ -95,9 +96,9 @@ def test_create_job_multiple_file(self):
9596
python_module=PythonModule.objects.filter(
9697
base_path="api_app.pivots_manager.pivots"
9798
).first(),
98-
playbook_to_execute=PlaybookConfig.objects.get(
99-
name="Sample_Static_Analysis"
100-
),
99+
playbook_to_execute=PlaybookConfig.objects.filter(
100+
disabled=False, type__icontains=AllTypes.FILE.value
101+
).first(),
101102
)
102103
with open("test_files/file.exe", "rb") as f:
103104
content = f.read()

0 commit comments

Comments
 (0)