Initial commit - PromptTech

tests/admin_test.py

@@ -0,0 +1,343 @@
+import requests
+import sys
+import json
+from datetime import datetime
+
+class AdminAPITester:
+    def __init__(self, base_url="https://dev-foundry.preview.emergentagent.com/api"):
+        self.base_url = base_url
+        self.admin_token = None
+        self.user_token = None
+        self.tests_run = 0
+        self.tests_passed = 0
+        self.test_results = []
+        self.created_product_id = None
+        self.created_service_id = None
+        self.created_order_id = None
+
+    def log_test(self, name, success, details=""):
+        """Log test result"""
+        self.tests_run += 1
+        if success:
+            self.tests_passed += 1
+        
+        result = {
+            "test": name,
+            "success": success,
+            "details": details,
+            "timestamp": datetime.now().isoformat()
+        }
+        self.test_results.append(result)
+        
+        status = "✅ PASS" if success else "❌ FAIL"
+        print(f"{status} - {name}")
+        if details:
+            print(f"    Details: {details}")
+
+    def run_test(self, name, method, endpoint, expected_status, data=None, headers=None, use_admin=True):
+        """Run a single API test"""
+        url = f"{self.base_url}/{endpoint}"
+        test_headers = {'Content-Type': 'application/json'}
+        
+        token = self.admin_token if use_admin else self.user_token
+        if token:
+            test_headers['Authorization'] = f'Bearer {token}'
+        
+        if headers:
+            test_headers.update(headers)
+
+        try:
+            if method == 'GET':
+                response = requests.get(url, headers=test_headers, timeout=15)
+            elif method == 'POST':
+                response = requests.post(url, json=data, headers=test_headers, timeout=15)
+            elif method == 'PUT':
+                response = requests.put(url, json=data, headers=test_headers, timeout=15)
+            elif method == 'DELETE':
+                response = requests.delete(url, headers=test_headers, timeout=15)
+
+            success = response.status_code == expected_status
+            details = f"Status: {response.status_code}, Expected: {expected_status}"
+            
+            if not success:
+                try:
+                    error_data = response.json()
+                    details += f", Response: {error_data}"
+                except:
+                    details += f", Response: {response.text[:200]}"
+            
+            self.log_test(name, success, details)
+            
+            if success:
+                try:
+                    return response.json()
+                except:
+                    return {"status": "success"}
+            return None
+
+        except Exception as e:
+            self.log_test(name, False, f"Error: {str(e)}")
+            return None
+
+    def test_admin_login(self):
+        """Test admin login"""
+        login_data = {
+            "email": "admin@techzone.com",
+            "password": "admin123"
+        }
+        
+        response = self.run_test("Admin Login", "POST", "auth/login", 200, login_data, use_admin=False)
+        if response and 'access_token' in response:
+            self.admin_token = response['access_token']
+            return True
+        return False
+
+    def test_user_registration_and_login(self):
+        """Test regular user registration for order testing"""
+        test_user_data = {
+            "name": f"Test User {datetime.now().strftime('%H%M%S')}",
+            "email": f"testuser_{datetime.now().strftime('%H%M%S')}@example.com",
+            "password": "testpass123"
+        }
+        
+        response = self.run_test("User Registration for Orders", "POST", "auth/register", 200, test_user_data, use_admin=False)
+        if response and 'access_token' in response:
+            self.user_token = response['access_token']
+            return True
+        return False
+
+    def test_admin_dashboard(self):
+        """Test admin dashboard stats"""
+        return self.run_test("Admin Dashboard Stats", "GET", "admin/dashboard", 200)
+
+    def test_admin_products_crud(self):
+        """Test admin product CRUD operations"""
+        # Get existing products
+        self.run_test("Admin Get Products", "GET", "admin/products", 200)
+        
+        # Create new product
+        product_data = {
+            "name": "Test Product Admin",
+            "description": "Test product created by admin API test",
+            "price": 99.99,
+            "category": "accessories",
+            "image_url": "https://images.unsplash.com/photo-1505740420928-5e560c06d30e?w=800",
+            "stock": 50,
+            "brand": "TestBrand"
+        }
+        
+        create_response = self.run_test("Admin Create Product", "POST", "admin/products", 200, product_data)
+        if create_response and 'id' in create_response:
+            self.created_product_id = create_response['id']
+            
+            # Update the product
+            update_data = {
+                "name": "Updated Test Product",
+                "price": 149.99,
+                "stock": 25
+            }
+            self.run_test("Admin Update Product", "PUT", f"admin/products/{self.created_product_id}", 200, update_data)
+            
+            # Delete the product (soft delete)
+            self.run_test("Admin Delete Product", "DELETE", f"admin/products/{self.created_product_id}", 200)
+            
+            return True
+        return False
+
+    def test_admin_services_crud(self):
+        """Test admin service CRUD operations"""
+        # Get existing services
+        self.run_test("Admin Get Services", "GET", "admin/services", 200)
+        
+        # Create new service
+        service_data = {
+            "name": "Test Service Admin",
+            "description": "Test service created by admin API test",
+            "price": 79.99,
+            "duration": "1 hour",
+            "image_url": "https://images.unsplash.com/photo-1581092918056-0c4c3acd3789?w=800",
+            "category": "repair"
+        }
+        
+        create_response = self.run_test("Admin Create Service", "POST", "admin/services", 200, service_data)
+        if create_response and 'id' in create_response:
+            self.created_service_id = create_response['id']
+            
+            # Update the service
+            update_data = {
+                "name": "Updated Test Service",
+                "price": 99.99,
+                "duration": "2 hours"
+            }
+            self.run_test("Admin Update Service", "PUT", f"admin/services/{self.created_service_id}", 200, update_data)
+            
+            # Delete the service (soft delete)
+            self.run_test("Admin Delete Service", "DELETE", f"admin/services/{self.created_service_id}", 200)
+            
+            return True
+        return False
+
+    def test_create_order_for_admin_testing(self):
+        """Create an order using regular user to test admin order management"""
+        if not self.user_token:
+            return False
+            
+        # Get products first
+        products_response = self.run_test("Get Products for Order", "GET", "products", 200, use_admin=False)
+        if not products_response or len(products_response) == 0:
+            return False
+
+        product_id = products_response[0]['id']
+
+        # Add to cart
+        add_cart_data = {"product_id": product_id, "quantity": 1}
+        self.run_test("Add to Cart for Order", "POST", "cart/add", 200, add_cart_data, use_admin=False)
+
+        # Create order
+        order_data = {
+            "shipping_address": {
+                "street": "123 Test St",
+                "city": "Test City",
+                "state": "TS",
+                "zip": "12345",
+                "country": "USA"
+            },
+            "notes": "Test order for admin testing"
+        }
+        
+        order_response = self.run_test("Create Order for Admin Testing", "POST", "orders", 200, order_data, use_admin=False)
+        if order_response and 'order_id' in order_response:
+            self.created_order_id = order_response['order_id']
+            return True
+        return False
+
+    def test_admin_orders_management(self):
+        """Test admin order management"""
+        # Get all orders
+        orders_response = self.run_test("Admin Get Orders", "GET", "admin/orders", 200)
+        
+        if self.created_order_id:
+            # Update order status
+            status_update = {
+                "status": "processing",
+                "notes": "Order is being processed",
+                "tracking_number": "TEST123456"
+            }
+            self.run_test("Admin Update Order Status", "PUT", f"admin/orders/{self.created_order_id}/status", 200, status_update)
+        
+        # Test order filtering
+        self.run_test("Admin Get Orders by Status", "GET", "admin/orders?status=pending", 200)
+        
+        return True
+
+    def test_admin_inventory_management(self):
+        """Test admin inventory management"""
+        # Get inventory
+        inventory_response = self.run_test("Admin Get Inventory", "GET", "admin/inventory", 200)
+        
+        if inventory_response and len(inventory_response) > 0:
+            product_id = inventory_response[0]['id']
+            
+            # Adjust inventory
+            adjust_data = {
+                "quantity_change": 10,
+                "notes": "Test inventory adjustment"
+            }
+            self.run_test("Admin Adjust Inventory", "POST", f"admin/inventory/{product_id}/adjust", 200, adjust_data)
+            
+            # Get inventory logs
+            self.run_test("Admin Get Inventory Logs", "GET", f"admin/inventory/{product_id}/logs", 200)
+        
+        return True
+
+    def test_admin_bookings_management(self):
+        """Test admin bookings management"""
+        # Get bookings
+        bookings_response = self.run_test("Admin Get Bookings", "GET", "admin/bookings", 200)
+        
+        # Test booking status filter
+        self.run_test("Admin Get Pending Bookings", "GET", "admin/bookings?status=pending", 200)
+        
+        return True
+
+    def test_admin_users_management(self):
+        """Test admin users management"""
+        return self.run_test("Admin Get Users", "GET", "admin/users", 200)
+
+    def test_admin_reports(self):
+        """Test admin reports functionality"""
+        # Test sales reports
+        self.run_test("Admin Sales Report Daily", "GET", "admin/reports/sales?period=daily", 200)
+        self.run_test("Admin Sales Report Weekly", "GET", "admin/reports/sales?period=weekly", 200)
+        self.run_test("Admin Sales Report Monthly", "GET", "admin/reports/sales?period=monthly", 200)
+        
+        return True
+
+    def test_admin_export_reports(self):
+        """Test admin report exports"""
+        # Test CSV exports
+        csv_response = self.run_test("Admin Export CSV Sales", "GET", "admin/reports/export/csv?report_type=sales&period=monthly", 200)
+        csv_inventory = self.run_test("Admin Export CSV Inventory", "GET", "admin/reports/export/csv?report_type=inventory", 200)
+        csv_orders = self.run_test("Admin Export CSV Orders", "GET", "admin/reports/export/csv?report_type=orders", 200)
+        
+        # Test PDF exports
+        pdf_response = self.run_test("Admin Export PDF Sales", "GET", "admin/reports/export/pdf?report_type=sales&period=monthly", 200)
+        pdf_inventory = self.run_test("Admin Export PDF Inventory", "GET", "admin/reports/export/pdf?report_type=inventory", 200)
+        
+        return True
+
+    def run_all_admin_tests(self):
+        """Run all admin API tests"""
+        print("🔐 Starting TechZone Admin API Tests...")
+        print(f"📍 Testing against: {self.base_url}")
+        print("=" * 60)
+
+        # Test admin authentication
+        if not self.test_admin_login():
+            print("❌ Admin login failed - cannot continue with admin tests")
+            return False
+
+        # Test user registration for order testing
+        self.test_user_registration_and_login()
+
+        # Test admin dashboard
+        self.test_admin_dashboard()
+
+        # Test admin CRUD operations
+        self.test_admin_products_crud()
+        self.test_admin_services_crud()
+
+        # Create order for admin testing
+        self.test_create_order_for_admin_testing()
+
+        # Test admin management features
+        self.test_admin_orders_management()
+        self.test_admin_inventory_management()
+        self.test_admin_bookings_management()
+        self.test_admin_users_management()
+
+        # Test admin reports
+        self.test_admin_reports()
+        self.test_admin_export_reports()
+
+        # Print summary
+        print("=" * 60)
+        print(f"📊 Admin tests completed: {self.tests_passed}/{self.tests_run}")
+        success_rate = (self.tests_passed / self.tests_run * 100) if self.tests_run > 0 else 0
+        print(f"📈 Success rate: {success_rate:.1f}%")
+        
+        if self.tests_passed < self.tests_run:
+            print("\n❌ Failed tests:")
+            for result in self.test_results:
+                if not result['success']:
+                    print(f"   - {result['test']}: {result['details']}")
+
+        return self.tests_passed == self.tests_run
+
+def main():
+    tester = AdminAPITester()
+    success = tester.run_all_admin_tests()
+    return 0 if success else 1
+
+if __name__ == "__main__":
+    sys.exit(main())