conftest.py

import os
import sys
import tempfile
import shutil
from pathlib import Path
from unittest.mock import Mock, patch
import pytest

# Add the project root to the Python path
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))


@pytest.fixture
def temp_dir():
    """Create a temporary directory for test files."""
    temp_path = tempfile.mkdtemp()
    yield Path(temp_path)
    shutil.rmtree(temp_path)


@pytest.fixture
def mock_config():
    """Provide a mock configuration object."""
    config = Mock()
    config.debug = False
    config.verbose = False
    config.timeout = 30
    config.max_retries = 3
    return config


@pytest.fixture
def sample_data():
    """Provide sample test data."""
    return {
        'test_string': 'Hello, World!',
        'test_bytes': b'Hello, World!',
        'test_list': [1, 2, 3, 4, 5],
        'test_dict': {'key1': 'value1', 'key2': 'value2'},
        'test_int': 42,
        'test_float': 3.14159
    }


@pytest.fixture
def mock_file_system(temp_dir):
    """Create a mock file system structure."""
    # Create directories
    (temp_dir / 'modules').mkdir()
    (temp_dir / 'core').mkdir()
    (temp_dir / 'utils').mkdir()
    
    # Create some mock files
    (temp_dir / 'test_file.txt').write_text('Test content')
    (temp_dir / 'modules' / 'module1.py').write_text('# Module 1')
    (temp_dir / 'core' / 'core1.py').write_text('# Core 1')
    
    return temp_dir


@pytest.fixture
def mock_network():
    """Mock network-related functions."""
    with patch('socket.socket') as mock_socket:
        mock_instance = Mock()
        mock_socket.return_value = mock_instance
        mock_instance.connect.return_value = None
        mock_instance.send.return_value = 10
        mock_instance.recv.return_value = b'response'
        yield mock_instance


@pytest.fixture
def mock_subprocess():
    """Mock subprocess calls."""
    with patch('subprocess.run') as mock_run:
        mock_run.return_value = Mock(
            stdout='Success',
            stderr='',
            returncode=0
        )
        yield mock_run


@pytest.fixture(autouse=True)
def reset_environment():
    """Reset environment variables before each test."""
    original_env = os.environ.copy()
    yield
    os.environ.clear()
    os.environ.update(original_env)


@pytest.fixture
def capture_logs():
    """Capture log messages during tests."""
    import logging
    from io import StringIO
    
    log_capture = StringIO()
    handler = logging.StreamHandler(log_capture)
    handler.setLevel(logging.DEBUG)
    
    logger = logging.getLogger()
    logger.addHandler(handler)
    logger.setLevel(logging.DEBUG)
    
    yield log_capture
    
    logger.removeHandler(handler)


@pytest.fixture
def mock_crypto():
    """Mock cryptographic operations."""
    with patch('Crypto.Cipher.AES.new') as mock_aes:
        mock_cipher = Mock()
        mock_aes.return_value = mock_cipher
        mock_cipher.encrypt.return_value = b'encrypted_data'
        mock_cipher.decrypt.return_value = b'decrypted_data'
        yield mock_cipher


@pytest.fixture(scope="session")
def test_resources():
    """Provide paths to test resource files."""
    resources_dir = Path(__file__).parent / 'resources'
    return {
        'base_dir': resources_dir,
        'sample_dll': resources_dir / 'sample.dll',
        'sample_exe': resources_dir / 'sample.exe',
        'sample_config': resources_dir / 'config.json'
    }


# Markers for different test types
def pytest_configure(config):
    """Configure pytest with custom markers."""
    config.addinivalue_line(
        "markers", "unit: mark test as a unit test"
    )
    config.addinivalue_line(
        "markers", "integration: mark test as an integration test"
    )
    config.addinivalue_line(
        "markers", "slow: mark test as slow running"
    )