p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

island: Add unit test to verify '~' expanded in data_dir

This commit is contained in:
Mike Salvatore 2021-06-01 03:33:34 -04:00
parent f3fccb3e06
commit 0ab20d558b
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
monkey/tests/unit_tests/monkey_island/setup/test_island_config_options.py
View File

@ -1,3 +1,5 @@
import os
from monkey_island.cc.server_utils.consts import ( from monkey_island.cc.server_utils.consts import (
DEFAULT_DATA_DIR, DEFAULT_DATA_DIR,
DEFAULT_LOG_LEVEL, DEFAULT_LOG_LEVEL,
@ -37,3 +39,16 @@ def test_island_config_options__mongodb():
assert options.start_mongodb == DEFAULT_START_MONGO_DB assert options.start_mongodb == DEFAULT_START_MONGO_DB
options = IslandConfigOptions(TEST_CONFIG_FILE_CONTENTS_NO_STARTMONGO) options = IslandConfigOptions(TEST_CONFIG_FILE_CONTENTS_NO_STARTMONGO)
assert options.start_mongodb == DEFAULT_START_MONGO_DB assert options.start_mongodb == DEFAULT_START_MONGO_DB
def set_home_env(monkeypatch, tmpdir):
monkeypatch.setenv("HOME", str(tmpdir))
def test_island_config_options__data_dir_expanduser(monkeypatch, tmpdir):
set_home_env(monkeypatch, tmpdir)
DATA_DIR_NAME = "test_data_dir"
options = IslandConfigOptions({"data_dir": f"~/{DATA_DIR_NAME}"})
assert options.data_dir == os.path.join(tmpdir, DATA_DIR_NAME)