feat(reference): add comprehensive test script for POST and PUT reference data endpoints

test_reference_crud.py

@@ -0,0 +1,253 @@
+#!/usr/bin/env python3
+"""
+Comprehensive test script for all POST and PUT reference data endpoints.
+
+This script tests:
+1. POST endpoints (create) for all reference data types
+2. PUT endpoints (update) for all reference data types
+3. Validates responses and error handling
+
+Reference data types tested:
+- States
+- Countries
+- Company Types
+- Lead Sources
+- Payment Terms
+- Purchase Prices
+- Rental Prices
+- Barrier Sizes
+- Product Applications
+- FOBs
+- Estimated Ship Dates
+- Estimated Freights
+- Status Info
+- Priorities
+
+Usage:
+python test_reference_crud.py
+"""
+
+import requests
+import json
+import sys
+from typing import Dict, Any
+
+BASE_URL = "http://localhost:8001"  # Adjust port as needed
+
+# Test data for creating reference records
+TEST_DATA = {
+    "states": {
+        "create": {"description": "Test State", "customer_type_id": 1, "enabled": True},
+        "update": {"description": "Updated Test State", "enabled": False}
+    },
+    "countries": {
+        "create": {"description": "Test Country", "customer_type_id": 1, "enabled": True},
+        "update": {"description": "Updated Test Country", "enabled": False}
+    },
+    "company_types": {
+        "create": {"description": "Test Company Type", "customer_type_id": 1, "inactive": False},
+        "update": {"description": "Updated Test Company Type", "inactive": True}
+    },
+    "lead_sources": {
+        "create": {"source_name": "Test Lead Source", "description": "Test description", "enabled": True},
+        "update": {"source_name": "Updated Test Lead Source", "description": "Updated description", "enabled": False}
+    },
+    "payment_terms": {
+        "create": {"description": "Test Payment Term", "enabled": True},
+        "update": {"description": "Updated Test Payment Term", "enabled": False}
+    },
+    "purchase_prices": {
+        "create": {"price": "100.00", "enabled": True},
+        "update": {"price": "150.00", "enabled": False}
+    },
+    "rental_prices": {
+        "create": {"price": "50.00", "enabled": True},
+        "update": {"price": "75.00", "enabled": False}
+    },
+    "barrier_sizes": {
+        "create": {"length": 10.5, "height": 5.0, "width": 2.0, "cableunits": 1.0, "price": 100.00, "is_standard": True},
+        "update": {"length": 12.0, "height": 6.0, "price": 120.00, "is_standard": False}
+    },
+    "product_applications": {
+        "create": {"customer_type_id": "COM", "description": "Test Application", "enabled": True},
+        "update": {"customer_type_id": "RES", "description": "Updated Application", "enabled": False}
+    },
+    "fobs": {
+        "create": {"fob_description": "Test FOB"},
+        "update": {"fob_description": "Updated Test FOB"}
+    },
+    "est_ship_dates": {
+        "create": {"est_ship_date_description": "Test Ship Date"},
+        "update": {"est_ship_date_description": "Updated Test Ship Date"}
+    },
+    "est_freights": {
+        "create": {"est_freight_description": "Test Freight"},
+        "update": {"est_freight_description": "Updated Test Freight"}
+    },
+    "status_info": {
+        "create": {"description": "Test Status", "abrv": "TST"},
+        "update": {"description": "Updated Test Status", "abrv": "UTST"}
+    },
+    "priorities": {
+        "create": {"customer_type_id": 1, "description": "Test Priority"},
+        "update": {"customer_type_id": 2, "description": "Updated Test Priority"}
+    }
+}
+
+# Endpoint mappings
+ENDPOINTS = {
+    "states": {"post": "/api/v1/reference/states", "put": "/api/v1/reference/states/{id}"},
+    "countries": {"post": "/api/v1/reference/countries", "put": "/api/v1/reference/countries/{id}"},
+    "company_types": {"post": "/api/v1/reference/company-types", "put": "/api/v1/reference/company-types/{id}"},
+    "lead_sources": {"post": "/api/v1/reference/lead-sources", "put": "/api/v1/reference/lead-sources/{id}"},
+    "payment_terms": {"post": "/api/v1/reference/payment-terms", "put": "/api/v1/reference/payment-terms/{id}"},
+    "purchase_prices": {"post": "/api/v1/reference/purchase-prices", "put": "/api/v1/reference/purchase-prices/{id}"},
+    "rental_prices": {"post": "/api/v1/reference/rental-prices", "put": "/api/v1/reference/rental-prices/{id}"},
+    "barrier_sizes": {"post": "/api/v1/reference/barrier-sizes", "put": "/api/v1/reference/barrier-sizes/{id}"},
+    "product_applications": {"post": "/api/v1/reference/product-applications", "put": "/api/v1/reference/product-applications/{id}"},
+    "fobs": {"post": "/api/v1/reference/fobs", "put": "/api/v1/reference/fobs/{id}"},
+    "est_ship_dates": {"post": "/api/v1/reference/est-ship-dates", "put": "/api/v1/reference/est-ship-dates/{id}"},
+    "est_freights": {"post": "/api/v1/reference/est-freights", "put": "/api/v1/reference/est-freights/{id}"},
+    "status_info": {"post": "/api/v1/reference/status-info", "put": "/api/v1/reference/status-info/{id}"},
+    "priorities": {"post": "/api/v1/reference/priorities", "put": "/api/v1/reference/priorities/{id}"}
+}
+
+# ID field mappings for PUT requests
+ID_FIELDS = {
+    "states": "state_id",
+    "countries": "country_id",
+    "company_types": "company_type_id",
+    "lead_sources": "lead_generated_from_id",
+    "payment_terms": "payment_term_id",
+    "purchase_prices": "purchase_price_id",
+    "rental_prices": "rental_price_id",
+    "barrier_sizes": "barrier_size_id",
+    "product_applications": "application_id",
+    "fobs": "fob_id",
+    "est_ship_dates": "est_ship_date_id",
+    "est_freights": "est_freight_id",
+    "status_info": "status_info_id",
+    "priorities": "priority_id"
+}
+
+def test_post_endpoint(endpoint_name: str, endpoint_url: str, test_data: Dict[str, Any]) -> Dict[str, Any]:
+    """Test POST endpoint for creating a record"""
+    print(f"\n🧪 Testing POST {endpoint_name}")
+    print(f"   URL: {endpoint_url}")
+    print(f"   Data: {json.dumps(test_data, indent=2, default=str)}")
+
+    try:
+        response = requests.post(f"{BASE_URL}{endpoint_url}", json=test_data, timeout=10)
+
+        print(f"   Status Code: {response.status_code}")
+
+        if response.status_code == 201:
+            result = response.json()
+            print(f"   ✅ POST {endpoint_name} successful")
+            print(f"   Created record: {json.dumps(result, indent=2, default=str)}")
+            return result
+        else:
+            print(f"   ❌ POST {endpoint_name} failed")
+            print(f"   Response: {response.text}")
+            return None
+
+    except Exception as e:
+        print(f"   ❌ POST {endpoint_name} error: {str(e)}")
+        return None
+
+def test_put_endpoint(endpoint_name: str, endpoint_url: str, record_id: int, test_data: Dict[str, Any]) -> bool:
+    """Test PUT endpoint for updating a record"""
+    url = endpoint_url.replace("{id}", str(record_id))
+    print(f"\n🔄 Testing PUT {endpoint_name}")
+    print(f"   URL: {BASE_URL}{url}")
+    print(f"   Data: {json.dumps(test_data, indent=2, default=str)}")
+
+    try:
+        response = requests.put(f"{BASE_URL}{url}", json=test_data, timeout=10)
+
+        print(f"   Status Code: {response.status_code}")
+
+        if response.status_code == 200:
+            result = response.json()
+            print(f"   ✅ PUT {endpoint_name} successful")
+            print(f"   Updated record: {json.dumps(result, indent=2, default=str)}")
+            return True
+        else:
+            print(f"   ❌ PUT {endpoint_name} failed")
+            print(f"   Response: {response.text}")
+            return False
+
+    except Exception as e:
+        print(f"   ❌ PUT {endpoint_name} error: {str(e)}")
+        return False
+
+def main():
+    """Main test function"""
+    print("🚀 Starting Comprehensive Reference Data CRUD Tests")
+    print("=" * 60)
+
+    # Skip server check for now
+    print("✅ Assuming server is running on localhost:8001")
+
+    results = {
+        "post_success": 0,
+        "post_total": 0,
+        "put_success": 0,
+        "put_total": 0,
+        "created_records": {}
+    }
+
+    # Test POST endpoints
+    print("\n📝 Testing POST (Create) Endpoints")
+    print("=" * 40)
+
+    for ref_type, endpoints in ENDPOINTS.items():
+        results["post_total"] += 1
+
+        test_data = TEST_DATA[ref_type]["create"]
+        result = test_post_endpoint(ref_type, endpoints["post"], test_data)
+
+        if result:
+            results["post_success"] += 1
+            # Store the created record for PUT testing
+            id_field = ID_FIELDS[ref_type]
+            record_id = result.get(id_field)
+            if record_id:
+                results["created_records"][ref_type] = {"id": record_id, "data": result}
+
+    # Test PUT endpoints
+    print("\n🔄 Testing PUT (Update) Endpoints")
+    print("=" * 40)
+
+    for ref_type, record_info in results["created_records"].items():
+        results["put_total"] += 1
+
+        record_id = record_info["id"]
+        test_data = TEST_DATA[ref_type]["update"]
+        endpoints = ENDPOINTS[ref_type]
+
+        success = test_put_endpoint(ref_type, endpoints["put"], record_id, test_data)
+
+        if success:
+            results["put_success"] += 1
+
+    # Summary
+    print("\n📊 Test Results Summary")
+    print("=" * 40)
+    print(f"POST Tests: {results['post_success']}/{results['post_total']} passed")
+    print(f"PUT Tests: {results['put_success']}/{results['put_total']} passed")
+
+    total_success = results['post_success'] + results['put_success']
+    total_tests = results['post_total'] + results['put_total']
+
+    print(f"\nOverall: {total_success}/{total_tests} tests passed")
+
+    if total_success == total_tests:
+        print("🎉 All tests passed! Reference data CRUD operations are working correctly.")
+        return 0
+    else:
+        print("⚠️  Some tests failed. Check the output above for details.")
+        return 1
+
+if __name__ == "__main__":
+    sys.exit(main())