test_web_app.py

#!/usr/bin/env python3
"""
Test script for the Drug Barcode Lookup Web Application
"""

import requests
import json
import time
import os

def test_web_app():
    """Test the web application functionality"""
    base_url = "http://localhost:5000"
    
    print("Testing Drug Barcode Lookup Web Application")
    print("=" * 50)
    
    # Test 1: Check if server is running
    print("1. Testing server availability...")
    try:
        response = requests.get(base_url, timeout=5)
        if response.status_code == 200:
            print("✓ Server is running")
        else:
            print(f"✗ Server returned status code: {response.status_code}")
            return False
    except requests.exceptions.RequestException as e:
        print(f"✗ Server is not running: {e}")
        print("Please start the server first: python app.py")
        return False
    
    # Test 2: Test API endpoint with known barcode
    print("\n2. Testing API endpoint with known barcode...")
    known_barcode = "8901571007363"  # From our database
    try:
        response = requests.get(f"{base_url}/api/lookup/{known_barcode}", timeout=10)
        if response.status_code == 200:
            data = response.json()
            if data.get('found'):
                print(f"✓ Found barcode: {data.get('drug_name', 'Unknown')}")
            else:
                print(f"✗ Barcode not found in database")
        else:
            print(f"✗ API returned status code: {response.status_code}")
    except requests.exceptions.RequestException as e:
        print(f"✗ API test failed: {e}")
    
    # Test 3: Test API endpoint with unknown barcode
    print("\n3. Testing API endpoint with unknown barcode...")
    unknown_barcode = "9999999999999"
    try:
        response = requests.get(f"{base_url}/api/lookup/{unknown_barcode}", timeout=10)
        if response.status_code == 200:
            data = response.json()
            if not data.get('found'):
                print("✓ Correctly identified unknown barcode")
            else:
                print("✗ Should have identified as unknown")
        else:
            print(f"✗ API returned status code: {response.status_code}")
    except requests.exceptions.RequestException as e:
        print(f"✗ API test failed: {e}")
    
    # Test 4: Test database endpoint
    print("\n4. Testing database endpoint...")
    try:
        response = requests.get(f"{base_url}/database", timeout=10)
        if response.status_code == 200:
            print("✓ Database endpoint is accessible")
        else:
            print(f"✗ Database endpoint returned status code: {response.status_code}")
    except requests.exceptions.RequestException as e:
        print(f"✗ Database test failed: {e}")
    
    print("\n" + "=" * 50)
    print("Web application test completed!")
    print("You can now:")
    print("1. Open http://localhost:5000 in your browser")
    print("2. Upload an image with a barcode")
    print("3. View the results and drug information")
    print("4. Check the database for all processed images")

def main():
    """Main test function"""
    # Check if database exists
    if not os.path.exists('barcode_text_results.json'):
        print("✗ Database file not found!")
        print("Please run: python barcode_scanner.py first")
        return
    
    # Run tests
    test_web_app()

if __name__ == "__main__":
    main()