diff --git a/tests/integration/workflows/test_integration.py b/tests/integration/workflows/test_integration.py index 412ce915..444bfd97 100644 --- a/tests/integration/workflows/test_integration.py +++ b/tests/integration/workflows/test_integration.py @@ -47,8 +47,11 @@ class TestIntegrationFlows: # 3. List all vendors response = client.get("/api/v1/admin/vendors", headers=admin_headers) assert response.status_code == 200 - vendors = response.json() - assert any(v["vendor_code"] == vendor["vendor_code"] for v in vendors) + vendors_response = response.json() + assert any( + v["vendor_code"] == vendor["vendor_code"] + for v in vendors_response["vendors"] + ) def test_admin_product_listing_workflow( self, client, admin_headers, test_marketplace_product