MC-19235: Quote improvement in TestFramework

Author: SeruyV

dev/tests/integration/framework/Magento/TestFramework/Quote/Model/GetQuoteByReservedOrderId.php

@@ -0,0 +1,48 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\TestFramework\Quote\Model;
+
+use Magento\Framework\Api\SearchCriteriaBuilder;
+use Magento\Quote\Api\CartRepositoryInterface;
+use Magento\Quote\Api\Data\CartInterface;
+
+/**
+ * Search and return quote by reserved order id.
+ */
+class GetQuoteByReservedOrderId
+{
+    /** @var SearchCriteriaBuilder */
+    private $searchCriteriaBuilder;
+
+    /** @var CartRepositoryInterface */
+    private $cartRepository;
+
+    /**
+     * @param SearchCriteriaBuilder $searchCriteriaBuilder
+     * @param CartRepositoryInterface $cartRepository
+     */
+    public function __construct(SearchCriteriaBuilder $searchCriteriaBuilder, CartRepositoryInterface $cartRepository)
+    {
+        $this->searchCriteriaBuilder = $searchCriteriaBuilder;
+        $this->cartRepository = $cartRepository;
+    }
+
+    /**
+     * Return quote by reserved order id.
+     *
+     * @param string $reservedOrderId
+     * @return CartInterface|null
+     */
+    public function execute(string $reservedOrderId): ?CartInterface
+    {
+        $searchCriteria = $this->searchCriteriaBuilder->addFilter('reserved_order_id', $reservedOrderId)->create();
+        $quote = $this->cartRepository->getList($searchCriteria)->getItems();
+
+        return array_shift($quote);
+    }
+}

dev/tests/integration/testsuite/Magento/Quote/_files/quote_for_customer_and_custom_store.php

@@ -0,0 +1,49 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+use Magento\Quote\Api\CartRepositoryInterface;
+use Magento\Quote\Model\Quote;
+use Magento\Quote\Model\Quote\Address;
+use Magento\Store\Api\StoreRepositoryInterface;
+use Magento\TestFramework\Helper\Bootstrap;
+use Magento\TestFramework\ObjectManager;
+use Magento\Customer\Api\CustomerRepositoryInterface;
+
+/** @var ObjectManager $objectManager */
+$objectManager = Bootstrap::getObjectManager();
+
+$customer = $objectManager->get(CustomerRepositoryInterface::class)->get('[email protected]');
+$store = $objectManager->get(StoreRepositoryInterface::class)->get('secondstore');
+
+$addressData = include __DIR__ . '/../../Customer/Fixtures/address_data.php';
+/** @var Address $shippingAddress */
+$shippingAddress = $objectManager->create(Address::class, ['data' => $addressData[0]]);
+$shippingAddress->setAddressType('shipping');
+
+$billingAddress = clone $shippingAddress;
+$billingAddress->setId(null)
+    ->setAddressType('billing');
+
+/** @var Quote $quote */
+$quote = $objectManager->create(
+    Quote::class,
+    [
+        'data' => [
+            'customer_id' => $customer->getId(),
+            'store_id' => $store->getId(),
+            'reserved_order_id' => 'tsg-123456789',
+            'is_active' => true,
+            'is_multishipping' => false
+        ],
+    ]
+);
+$quote->setShippingAddress($shippingAddress)
+    ->setBillingAddress($billingAddress);
+
+/** @var CartRepositoryInterface $repository */
+$repository = $objectManager->get(CartRepositoryInterface::class);
+$repository->save($quote);

dev/tests/integration/testsuite/Magento/Quote/_files/quote_for_customer_and_custom_store_rollback.php

@@ -0,0 +1,29 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+use Magento\Framework\Registry;
+use Magento\Quote\Api\CartRepositoryInterface;
+use Magento\TestFramework\Helper\Bootstrap;
+use Magento\TestFramework\Quote\Model\GetQuoteByReservedOrderId;
+
+/** @var ObjectManager $objectManager */
+$objectManager = Bootstrap::getObjectManager();
+
+/** @var Registry $registry */
+$registry = $objectManager->get(Registry::class);
+$registry->unregister('isSecureArea');
+$registry->register('isSecureArea', true);
+
+$quote = $objectManager->get(GetQuoteByReservedOrderId::class)->execute('tsg-123456789');
+if ($quote !== null) {
+    /** @var CartRepositoryInterface $cartRepository */
+    $cartRepository = $objectManager->get(CartRepositoryInterface::class);
+    $cartRepository->delete($quote);
+}
+
+$registry->unregister('isSecureArea');
+$registry->register('isSecureArea', false);