This commit is contained in:
57
tests/conftest.py
Normal file
57
tests/conftest.py
Normal file
@@ -0,0 +1,57 @@
|
||||
import pytest
|
||||
from code_privacy_shield.patterns import PatternLibrary
|
||||
from code_privacy_shield.redactor import Redactor
|
||||
from code_privacy_shield.config import Config
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def pattern_library():
|
||||
return PatternLibrary()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def redactor():
|
||||
return Redactor()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def config():
|
||||
return Config()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def sample_sensitive_code():
|
||||
return '''
|
||||
import os
|
||||
import requests
|
||||
|
||||
API_KEY = "sk-abc123def456ghi789"
|
||||
DATABASE_URL = "postgresql://user:password@localhost:5432/mydb"
|
||||
EMAIL = "developer@company.com"
|
||||
AWS_SECRET = "wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY"
|
||||
|
||||
# Environment variable access
|
||||
db_password = os.environ.get("DB_PASSWORD")
|
||||
|
||||
# Database connection
|
||||
conn_string = "mysql://admin:secretpass@db.example.com:3306/users"
|
||||
|
||||
# Authorization header
|
||||
headers = {"Authorization": "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9"}
|
||||
'''
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def sample_clean_code():
|
||||
return '''
|
||||
def calculate_sum(a, b):
|
||||
"""Calculate sum of two numbers."""
|
||||
return a + b
|
||||
|
||||
class MyClass:
|
||||
def __init__(self, value):
|
||||
self.value = value
|
||||
|
||||
def get_value(self):
|
||||
return self.value
|
||||
'''
|
||||
Reference in New Issue
Block a user