Update couple of tests to new api

Author: mvantellingen

tests/platform/test_custom_fields.py

@@ -9,12 +9,12 @@ def test_serialize_field_container():
         fields=models.FieldContainer(foobar=10),
     )
 
-    result = CustomFieldsDraftSchema().dump(draft)
+    result = draft.serialize()
     expected = {
         "fields": {"foobar": 10},
         "type": {"typeId": "type", "id": "foobar"},
     }
     assert expected == result
 
-    roundtrip = CustomFieldsDraftSchema().load(expected)
+    roundtrip = models.CustomFieldsDraft.deserialize(expected)
     assert draft == roundtrip

tests/platform/test_errors.py

@@ -1,7 +1,7 @@
 from commercetools.platform import models
 from commercetools.platform.client import Client as PlatformClient
 from commercetools.platform.models import ExtensionNoResponseError, QueryTimedOutError
-from commercetools.platform.models._schemas.error import ErrorResponseSchema
+from commercetools.platform.models import ErrorResponse
 
 
 def test_raises_exception(ct_platform_client: PlatformClient):
@@ -52,7 +52,7 @@ def test_extension_no_response_error():
         "statusCode": 504,
     }
 
-    obj = ErrorResponseSchema().load(error_response)
+    obj = ErrorResponse.deserialize(error_response)
     error = obj.errors[0]
 
     assert isinstance(error, ExtensionNoResponseError)
@@ -64,7 +64,7 @@ def test_query_timeout_error():
         "message": "The query timed out.",
         "statusCode": 400,
     }
-    obj = ErrorResponseSchema().load(error_response)
+    obj = ErrorResponse.deserialize(error_response)
     error = obj.errors[0]
 
     assert isinstance(error, QueryTimedOutError)

tests/platform/test_expand.py

@@ -1,39 +1,60 @@
 import commercetools
 from commercetools.platform import models
+from commercetools.platform import Client as PlatformClient
 from tests.platform.test_service_order import get_test_order
 
 
-def test_unknown_expand_terms(old_client: commercetools.Client):
-    cart = old_client.carts.create(models.CartDraft(currency="EUR"))
-
-    order = old_client.orders.create(
-        models.OrderFromCartDraft(
-            id=cart.id, version=1, cart=cart, order_number="test-order"
-        ),
-        expand="nonExisting",
+def test_unknown_expand_terms(ct_platform_client: PlatformClient):
+    cart = (
+        ct_platform_client.with_project_key("unittest")
+        .carts()
+        .post(models.CartDraft(currency="EUR"))
+    )
+    assert cart
+
+    order = (
+        ct_platform_client.with_project_key("unittest")
+        .orders()
+        .post(
+            models.OrderFromCartDraft(
+                id=cart.id,
+                version=1,
+                cart=models.CartResourceIdentifier(id=cart.id),
+                order_number="test-order",
+            ),
+            expand=["nonExisting"],
+        )
     )
 
     assert order.id
 
 
-def test_optional_expanded_terms(old_client, commercetools_api):
+def test_optional_expanded_terms(ct_platform_client: PlatformClient, commercetools_api):
     order = get_test_order()
     commercetools_api.orders.add_existing(order)
 
-    expanded_order = old_client.orders.get_by_id(
-        order.id, expand="discountCodes[*].discountCode"
+    expanded_order = (
+        ct_platform_client.with_project_key("unittest")
+        .orders()
+        .with_id(order.id)
+        .get(expand=["discountCodes[*].discountCode"])
     )
 
     assert expanded_order.id
     assert expanded_order.discount_codes is None
 
 
-def test_unknown_reference_expand_terms(old_client, commercetools_api):
+def test_unknown_reference_expand_terms(
+    ct_platform_client: PlatformClient, commercetools_api
+):
     order = get_test_order()
     commercetools_api.orders.add_existing(order)
 
-    expanded_order = old_client.orders.get_by_id(
-        order.id, expand="shippingInfo.shippingMethod"
+    expanded_order = (
+        ct_platform_client.with_project_key("unittest")
+        .orders()
+        .with_id(order.id)
+        .get(expand=["shippingInfo.shippingMethod"])
     )
 
     assert expanded_order.id