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

island: move set_home_env() to conftest.py so it can be reused

This commit is contained in:
Mike Salvatore 2021-06-07 13:17:22 -04:00
parent 0519153aaf
commit 8744011297
2 changed files with 21 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
monkey/tests/unit_tests/monkey_island/cc/setup/test_island_config_options.py
View File

@ -32,28 +32,24 @@ def test_data_dir_uses_default():
assert_data_dir_equals(TEST_CONFIG_FILE_CONTENTS_UNSPECIFIED, DEFAULT_DATA_DIR)
def test_data_dir_expanduser(monkeypatch, tmpdir):
set_home_env(monkeypatch, tmpdir)
def test_data_dir_expanduser(patched_home_env):
DATA_DIR_NAME = "test_data_dir"
assert_data_dir_equals(
{"data_dir": os.path.join("~", DATA_DIR_NAME)}, os.path.join(tmpdir, DATA_DIR_NAME)
{"data_dir": os.path.join("~", DATA_DIR_NAME)},
os.path.join(patched_home_env, DATA_DIR_NAME),
)
def test_data_dir_expandvars(monkeypatch, tmpdir):
set_home_env(monkeypatch, tmpdir)
def test_data_dir_expandvars(patched_home_env):
DATA_DIR_NAME = "test_data_dir"
assert_data_dir_equals(
{"data_dir": os.path.join("$HOME", DATA_DIR_NAME)}, os.path.join(tmpdir, DATA_DIR_NAME)
{"data_dir": os.path.join("$HOME", DATA_DIR_NAME)},
os.path.join(patched_home_env, DATA_DIR_NAME),
)
def set_home_env(monkeypatch, tmpdir):
monkeypatch.setenv("HOME", str(tmpdir))
def assert_data_dir_equals(config_file_contents, expected_data_dir):
assert_island_config_option_equals(config_file_contents, "data_dir", expected_data_dir)
@ -85,23 +81,21 @@ def test_crt_path_specified():
)
def test_crt_path_expanduser(monkeypatch, tmpdir):
set_home_env(monkeypatch, tmpdir)
def test_crt_path_expanduser(patched_home_env):
FILE_NAME = "test.crt"
assert_ssl_certificate_file_equals(
{"ssl_certificate": {"ssl_certificate_file": os.path.join("~", FILE_NAME)}},
os.path.join(tmpdir, FILE_NAME),
os.path.join(patched_home_env, FILE_NAME),
)
def test_crt_path_expandvars(monkeypatch, tmpdir):
set_home_env(monkeypatch, tmpdir)
def test_crt_path_expandvars(patched_home_env):
FILE_NAME = "test.crt"
assert_ssl_certificate_file_equals(
{"ssl_certificate": {"ssl_certificate_file": os.path.join("$HOME", FILE_NAME)}},
os.path.join(tmpdir, FILE_NAME),
os.path.join(patched_home_env, FILE_NAME),
)
@ -122,23 +116,21 @@ def test_key_path_specified():
)
def test_key_path_expanduser(monkeypatch, tmpdir):
set_home_env(monkeypatch, tmpdir)
def test_key_path_expanduser(patched_home_env):
FILE_NAME = "test.key"
assert_ssl_certificate_key_file_equals(
{"ssl_certificate": {"ssl_certificate_key_file": os.path.join("~", FILE_NAME)}},
os.path.join(tmpdir, FILE_NAME),
os.path.join(patched_home_env, FILE_NAME),
)
def test_key_path_expandvars(monkeypatch, tmpdir):
set_home_env(monkeypatch, tmpdir)
def test_key_path_expandvars(patched_home_env):
FILE_NAME = "test.key"
assert_ssl_certificate_key_file_equals(
{"ssl_certificate": {"ssl_certificate_key_file": os.path.join("$HOME", FILE_NAME)}},
os.path.join(tmpdir, FILE_NAME),
os.path.join(patched_home_env, FILE_NAME),
)

7
monkey/tests/unit_tests/monkey_island/conftest.py
View File

@ -6,3 +6,10 @@ import pytest
@pytest.fixture(scope="module")
def server_configs_dir(data_for_tests_dir):
return os.path.join(data_for_tests_dir, "server_configs")
@pytest.fixture
def patched_home_env(monkeypatch, tmpdir):
monkeypatch.setenv("HOME", str(tmpdir))
return tmpdir