Spaces:
Runtime error
Runtime error
| import sys | |
| import os | |
| import json | |
| sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../'))) | |
| from main import app, db | |
| from utils.db.db import Doctor, ChatSession, ChatMessage, Patient | |
| def verify_workflow(): | |
| with app.test_client() as client: | |
| # Preamble: Create context | |
| with app.app_context(): | |
| # Ensure Test Patient | |
| if not Patient.query.get(999): | |
| p = Patient(id=999, name="Test Patient", email="test@patient.com", password="pwd") | |
| db.session.add(p) | |
| # Ensure Test Doctor | |
| doc = Doctor.query.filter_by(email="testdoc@example.com").first() | |
| if not doc: | |
| doc = Doctor(name="Test Doc", specialization="General", email="testdoc@example.com", phone="123", password="password", bio="Test Bio") | |
| db.session.add(doc) | |
| db.session.commit() | |
| # 1. Login | |
| print("\n--- Test 1: Login ---") | |
| client.post('/login/doctor', data={'name': 'Test Doc', 'password': 'password'}, follow_redirects=True) | |
| print("PASS: Login attempt made.") | |
| # 2. Start Session (Patient Search) | |
| print("\n--- Test 2: Start Session with Patient ID ---") | |
| # Simulating frontend call: sends "patient search on" then calls ID | |
| resp = client.post('/api/chat', json={'message': 'patient search on'}) # session_id=None | |
| # Now send ID (simulating modal submit) | |
| # Note: In real flow, frontend clears session_id then sends ID. | |
| # Main.py logic: if no session_id provided, it creates one. | |
| # If we send patient ID logic keywords, backend links it. | |
| # Actually frontend sends `patientId` *as message body*? Yes: fetchAIResponse(patientId) | |
| resp = client.post('/api/chat', json={'message': '999'}) | |
| data = resp.get_json() | |
| session_id = data.get('session_id') | |
| print(f"Response Session ID: {session_id}") | |
| if session_id: | |
| print("PASS: Session created.") | |
| else: | |
| print("FAIL: No session ID returned.") | |
| return | |
| # 3. Chat in Session | |
| print("\n--- Test 3: Chat in Session ---") | |
| resp = client.post('/api/chat', json={'message': 'Hello Doctor', 'session_id': session_id}) | |
| data = resp.get_json() | |
| # Verify message count in DB | |
| with app.app_context(): | |
| count = ChatMessage.query.filter_by(session_id=session_id).count() | |
| print(f"Messages in DB for Session {session_id}: {count}") | |
| if count >= 2: # At least User + AI response from step 2 and step 3... wait, step 2 was search result. | |
| print("PASS: Messages saved.") | |
| else: | |
| print("FAIL: Messages not saved correctly.") | |
| # 4. Get History | |
| print("\n--- Test 4: Get History ---") | |
| resp = client.get('/api/history') | |
| hist = resp.get_json()['history'] | |
| print(f"History Sessions: {len(hist)}") | |
| found = any(s['session_id'] == session_id for s in hist) | |
| if found: | |
| print("PASS: Session found in history.") | |
| else: | |
| print("FAIL: Session NOT found in history.") | |
| # 5. Restore Session | |
| print("\n--- Test 5: Restore Session ---") | |
| resp = client.get(f'/api/history/{session_id}') | |
| msgs = resp.get_json()['messages'] | |
| print(f"Restored Messages: {len(msgs)}") | |
| if len(msgs) > 0: | |
| print("PASS: Messages restored.") | |
| else: | |
| print("FAIL: Restoration failed.") | |
| if __name__ == "__main__": | |
| verify_workflow() | |