From 297adcf015d4a04fcd4fa51cc5f6196558eb7c2c Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Wed, 23 Jun 2021 07:10:55 -0400 Subject: [PATCH] agent: Don't redefine EXTENSION in ransomware tests --- .../infection_monkey/ransomware/test_ransomware_payload.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/monkey/tests/unit_tests/infection_monkey/ransomware/test_ransomware_payload.py b/monkey/tests/unit_tests/infection_monkey/ransomware/test_ransomware_payload.py index 4cc3e8c96..9108bace5 100644 --- a/monkey/tests/unit_tests/infection_monkey/ransomware/test_ransomware_payload.py +++ b/monkey/tests/unit_tests/infection_monkey/ransomware/test_ransomware_payload.py @@ -6,9 +6,7 @@ from pathlib import Path import pytest from tests.utils import is_user_admin -from infection_monkey.ransomware.ransomware_payload import RansomewarePayload - -EXTENSION = ".m0nk3y" +from infection_monkey.ransomware.ransomware_payload import EXTENSION, RansomewarePayload SUBDIR = "subdir" ALL_ZEROS_PDF = "all_zeros.pdf"