tests: Modify unit tests for ransomware utils to test for a file like file.jpg.zip

This commit is contained in:
Shreya 2021-06-22 19:03:54 +05:30
parent 8abb9d95ce
commit 48967a2e5b
1 changed files with 2 additions and 2 deletions

View File

@ -3,8 +3,8 @@ import os
import infection_monkey.ransomware.utils import infection_monkey.ransomware.utils
VALID_FILE_EXTENSION_1 = "file.3ds" VALID_FILE_EXTENSION_1 = "file.3ds"
VALID_FILE_EXTENSION_2 = "file.zip" VALID_FILE_EXTENSION_2 = "file.jpg.zip"
INVALID_FILE_EXTENSION_1 = "file.jpe" INVALID_FILE_EXTENSION_1 = "file.pqr"
INVALID_FILE_EXTENSION_2 = "file.xyz" INVALID_FILE_EXTENSION_2 = "file.xyz"
SUBDIR_1 = "subdir1" SUBDIR_1 = "subdir1"
SUBDIR_2 = "subdir2" SUBDIR_2 = "subdir2"