Base scripts and templates added

test_app.py

@@ -0,0 +1,131 @@
+#!/usr/bin/env python3
+"""
+Test script for Flask 2FA Authentication Application
+
+This script performs basic functionality tests to ensure the application
+is working correctly.
+"""
+
+import os
+import sys
+import requests
+import time
+
+# Add the project root to Python path
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from app import create_app, db
+from app.models import User
+
+
+def test_database():
+    """Test database connectivity and User model."""
+    print("Testing database connectivity...")
+    
+    app = create_app('development')
+    
+    with app.app_context():
+        try:
+            # Test database connection
+            db.create_all()
+            
+            # Test User model
+            test_user = User(username='testuser', email='test@example.com')
+            test_user.set_password('testpassword123')
+            test_user.generate_totp_secret()
+            
+            print("✓ Database connection successful")
+            print("✓ User model working correctly")
+            print("✓ Password hashing working")
+            print("✓ TOTP secret generation working")
+            
+            # Test TOTP URI generation
+            uri = test_user.generate_totp_uri()
+            print(f"✓ TOTP URI generated: {uri[:50]}...")
+            
+            # Test QR code generation
+            qr_code = test_user.generate_qr_code()
+            print(f"✓ QR code generated ({len(qr_code)} characters)")
+            
+            return True
+            
+        except Exception as e:
+            print(f"✗ Database test failed: {e}")
+            return False
+
+
+def test_application_startup():
+    """Test if the application starts without errors."""
+    print("\nTesting application startup...")
+    
+    try:
+        app = create_app('development')
+        
+        with app.test_client() as client:
+            # Test home page
+            response = client.get('/')
+            if response.status_code == 200:
+                print("✓ Home page loads successfully")
+            else:
+                print(f"✗ Home page failed: {response.status_code}")
+                return False
+            
+            # Test registration page
+            response = client.get('/auth/register')
+            if response.status_code == 200:
+                print("✓ Registration page loads successfully")
+            else:
+                print(f"✗ Registration page failed: {response.status_code}")
+                return False
+            
+            # Test login page
+            response = client.get('/auth/login')
+            if response.status_code == 200:
+                print("✓ Login page loads successfully")
+            else:
+                print(f"✗ Login page failed: {response.status_code}")
+                return False
+            
+            return True
+            
+    except Exception as e:
+        print(f"✗ Application startup test failed: {e}")
+        return False
+
+
+def run_all_tests():
+    """Run all tests."""
+    print("=" * 60)
+    print("Flask 2FA Authentication Application - Test Suite")
+    print("=" * 60)
+    
+    tests_passed = 0
+    total_tests = 2
+    
+    # Test database
+    if test_database():
+        tests_passed += 1
+    
+    # Test application startup
+    if test_application_startup():
+        tests_passed += 1
+    
+    # Results
+    print("\n" + "=" * 60)
+    print("TEST RESULTS")
+    print("=" * 60)
+    print(f"Tests passed: {tests_passed}/{total_tests}")
+    
+    if tests_passed == total_tests:
+        print("✓ All tests passed! The application is ready to use.")
+        print("\nTo start the application:")
+        print("  python run.py")
+        print("\nThen open your browser to: http://127.0.0.1:5000")
+    else:
+        print("✗ Some tests failed. Please check the errors above.")
+    
+    print("=" * 60)
+
+
+if __name__ == '__main__':
+    run_all_tests()