77
88namespace Magento \GraphQl \Quote \Customer ;
99
10+ use Magento \Framework \Exception \NoSuchEntityException ;
11+ use Magento \Framework \Registry ;
1012use Magento \GraphQl \Quote \GetMaskedQuoteIdByReservedOrderId ;
1113use Magento \Integration \Api \CustomerTokenServiceInterface ;
14+ use Magento \Sales \Api \OrderRepositoryInterface ;
15+ use Magento \Sales \Model \ResourceModel \Order \CollectionFactory ;
1216use Magento \TestFramework \Helper \Bootstrap ;
1317use Magento \TestFramework \TestCase \GraphQlAbstract ;
1418
@@ -27,6 +31,21 @@ class PlaceOrderTest extends GraphQlAbstract
2731 */
2832 private $ getMaskedQuoteIdByReservedOrderId ;
2933
34+ /**
35+ * @var CollectionFactory
36+ */
37+ private $ orderCollectionFactory ;
38+
39+ /**
40+ * @var OrderRepositoryInterface
41+ */
42+ private $ orderRepository ;
43+
44+ /**
45+ * @var Registry
46+ */
47+ private $ registry ;
48+
3049 /**
3150 * @inheritdoc
3251 */
@@ -35,6 +54,9 @@ protected function setUp()
3554 $ objectManager = Bootstrap::getObjectManager ();
3655 $ this ->getMaskedQuoteIdByReservedOrderId = $ objectManager ->get (GetMaskedQuoteIdByReservedOrderId::class);
3756 $ this ->customerTokenService = $ objectManager ->get (CustomerTokenServiceInterface::class);
57+ $ this ->orderCollectionFactory = $ objectManager ->get (CollectionFactory::class);
58+ $ this ->orderRepository = $ objectManager ->get (OrderRepositoryInterface::class);
59+ $ this ->registry = Bootstrap::getObjectManager ()->get (Registry::class);
3860 }
3961
4062 /**
@@ -253,4 +275,22 @@ private function getHeaderMap(string $username = 'customer@example.com', string
253275 $ headerMap = ['Authorization ' => 'Bearer ' . $ customerToken ];
254276 return $ headerMap ;
255277 }
278+
279+ /**
280+ * @inheritdoc
281+ */
282+ public function tearDown ()
283+ {
284+ $ this ->registry ->unregister ('isSecureArea ' );
285+ $ this ->registry ->register ('isSecureArea ' , true );
286+
287+ $ orderCollection = $ this ->orderCollectionFactory ->create ();
288+ foreach ($ orderCollection as $ order ) {
289+ $ this ->orderRepository ->delete ($ order );
290+ }
291+ $ this ->registry ->unregister ('isSecureArea ' );
292+ $ this ->registry ->register ('isSecureArea ' , false );
293+
294+ parent ::tearDown ();
295+ }
256296}
0 commit comments