Spaces:
Runtime error
Runtime error
| import sys | |
| import os | |
| sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | |
| from flask import Flask, session, request | |
| from utils.blueprints.assistant import assistant_bp | |
| import mongomock | |
| # Patch mongo with mock | |
| from utils.blueprints import assistant | |
| assistant.mongo.db = mongomock.MongoClient().db | |
| app = Flask(__name__, template_folder='templates') | |
| app.secret_key = 'test' | |
| app.register_blueprint(assistant_bp) | |
| def test_search_and_entry(): | |
| with app.test_request_context(): | |
| # Setup session | |
| session['user_type'] = 'assistant' | |
| session['role'] = 'data_entry' | |
| session['user_name'] = 'Test Assistant' | |
| # 1. Create a patient | |
| assistant.mongo.db.patient.insert_one({ | |
| "id": 101, "sql_id": 101, "name": "John Doe", "phone": "1234567890", "email": "john@example.com" | |
| }) | |
| # 2. Test Search by Phone | |
| with app.test_client() as client: | |
| with client.session_transaction() as sess: | |
| sess['user_type'] = 'assistant' | |
| sess['role'] = 'data_entry' | |
| resp = client.get('/assistant/dashboard/data_entry?q=1234567890') | |
| if b"John Doe" in resp.data: | |
| print("SUCCESS: Search by phone found patient.") | |
| else: | |
| print("FAILURE: Search by phone did not find patient.") | |
| # 3. Test Data Entry with String ID (Legacy Fix) | |
| with app.test_client() as client: | |
| with client.session_transaction() as sess: | |
| sess['user_type'] = 'assistant' | |
| sess['role'] = 'data_entry' | |
| sess['user_name'] = 'Test Assistant' | |
| # Simulate posting to a "string" ID url that matches the int ID in DB | |
| # The route is /add_record/<record_type>/<patient_id> | |
| # We pass "101" as string, backend should convert to int 101 and save. | |
| data = { | |
| "diagnosis_date": "2023-01-01", | |
| "condition": "Flu", | |
| "notes": "Test Note" | |
| } | |
| resp = client.post('/assistant/add_record/summary/101', data=data, follow_redirects=True) | |
| # Check DB | |
| record = assistant.mongo.db.medicalhistory.find_one({"patient_id": 101}) | |
| if record and record['condition'] == 'Flu': | |
| print("SUCCESS: Data entry saved with ID resolution.") | |
| else: | |
| print("FAILURE: Data entry NOT saved.") | |
| if __name__ == "__main__": | |
| test_search_and_entry() | |