p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Use fixtures in test_process_runner.py

This commit is contained in:
Shreya 2021-05-26 18:53:47 +05:30 committed by VakarisZ
parent 5f7e886310
commit 58745a0eb4
1 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
monkey/tests/unit_tests/monkey_island/cc/setup/test_process_runner.py
View File

@ -1,22 +1,26 @@
import os import os
import pytest
from monkey_island.cc.setup.mongo_process_runner import MongoDbRunner from monkey_island.cc.setup.mongo_process_runner import MongoDbRunner
TEST_DIR_NAME = "test_dir"
def test_create_db_dir(monkeypatch, tmpdir):
test_dir_name = "test_dir"
monkeypatch.setattr("monkey_island.cc.setup.mongo_process_runner.DB_DIR_NAME", test_dir_name)
expected_path = os.path.join(tmpdir, test_dir_name)
@pytest.fixture
def expected_path(monkeypatch, tmpdir):
monkeypatch.setattr("monkey_island.cc.setup.mongo_process_runner.DB_DIR_NAME", TEST_DIR_NAME)
expected_path = os.path.join(tmpdir, TEST_DIR_NAME)
return expected_path
def test_create_db_dir(monkeypatch, tmpdir, expected_path):
db_path = MongoDbRunner(tmpdir, tmpdir)._create_db_dir() db_path = MongoDbRunner(tmpdir, tmpdir)._create_db_dir()
assert os.path.isdir(expected_path) assert os.path.isdir(expected_path)
assert db_path == expected_path assert db_path == expected_path
def test_create_db_dir__already_created(monkeypatch, tmpdir): def test_create_db_dir__already_created(monkeypatch, tmpdir, expected_path):
test_dir_name = "test_dir"
monkeypatch.setattr("monkey_island.cc.setup.mongo_process_runner.DB_DIR_NAME", test_dir_name)
expected_path = os.path.join(tmpdir, test_dir_name)
os.mkdir(expected_path) os.mkdir(expected_path)
db_path = MongoDbRunner(tmpdir, tmpdir)._create_db_dir() db_path = MongoDbRunner(tmpdir, tmpdir)._create_db_dir()