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:
3
tests/__init__.py
Normal file
3
tests/__init__.py
Normal file
@@ -0,0 +1,3 @@
|
||||
"""
|
||||
测试模块
|
||||
"""
|
||||
44
tests/test_auth_config.py
Normal file
44
tests/test_auth_config.py
Normal 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
79
tests/test_login.py
Normal 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)
|
||||
Reference in New Issue
Block a user