address_type is null after setting billing and shipping addresses on cart

Author: XxXgeoXxX

app/code/Magento/QuoteGraphQl/etc/schema.graphqls

@@ -211,6 +211,11 @@ type SelectedPaymentMethod {
 type SelectedPaymentMethodAdditionalData {
 }
 
+enum AdressTypeEnum {
+    SHIPPING
+    BILLING
+}
+
 type AppliedCoupon {
     code: String!
 }

dev/tests/api-functional/testsuite/Magento/GraphQl/Quote/Customer/SetBillingAddressOnCartTest.php

@@ -90,6 +90,7 @@ public function testSetNewBillingAddress()
           code
           label
         }
+        address_type
       }
     }
   }
@@ -147,6 +148,7 @@ public function testSetNewBillingAddressWithUseForShippingParameter()
           code
           label
         }
+        address_type
       }
       shipping_addresses {
         firstname
@@ -160,6 +162,7 @@ public function testSetNewBillingAddressWithUseForShippingParameter()
           code
           label
         }
+        address_type
       }
     }
   }
@@ -174,7 +177,7 @@ public function testSetNewBillingAddressWithUseForShippingParameter()
         self::assertArrayHasKey('shipping_addresses', $cartResponse);
         $shippingAddressResponse = current($cartResponse['shipping_addresses']);
         $this->assertNewAddressFields($billingAddressResponse);
-        $this->assertNewAddressFields($shippingAddressResponse);
+        $this->assertNewAddressFields($shippingAddressResponse, 'shipping');
     }
 
     /**
@@ -403,9 +406,10 @@ public function testSetBillingAddressIfCustomerIsNotOwnerOfAddress()
     /**
      * Verify the all the whitelisted fields for a New Address Object
      *
-     * @param array $billingAddressResponse
+     * @param array $addressResponse
+     * @param string $addressType
      */
-    private function assertNewAddressFields(array $billingAddressResponse): void
+    private function assertNewAddressFields(array $addressResponse, string $addressType = 'billing'): void
     {
         $assertionMap = [
             ['response_field' => 'firstname', 'expected_value' => 'test firstname'],
@@ -416,9 +420,10 @@ private function assertNewAddressFields(array $billingAddressResponse): void
             ['response_field' => 'postcode', 'expected_value' => '887766'],
             ['response_field' => 'telephone', 'expected_value' => '88776655'],
             ['response_field' => 'country', 'expected_value' => ['code' => 'US', 'label' => 'US']],
+            ['response_field' => 'address_type', 'expected_value' => $addressType]
         ];
 
-        $this->assertResponseFields($billingAddressResponse, $assertionMap);
+        $this->assertResponseFields($addressResponse, $assertionMap);
     }
 
     /**

dev/tests/api-functional/testsuite/Magento/GraphQl/Quote/Guest/SetBillingAddressOnCartTest.php

@@ -82,6 +82,7 @@ public function testSetNewBillingAddress()
           code
           label
         }
+        address_type
       }
     }
   }
@@ -138,6 +139,7 @@ public function testSetNewBillingAddressWithUseForShippingParameter()
           code
           label
         }
+        address_type
       }
       shipping_addresses {
         firstname
@@ -151,6 +153,7 @@ public function testSetNewBillingAddressWithUseForShippingParameter()
           code
           label
         }
+        address_type
       }
     }
   }
@@ -165,7 +168,7 @@ public function testSetNewBillingAddressWithUseForShippingParameter()
         self::assertArrayHasKey('shipping_addresses', $cartResponse);
         $shippingAddressResponse = current($cartResponse['shipping_addresses']);
         $this->assertNewAddressFields($billingAddressResponse);
-        $this->assertNewAddressFields($shippingAddressResponse);
+        $this->assertNewAddressFields($shippingAddressResponse, 'shipping');
     }
 
     /**
@@ -244,9 +247,10 @@ public function testSetBillingAddressFromAddressBook()
     /**
      * Verify the all the whitelisted fields for a New Address Object
      *
-     * @param array $billingAddressResponse
+     * @param array $addressResponse
+     * @param string $addressType
      */
-    private function assertNewAddressFields(array $billingAddressResponse): void
+    private function assertNewAddressFields(array $addressResponse, string $addressType = 'billing'): void
     {
         $assertionMap = [
             ['response_field' => 'firstname', 'expected_value' => 'test firstname'],
@@ -257,9 +261,10 @@ private function assertNewAddressFields(array $billingAddressResponse): void
             ['response_field' => 'postcode', 'expected_value' => '887766'],
             ['response_field' => 'telephone', 'expected_value' => '88776655'],
             ['response_field' => 'country', 'expected_value' => ['code' => 'US', 'label' => 'US']],
+            ['response_field' => 'address_type', 'expected_value' => $addressType]
         ];
 
-        $this->assertResponseFields($billingAddressResponse, $assertionMap);
+        $this->assertResponseFields($addressResponse, $assertionMap);
     }
 
     /**

dev/tests/api-functional/testsuite/Magento/GraphQl/Quote/SetShippingAddressOnCartTest.php

@@ -94,6 +94,7 @@ public function testSetNewShippingAddressByGuest()
           code
           label
         }
+        address_type
       }
     }
   }
@@ -184,6 +185,7 @@ public function testSetNewShippingAddressByRegisteredCustomer()
           label
           code
         }
+        address_type
       }
     }
   }
@@ -472,6 +474,7 @@ private function assertNewShippingAddressFields(array $shippingAddressResponse):
             ['response_field' => 'postcode', 'expected_value' => '887766'],
             ['response_field' => 'telephone', 'expected_value' => '88776655'],
             ['response_field' => 'country', 'expected_value' => ['code' => 'US', 'label' => 'US']],
+            ['response_field' => 'address_type', 'expected_value' => 'shipping']
         ];
 
         $this->assertResponseFields($shippingAddressResponse, $assertionMap);