p59342861
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

agent: Refactor add_subdirs_to_dir in test_dir_utils.py

This commit is contained in:
Mike Salvatore 2021-06-22 14:07:31 -04:00
parent b643cd1edd
commit 30f88ca319
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/tests/unit_tests/infection_monkey/utils/test_dir_utils.py
View File

@ -5,14 +5,11 @@ from infection_monkey.utils.dir_utils import (
) )
FILES = ["file.jpg.zip", "file.xyz", "1.tar", "2.tgz"] FILES = ["file.jpg.zip", "file.xyz", "1.tar", "2.tgz"]
SUBDIR_1 = "subdir1" SUBDIRS = ["subdir1", "subdir2"]
SUBDIR_2 = "subdir2"
def add_subdirs_to_dir(parent_dir): def add_subdirs_to_dir(parent_dir):
subdir1 = parent_dir / SUBDIR_1 subdirs = [parent_dir / s for s in SUBDIRS]
subdir2 = parent_dir / SUBDIR_2
subdirs = [subdir1, subdir2]
for subdir in subdirs: for subdir in subdirs:
subdir.mkdir() subdir.mkdir()