Cobra
1eb35c9050
- Replace BigBrother -> SystemMonitor in display names and docstrings - Replace logger names: bb.* -> sensor.* - Replace process names: bb-* -> sensor-* - Replace home directory: ~/.bigbrother -> ~/.implant - Replace LUKS device: /dev/mapper/bb-* -> /dev/mapper/sensor-* - Updated 76 Python files across all modules - Improves OPSEC by removing obvious tool fingerprints from logs and runtime
118 lines
3.0 KiB
Python
118 lines
3.0 KiB
Python
"""Shared pytest fixtures for SystemMonitor tests."""
|
|
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Ensure project root is on sys.path
|
|
# ---------------------------------------------------------------------------
|
|
|
|
PROJECT_ROOT = str(Path(__file__).resolve().parent.parent)
|
|
if PROJECT_ROOT not in sys.path:
|
|
sys.path.insert(0, PROJECT_ROOT)
|
|
|
|
|
|
@pytest.fixture
|
|
def project_root():
|
|
"""Return the project root directory and ensure it is on sys.path."""
|
|
return PROJECT_ROOT
|
|
|
|
|
|
@pytest.fixture
|
|
def tmp_dir(tmp_path):
|
|
"""Provide a temporary directory for test artifacts."""
|
|
return str(tmp_path)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config():
|
|
"""Return a valid SystemMonitor config dict (no YAML file needed)."""
|
|
return {
|
|
"device": {
|
|
"platform": "generic",
|
|
"hostname": "test-device",
|
|
"install_path": "/tmp/bb-test",
|
|
},
|
|
"network": {
|
|
"primary_interface": "eth0",
|
|
"scope_enforcement": False,
|
|
},
|
|
"connectivity": {
|
|
"primary": "wireguard",
|
|
},
|
|
"security": {
|
|
"encryption": "aes-256-gcm",
|
|
"encryption_key_derive": "argon2id",
|
|
"kill_switch_enabled": True,
|
|
},
|
|
"capture": {
|
|
"interface": "eth0",
|
|
"snap_length": 65535,
|
|
"pcap_rotation_hours": 1,
|
|
"pcap_compression": "zstd",
|
|
"pcap_compression_level": 19,
|
|
},
|
|
"stealth": {
|
|
"mac_profile": "auto",
|
|
"process_disguise": True,
|
|
"log_suppression": True,
|
|
},
|
|
"engagement_phase": {
|
|
"mode": "passive_only",
|
|
"passive_days": 7,
|
|
},
|
|
"bettercap": {
|
|
"binary": "/usr/local/bin/bettercap",
|
|
"api_address": "127.0.0.1",
|
|
"api_port": 8083,
|
|
},
|
|
"storage_path": "/tmp/bb-test/storage",
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_bus():
|
|
"""Return a started EventBus instance. Stopped after test."""
|
|
from core.bus import EventBus
|
|
|
|
bus = EventBus()
|
|
bus.start()
|
|
yield bus
|
|
bus.stop()
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_state(tmp_path):
|
|
"""Return a StateManager backed by a temp SQLite database. Stopped after test."""
|
|
from core.state import StateManager
|
|
|
|
db_path = str(tmp_path / "test_state.db")
|
|
state = StateManager(db_path=db_path)
|
|
state.start()
|
|
yield state
|
|
state.stop()
|
|
|
|
|
|
@pytest.fixture
|
|
def hardware_tier_override():
|
|
"""Context manager fixture to patch detect_hardware_tier to return a specific tier.
|
|
|
|
Usage:
|
|
def test_foo(hardware_tier_override):
|
|
with hardware_tier_override("pi_zero"):
|
|
...
|
|
"""
|
|
from contextlib import contextmanager
|
|
|
|
@contextmanager
|
|
def _override(tier: str):
|
|
with patch("core.engine.detect_hardware_tier", return_value=tier):
|
|
yield
|
|
|
|
return _override
|