File size: 1,789 Bytes
a402b9b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 | """Shared fixtures for source_patcher tests.
The sample module is defined as an inline string and written to a temp file
at test time, avoiding CI complaints about fixture files without test registration.
"""
import importlib.util
import sys
import tempfile
from pathlib import Path
from types import ModuleType
import pytest
SAMPLE_MODULE_NAME = "_source_patcher_test_fixtures.sample_module"
SAMPLE_MODULE_SOURCE = '''\
GLOBAL_VAR = "global_value"
class HelperClass:
"""Utility class referenced by SampleClass to test cross-class calls."""
@staticmethod
def format_value(value: str) -> str:
return f"[{value}]"
class SampleClass:
def greet(self, name: str) -> str:
greeting = f"hello {name}"
return greeting
def compute(self, x: int) -> int:
result = x * 2 + 1
return result
def uses_global(self) -> str:
return f"value={GLOBAL_VAR}"
def uses_helper(self, value: str) -> str:
return HelperClass.format_value(value)
def standalone_function(a: int, b: int) -> int:
return a + b
'''
@pytest.fixture(scope="session")
def sample_module() -> ModuleType:
"""Load the sample module from a temp file and register it in sys.modules."""
if SAMPLE_MODULE_NAME in sys.modules:
return sys.modules[SAMPLE_MODULE_NAME]
tmpdir = tempfile.mkdtemp(prefix="source_patcher_fixtures_")
module_path = Path(tmpdir) / "sample_module.py"
module_path.write_text(SAMPLE_MODULE_SOURCE)
spec = importlib.util.spec_from_file_location(SAMPLE_MODULE_NAME, module_path)
assert spec is not None
assert spec.loader is not None
module = importlib.util.module_from_spec(spec)
sys.modules[SAMPLE_MODULE_NAME] = module
spec.loader.exec_module(module)
return module
|