teszt állományok áthelyezése és szelektálása

2026-06-05 10:50:25 +00:00
parent 18524a08f2
commit f03b5f3916
63 changed files with 633 additions and 527 deletions
--- a/tests/archive/test_asset_e2e_fetch.py.old
+++ b/tests/archive/test_asset_e2e_fetch.py.old
@@ -0,0 +1,128 @@
+import asyncio
+import httpx
+from sqlalchemy import select, delete
+from sqlalchemy.orm import selectinload
+from app.db.session import AsyncSessionLocal
+from app.models.identity import User
+from app.models.marketplace.organization import Organization, Branch, OrganizationMember
+from app.models.vehicle.asset import Asset
+from app.models.vehicle.vehicle_definitions import VehicleModelDefinition
+from app.core.security import create_tokens
+
+async def setup_test_data():
+    async with AsyncSessionLocal() as db:
+        # Just find any 2 users that belong to ANY organization
+        stmt = select(OrganizationMember).limit(2)
+        members = (await db.execute(stmt)).scalars().all()
+        if not members:
+            print("No org members found")
+            return None
+        
+        user1_id = members[0].user_id
+        org_id = members[0].organization_id
+        
+        user2_id = members[1].user_id if len(members)>1 else user1_id
+
+        # Find or create branch
+        branch_stmt = select(Branch).where(Branch.organization_id == org_id, Branch.is_main == True)
+        branch = (await db.execute(branch_stmt)).scalar()
+        if not branch:
+            branch = Branch(organization_id=org_id, name="Test Main Branch", is_main=True)
+            db.add(branch)
+            await db.flush()
+
+        # Clean old test data
+        await db.execute(delete(Asset).where(Asset.license_plate.in_(["TEST-123", "DRAFT-456", "OTHER-123"])))
+        await db.execute(delete(VehicleModelDefinition).where(VehicleModelDefinition.make == "Ford", VehicleModelDefinition.marketing_name == "Focus"))
+        await db.commit()
+
+        model_def = VehicleModelDefinition(
+            make="Ford",
+            marketing_name="Focus",
+            normalized_name="ford_focus",
+            year_from=2015,
+            year_to=2020,
+            power_kw=92,
+            data_status="verified"
+        )
+        db.add(model_def)
+        await db.commit()
+
+        token1, _ = create_tokens(data={"sub": str(user1_id)})
+        token2, _ = create_tokens(data={"sub": str(user2_id)})
+        
+        return token1, token2, org_id, branch.id, model_def.id
+
+async def run_tests():
+    print("--- SETUP ---")
+    result = await setup_test_data()
+    if not result:
+        return
+    token1, token2, org_id, branch_id, model_id = result
+    print("Test data created successfully.")
+
+    headers1 = {"Authorization": f"Bearer {token1}"}
+    headers2 = {"Authorization": f"Bearer {token2}"}
+
+    async with httpx.AsyncClient(base_url="http://127.0.0.1:8000") as client:
+        print("\n--- TEST 1: CATALOG MATCH & GARAGE TEST ---")
+        payload1 = {
+            "vin": "TESTVIN1234567890",
+            "license_plate": "TEST-123",
+            "brand": "Ford",
+            "model": "Focus",
+            "year_of_manufacture": 2018,
+            "organization_id": org_id,
+            "vehicle_class": "car",
+            "fuel_type": "petrol"
+        }
+        resp1 = await client.post("/api/v1/vehicles", json=payload1, headers=headers1)
+        print(f"Status: {resp1.status_code}")
+        try:
+            data1 = resp1.json()
+            print(f"Response: {data1}")
+            if resp1.status_code == 201:
+                assert data1["catalog_id"] == model_id, f"Matcher failed: expected {model_id}, got {data1.get('catalog_id')}"
+                assert data1["branch_id"] == str(branch_id), f"Branch failed: expected {branch_id}, got {data1.get('branch_id')}"
+                print("-> SUCCESS: Matcher assigned catalog and Main Branch assigned.")
+        except Exception as e:
+            print(f"-> FAILED: {e}")
+
+        print("\n--- TEST 2: DRAFT / INCOMPLETE TEST ---")
+        payload2 = {
+            "license_plate": "DRAFT-456",
+            "brand": "Ismeretlen",
+            "organization_id": org_id
+        }
+        resp2 = await client.post("/api/v1/vehicles", json=payload2, headers=headers1)
+        print(f"Status: {resp2.status_code}")
+        try:
+            data2 = resp2.json()
+            print(f"Response: {data2}")
+            if resp2.status_code == 201:
+                assert data2["status"] == "draft", f"Status failed: expected 'draft', got {data2.get('status')}"
+                assert data2["branch_id"] == str(branch_id), f"Branch failed: expected {branch_id}, got {data2.get('branch_id')}"
+                print("-> SUCCESS: Draft created and Main Branch assigned.")
+        except Exception as e:
+            print(f"-> FAILED: {e}")
+
+        print("\n--- TEST 3: OWNERSHIP TRANSFER TEST ---")
+        payload3 = {
+            "vin": "TESTVIN1234567890",
+            "license_plate": "OTHER-123",
+            "brand": "Ford",
+            "model": "Focus",
+            "organization_id": org_id
+        }
+        resp3 = await client.post("/api/v1/vehicles", json=payload3, headers=headers2)
+        print(f"Status: {resp3.status_code}")
+        try:
+            data3 = resp3.json()
+            print(f"Response: {data3}")
+            if resp3.status_code == 202:
+                print("-> SUCCESS: VIN collision detected (transfer_pending).")
+        except Exception as e:
+            print(f"-> FAILED: {e}")
+
+if __name__ == "__main__":
+    asyncio.run(run_tests())