| import os | |
| import re | |
| def align_test_paths(file_path): | |
| with open(file_path, 'r', encoding='utf-8') as f: | |
| content = f.read() | |
| # Rule: Point patches to the actual sub-module instead of the facade | |
| new_content = re.sub(r'patch\(["\']server\.api_routes\.knowledge_api\.RBACService["\']', | |
| 'patch(\"server.api_routes.knowledge.items.RBACService\"', content) | |
| new_content = re.sub(r'patch\(["\']server\.api_routes\.knowledge_api\.extract_text_from_document["\']', | |
| 'patch(\"server.api_routes.knowledge.upload.extract_text_from_document\"', new_content) | |
| new_content = re.sub(r'patch\(["\']server\.api_routes\.knowledge_api\.DocumentStorageService["\']', | |
| 'patch(\"server.api_routes.knowledge.upload.DocumentStorageService\"', new_content) | |
| if content != new_content: | |
| with open(file_path, 'w', encoding='utf-8') as f: | |
| f.write(new_content) | |
| return True | |
| return False | |
| targets = [ | |
| 'python/tests/server/api_routes/test_knowledge_api_security.py', | |
| 'python/tests/test_file_upload_integration.py' | |
| ] | |
| for t in targets: | |
| if os.path.exists(t): | |
| if align_test_paths(t): | |
| print(f"Aligned {t} to real paths.") | |