From 48967a2e5b4ce1188845f86d7c2ac58a6d4c878a Mon Sep 17 00:00:00 2001
From: Shreya <shreya.malviya@gmail.com>
Date: Tue, 22 Jun 2021 19:03:54 +0530
Subject: [PATCH] tests: Modify unit tests for ransomware utils to test for a
 file like file.jpg.zip

---
 .../unit_tests/infection_monkey/ransomware/test_utils.py      | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/monkey/tests/unit_tests/infection_monkey/ransomware/test_utils.py b/monkey/tests/unit_tests/infection_monkey/ransomware/test_utils.py
index 579b7e2c9..f3c0bea39 100644
--- a/monkey/tests/unit_tests/infection_monkey/ransomware/test_utils.py
+++ b/monkey/tests/unit_tests/infection_monkey/ransomware/test_utils.py
@@ -3,8 +3,8 @@ import os
 import infection_monkey.ransomware.utils
 
 VALID_FILE_EXTENSION_1 = "file.3ds"
-VALID_FILE_EXTENSION_2 = "file.zip"
-INVALID_FILE_EXTENSION_1 = "file.jpe"
+VALID_FILE_EXTENSION_2 = "file.jpg.zip"
+INVALID_FILE_EXTENSION_1 = "file.pqr"
 INVALID_FILE_EXTENSION_2 = "file.xyz"
 SUBDIR_1 = "subdir1"
 SUBDIR_2 = "subdir2"