MAGETWO-87966: Incorrect tax presented on checkout page

Author: OlgaVasyltsun

app/code/Magento/Checkout/Plugin/Model/Quote/ResetQuoteAddresses.php

@@ -0,0 +1,35 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\Checkout\Plugin\Model\Quote;
+
+use Magento\Quote\Model\Quote;
+
+/**
+ * Clear quote addresses after all items were removed.
+ */
+class ResetQuoteAddresses
+{
+    /**
+     * @param Quote $quote
+     * @param Quote $result
+     * @param mixed $itemId
+     *
+     * @return Quote
+     * @SuppressWarnings(PHPMD.UnusedFormalParameter)
+     */
+    public function afterRemoveItem(Quote $quote, Quote $result, $itemId): Quote
+    {
+        if (empty($result->getAllVisibleItems())) {
+            foreach ($result->getAllAddresses() as $address) {
+                $result->removeAddress($address->getId());
+            }
+        }
+
+        return $result;
+    }
+}

app/code/Magento/Checkout/etc/di.xml

@@ -49,4 +49,7 @@
             </argument>
         </arguments>
     </type>
+    <type name="Magento\Quote\Model\Quote">
+        <plugin name="clear_addresses_after_product_delete" type="Magento\Checkout\Plugin\Model\Quote\ResetQuoteAddresses"/>
+    </type>
 </config>

dev/tests/integration/testsuite/Magento/Checkout/Plugin/Model/Quote/ResetQuoteAddressesTest.php

@@ -0,0 +1,71 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\Checkout\Plugin\Model\Quote;
+
+use Magento\Checkout\Model\Cart;
+use Magento\Checkout\Model\Session;
+use Magento\Quote\Model\BillingAddressManagement;
+use Magento\Quote\Model\Quote;
+use Magento\Quote\Model\Quote\Address as QuoteAddress;
+use Magento\TestFramework\Helper\Bootstrap;
+
+/**
+ * Test for \Magento\Checkout\Plugin\Model\Quote\ResetQuoteAddresses
+ */
+class ResetQuoteAddressesTest extends \PHPUnit\Framework\TestCase
+{
+    /**
+     * @magentoDataFixture Magento/Checkout/_files/quote_with_virtual_product_and_address.php
+     *
+     * @return void
+     */
+    public function testAfterRemoveItem()
+    {
+        /** @var Quote $quote */
+        $quote = Bootstrap::getObjectManager()->create(Quote::class);
+        $quote->load('test_order_with_virtual_product', 'reserved_order_id');
+        /** @var QuoteAddress $quoteShippingAddress */
+        $quoteBillingAddress = Bootstrap::getObjectManager()->create(QuoteAddress::class);
+        $quoteBillingAddress->setRegion('CA')
+            ->setPostcode('90210')
+            ->setFirstname('a_unique_firstname')
+            ->setLastname('lastname')
+            ->setStreet('street')
+            ->setCity('Beverly Hills')
+            ->setEmail('admin@example.com')
+            ->setTelephone('1111111111')
+            ->setCountryId('US')
+            ->setAddressType('billing');
+
+        /** @var BillingAddressManagement $billingAddressManagement */
+        $billingAddressManagement = Bootstrap::getObjectManager()->create(BillingAddressManagement::class);
+        $billingAddressManagement->assign($quote->getId(), $quoteBillingAddress);
+        /** @var Session $checkoutSession */
+        $checkoutSession = Bootstrap::getObjectManager()->create(Session::class);
+        $checkoutSession->setQuoteId($quote->getId());
+        /** @var Cart $cart */
+        $cart = Bootstrap::getObjectManager()->create(Cart::class);
+
+        $activeQuote = $cart->getQuote();
+        $cart->removeItem($activeQuote->getAllVisibleItems()[0]->getId());
+        $cart->save();
+
+        /** @var Quote $quote */
+        $quote = Bootstrap::getObjectManager()->create(Quote::class);
+        $quote->load('test_order_with_virtual_product', 'reserved_order_id');
+        $quoteBillingAddressUpdated = $quote->getBillingAddress();
+        $customer = $quote->getCustomer();
+
+        $this->assertEquals($quoteBillingAddressUpdated->getEmail(), $customer->getEmail());
+        $this->assertEmpty($quoteBillingAddressUpdated->getCountryId());
+        $this->assertEmpty($quoteBillingAddressUpdated->getRegionId());
+        $this->assertEmpty($quoteBillingAddressUpdated->getRegion());
+        $this->assertEmpty($quoteBillingAddressUpdated->getPostcode());
+        $this->assertEmpty($quoteBillingAddressUpdated->getCity());
+    }
+}