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

Use stub for PBA

This commit is contained in:
Shreya 2021-02-19 00:44:28 +05:30
parent 8bd30ceb4c
commit 2bc27b48de
1 changed files with 9 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
monkey/infection_monkey/telemetry/tests/test_post_breach_telem.py
View File

@ -1,31 +1,24 @@
import pytest import pytest
from common.data.post_breach_consts import POST_BREACH_JOB_SCHEDULING
from infection_monkey.post_breach.actions.schedule_jobs import ScheduleJobs
from infection_monkey.post_breach.job_scheduling.linux_job_scheduling import (
get_linux_commands_to_schedule_jobs,
)
from infection_monkey.post_breach.job_scheduling.windows_job_scheduling import (
get_windows_commands_to_schedule_jobs,
)
from infection_monkey.telemetry.post_breach_telem import PostBreachTelem from infection_monkey.telemetry.post_breach_telem import PostBreachTelem
from infection_monkey.utils.environment import is_windows_os
HOSTNAME = "hostname" HOSTNAME = "hostname"
IP = "0.0.0.0" IP = "0.0.0.0"
PBA = ScheduleJobs() PBA_COMMAND = "run some pba"
PBA_COMMAND = ( PBA_NAME = "some pba"
get_windows_commands_to_schedule_jobs()
if is_windows_os()
else " ".join(get_linux_commands_to_schedule_jobs())
)
PBA_NAME = POST_BREACH_JOB_SCHEDULING
RESULT = False RESULT = False
class StubSomePBA:
def __init__(self):
self.name = PBA_NAME
self.command = PBA_COMMAND
@pytest.fixture @pytest.fixture
def post_breach_telem_test_instance(monkeypatch): def post_breach_telem_test_instance(monkeypatch):
PBA = StubSomePBA()
monkeypatch.setattr(PostBreachTelem, "_get_hostname_and_ip", lambda: (HOSTNAME, IP)) monkeypatch.setattr(PostBreachTelem, "_get_hostname_and_ip", lambda: (HOSTNAME, IP))
return PostBreachTelem(PBA, RESULT) return PostBreachTelem(PBA, RESULT)