Add Yonyou BIP automation framework with authentication

Features:
- Add auth module (utils/auth.py) for Yonyou BIP login/logout
- Add environment configuration support with python-dotenv
- Add Playwright browser automation with customizable path
- Add comprehensive test suite (tests/) with login/logout tests
- Add dependency management (requirements.txt)
- Add environment variable template (.env.example)

Key capabilities:
- Login with username/password from environment variables
- Auto-handle force login popup
- Logout with confirmation dialog handling
- Configurable headless mode and HTTPS error handling
- Auto-close browser option

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Misaka_Company
2026-03-27 11:11:15 +08:00
parent 492c98d4cc
commit 90e38821a5
6 changed files with 311 additions and 0 deletions

3
tests/__init__.py Normal file
View File

@@ -0,0 +1,3 @@
"""
测试模块
"""

44
tests/test_auth_config.py Normal file
View File

@@ -0,0 +1,44 @@
"""
Test if auth module configuration is correct
"""
from playwright.sync_api import sync_playwright
from dotenv import load_dotenv
from pathlib import Path
import os
# Load environment variables
PROJECT_ROOT = Path(__file__).resolve().parent.parent
load_dotenv(PROJECT_ROOT / ".env")
# Configure browser path
os.environ["PLAYWRIGHT_BROWSERS_PATH"] = os.getenv("PLAYWRIGHT_BROWSERS_PATH")
print("=" * 50)
print("ERP System Configuration Check")
print("=" * 50)
# Display configuration information
print(f"Browser Path: {os.getenv('PLAYWRIGHT_BROWSERS_PATH')}")
print(f"ERP URL: {os.getenv('ERP_URL')}")
print(f"Username: {os.getenv('ERP_USERNAME')}")
print(f"Password: {'*' * len(os.getenv('ERP_PASSWORD', ''))}")
print(f"Headless Mode: {os.getenv('ERP_HEADLESS')}")
print(f"Ignore HTTPS Errors: {os.getenv('ERP_IGNORE_HTTPS_ERRORS')}")
print(f"Auto Close Browser: {os.getenv('ERP_AUTO_CLOSE_BROWSER')}")
print("\n" + "=" * 50)
print("Check Playwright Browser")
print("=" * 50)
with sync_playwright() as p:
chromium_path = p.chromium.executable_path
print(f"Chromium Path: {chromium_path}")
# Check if browser file exists
if os.path.exists(chromium_path):
print("[OK] Browser file exists")
else:
print("[ERROR] Browser file not found")
print("\n[OK] Configuration check completed!")

79
tests/test_login.py Normal file
View File

@@ -0,0 +1,79 @@
"""
Test Yonyou BIP system login functionality
"""
import sys
from pathlib import Path
# Add project root to Python path
PROJECT_ROOT = Path(__file__).resolve().parent.parent
sys.path.insert(0, str(PROJECT_ROOT))
from playwright.sync_api import sync_playwright
from utils.auth import login, logout, close_session
import os
# Load environment variables
from dotenv import load_dotenv
load_dotenv(PROJECT_ROOT / ".env")
# Configure browser path
os.environ["PLAYWRIGHT_BROWSERS_PATH"] = os.getenv("PLAYWRIGHT_BROWSERS_PATH")
print("=" * 60)
print("Testing Yonyou BIP Login and Logout")
print("=" * 60)
# Display configuration information
print(f"\nConfiguration:")
print(f" URL: {os.getenv('ERP_URL')}")
print(f" Username: {os.getenv('ERP_USERNAME')}")
print(f" Headless: {os.getenv('ERP_HEADLESS')}")
print(f" Ignore HTTPS Errors: {os.getenv('ERP_IGNORE_HTTPS_ERRORS')}")
try:
with sync_playwright() as p:
print("\n[1/5] Starting browser...")
browser, context, page, main_frame = login(
playwright=p,
verbose=True
)
print("\n[2/5] Login successful!")
# Wait a moment to observe the page after login
print("\n[3/5] Waiting 3 seconds to observe the page after login...")
import time
time.sleep(3)
# Test logout
print("\n[4/5] Testing logout...")
logout(main_frame, verbose=True)
print("\n[5/5] Logout successful!")
# Wait a moment to observe the page after logout
print("\nWaiting 2 seconds to observe the page after logout...")
time.sleep(2)
print("\n[SUCCESS] Login and logout test completed successfully!")
# Check if auto-close browser
auto_close = os.getenv("ERP_AUTO_CLOSE_BROWSER", "true").lower() in ("true", "1", "yes")
if not auto_close:
print("\n[INFO] Browser will stay open for 10 seconds for manual observation...")
time.sleep(10)
# Close browser
print("\n[CLEANUP] Closing browser session...")
close_session(browser, context)
except Exception as e:
print(f"\n[ERROR] Login/logout test failed!")
print(f"Error: {e}")
import traceback
traceback.print_exc()
print("\n" + "=" * 60)
print("Test completed")
print("=" * 60)