mirror of
https://github.com/SpecterOps/Nemesis
synced 2026-06-08 12:36:42 +00:00
Lee Chagolla-Christensen
3b4f110686
* make all enrichment (should)process methods async * fix office2john resource leak * check for none * more resource errors, add ole hash tests, pytest configs * update chromium to use async pool * update registry module to use asyncpg pool * more asyncpg * use async get_file_enriched * update sleep * fix dapr configs, new ones for workflow services * remove deprecate/redundant httprequestmax setting * disable workflow tracking (for now) --------- Co-authored-by: Lee Chagolla-Christensen <lee@localhost>
54 lines
1.1 KiB
Python
54 lines
1.1 KiB
Python
import os
|
|
import warnings
|
|
from collections.abc import Callable
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
# Ensure dev mode is enabled (catches many other bugs)
|
|
os.environ["PYTHONDEVMODE"] = "1"
|
|
|
|
|
|
# Configure warnings
|
|
def pytest_configure(config):
|
|
# At a minimum, we should always report resource warnings
|
|
# warnings.simplefilter("error", ResourceWarning)
|
|
warnings.simplefilter("error")
|
|
|
|
|
|
def _get_file_path(name: str) -> Path:
|
|
return Path(__file__).parent / "fixtures" / name
|
|
|
|
|
|
def _read_file_text(name: str) -> str:
|
|
return _get_file_path(name).read_text()
|
|
|
|
|
|
def _read_file_bytes(name: str) -> bytes:
|
|
return _get_file_path(name).read_bytes()
|
|
|
|
|
|
@pytest.fixture
|
|
def read_file_text() -> Callable[[str], str]:
|
|
return _read_file_text
|
|
|
|
|
|
@pytest.fixture
|
|
def read_file_bytes() -> Callable[[str], bytes]:
|
|
return _read_file_bytes
|
|
|
|
|
|
@pytest.fixture
|
|
def get_file_path() -> Callable[[str], Path]:
|
|
return _get_file_path
|
|
|
|
|
|
@pytest.fixture
|
|
def blob_without_entropy() -> bytes:
|
|
return _read_file_bytes("blob_without_entropy.bin")
|
|
|
|
|
|
@pytest.fixture
|
|
def blob_with_entropy() -> bytes:
|
|
return _read_file_bytes("blob_with_entropy.bin")
|