149 lines
6.8 KiB
Python
149 lines
6.8 KiB
Python
"""Debug script to trace order status sync in logistics flow."""
|
|
import sys, os
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
os.environ['APP_ENV'] = 'test'
|
|
os.environ['MYSQL_DATABASE'] = ':memory:'
|
|
os.environ['SECRET_KEY'] = 'test-secret-key-for-testing-only'
|
|
os.environ['JWT_EXPIRE_MINUTES'] = '60'
|
|
os.environ['AI_PROVIDER'] = 'mock'
|
|
os.environ['AUTH_TOKEN_STORE_PATH'] = 'D:/tmp/order-flow-test/revoked_tokens.json'
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import Session
|
|
from backend.app.db import Base
|
|
from backend.app.core.security import create_access_token, hash_password
|
|
from backend.app.main import app
|
|
from backend.app.api.deps import get_db_session
|
|
from backend.tests.conftest import SyncTestClient
|
|
from backend.app.models.system import Role, User, Menu, RoleMenu
|
|
from backend.app.models.business import Customer, Product, Supplier, SalesOrder, SalesOrderItem
|
|
|
|
engine = create_engine('sqlite:///:memory:', echo=False, connect_args={'check_same_thread': False})
|
|
Base.metadata.create_all(engine)
|
|
connection = engine.connect()
|
|
transaction = connection.begin()
|
|
session = Session(bind=connection)
|
|
|
|
# Seed data
|
|
roles = [
|
|
Role(id=1, role_name='Admin', role_code='admin', status=1),
|
|
Role(id=2, role_name='Sales', role_code='salesman', status=1),
|
|
Role(id=3, role_name='Manager', role_code='manager', status=1),
|
|
Role(id=4, role_name='Driver', role_code='driver', status=1),
|
|
]
|
|
session.add_all(roles)
|
|
session.flush()
|
|
|
|
users = [
|
|
User(id=1, username='admin01', password_hash=hash_password('admin123'), real_name='Admin', mobile='13800000001', role_id=1, status=1),
|
|
User(id=5, username='driver01', password_hash=hash_password('driver123'), real_name='Driver', mobile='13800000005', role_id=4, status=1),
|
|
]
|
|
session.add_all(users)
|
|
session.flush()
|
|
|
|
menus = [
|
|
Menu(id=1, menu_name='Orders', menu_path='/orders', menu_type='page', permission_code='order:list', sort_no=1, status=1),
|
|
Menu(id=8, menu_name='Driver', menu_path='/driver', menu_type='page', permission_code='driver:task:list', sort_no=6, status=1),
|
|
Menu(id=9, menu_name='Accept', menu_path='', menu_type='button', permission_code='driver:task:accept', parent_id=8, sort_no=1, status=1),
|
|
Menu(id=10, menu_name='Pickup', menu_path='', menu_type='button', permission_code='driver:task:pickup', parent_id=8, sort_no=2, status=1),
|
|
Menu(id=11, menu_name='Deliver', menu_path='', menu_type='button', permission_code='driver:task:deliver', parent_id=8, sort_no=3, status=1),
|
|
Menu(id=25, menu_name='TaskList', menu_path='', menu_type='button', permission_code='logistics:task:list', parent_id=8, sort_no=4, status=1),
|
|
Menu(id=26, menu_name='TaskCreate', menu_path='', menu_type='button', permission_code='logistics:task:create', parent_id=8, sort_no=5, status=1),
|
|
Menu(id=27, menu_name='TaskCancel', menu_path='', menu_type='button', permission_code='logistics:task:cancel', parent_id=8, sort_no=6, status=1),
|
|
Menu(id=28, menu_name='TraceList', menu_path='', menu_type='button', permission_code='logistics:trace:list', parent_id=8, sort_no=7, status=1),
|
|
Menu(id=29, menu_name='TraceCreate', menu_path='', menu_type='button', permission_code='logistics:trace:create', parent_id=8, sort_no=8, status=1),
|
|
Menu(id=12, menu_name='Submit', menu_path='', menu_type='button', permission_code='order:submit', parent_id=1, sort_no=3, status=1),
|
|
]
|
|
session.add_all(menus)
|
|
session.flush()
|
|
|
|
role_menus = [
|
|
*[RoleMenu(role_id=1, menu_id=m.id) for m in menus],
|
|
RoleMenu(role_id=4, menu_id=8), RoleMenu(role_id=4, menu_id=9),
|
|
RoleMenu(role_id=4, menu_id=10), RoleMenu(role_id=4, menu_id=11),
|
|
RoleMenu(role_id=4, menu_id=28), RoleMenu(role_id=4, menu_id=29),
|
|
]
|
|
session.add_all(role_menus)
|
|
session.flush()
|
|
|
|
def override_get_db():
|
|
try:
|
|
yield session
|
|
finally:
|
|
pass
|
|
|
|
app.dependency_overrides[get_db_session] = override_get_db
|
|
|
|
admin_token = create_access_token({'user_id': 1, 'role_code': 'admin', 'username': 'admin01'})
|
|
driver_token = create_access_token({'user_id': 5, 'role_code': 'driver', 'username': 'driver01'})
|
|
|
|
with SyncTestClient(app, raise_server_exceptions=False) as c:
|
|
admin_h = {'Authorization': f'Bearer {admin_token}'}
|
|
driver_h = {'Authorization': f'Bearer {driver_token}'}
|
|
|
|
# Create test data directly in DB
|
|
customer = Customer(customer_name='Test', mobile='13800001111', settlement_type='immediate',
|
|
settlement_days=0, salesman_id=2, credit_limit=0, deleted=0)
|
|
product = Product(product_name='TestProd', specification='100x100', unit='pcs',
|
|
category='cat', cost_price=60.0, sale_price=100.0, status=1)
|
|
supplier = Supplier(supplier_name='TestFactory', supplier_type='factory',
|
|
contact_name='Contact', contact_mobile='13900001111', status=1)
|
|
session.add_all([customer, product, supplier])
|
|
session.flush()
|
|
|
|
order = SalesOrder(
|
|
order_no='TEST-001', customer_id=customer.id, customer_name=customer.customer_name,
|
|
customer_mobile=customer.mobile, salesman_id=2, order_status='approved',
|
|
factory_id=supplier.id, sale_price_total=1000.0, cost_price_total=600.0,
|
|
profit_total=400.0, profit_rate=40.0, deleted=0,
|
|
)
|
|
session.add(order)
|
|
session.flush()
|
|
print(f'1. Order created: id={order.id}, status={order.order_status}')
|
|
|
|
# Create task via API
|
|
resp = c.post('/api/logistics/tasks', headers=admin_h, json={
|
|
'order_id': order.id, 'driver_id': 5,
|
|
'pickup_address': 'Factory', 'delivery_address': 'Warehouse',
|
|
'pickup_content': 'Goods', 'quantity': 10,
|
|
})
|
|
print(f'2. Create task: status={resp.status_code}')
|
|
if resp.status_code == 200:
|
|
task_id = resp.json()['data']['task_id']
|
|
print(f' task_id={task_id}')
|
|
else:
|
|
print(f' Error: {resp.text[:200]}')
|
|
sys.exit(1)
|
|
|
|
# Check order status in DB
|
|
session.expire_all()
|
|
order = session.get(SalesOrder, order.id)
|
|
print(f'3. Order status after create_task: {order.order_status}')
|
|
|
|
# Check task in DB
|
|
from backend.app.models.business import LogisticsTask
|
|
task = session.get(LogisticsTask, task_id)
|
|
print(f'4. Task: id={task.id}, order_id={task.order_id}, status={task.status}')
|
|
|
|
# Accept task
|
|
resp = c.post(f'/api/driver/tasks/{task_id}/accept', headers=driver_h, json={})
|
|
print(f'5. Accept task: status={resp.status_code}')
|
|
if resp.status_code != 200:
|
|
print(f' Error: {resp.text[:200]}')
|
|
|
|
# Check order status
|
|
session.expire_all()
|
|
order = session.get(SalesOrder, order.id)
|
|
print(f'6. Order status after accept: {order.order_status}')
|
|
|
|
# Check task status
|
|
session.expire_all()
|
|
task = session.get(LogisticsTask, task_id)
|
|
print(f'7. Task status after accept: {task.status}')
|
|
|
|
app.dependency_overrides.clear()
|
|
transaction.rollback()
|
|
session.close()
|
|
connection.close()
|