MAGETWO-38155: Create block for order level

Author: okarpenko

dev/tests/api-functional/testsuite/Magento/Integration/Model/CustomerTokenServiceTest.php

@@ -13,6 +13,7 @@
 use Magento\TestFramework\TestCase\WebapiAbstract;
 use Magento\User\Model\User as UserModel;
 use Magento\Framework\Webapi\Exception as HTTPExceptionCodes;
+use Magento\Integration\Model\Resource\Oauth\Token\CollectionFactory;
 
 /**
  * api-functional test for \Magento\Integration\Model\CustomerTokenService.
@@ -35,9 +36,9 @@ class CustomerTokenServiceTest extends WebapiAbstract
     private $customerAccountManagement;
 
     /**
-     * @var TokenModel
+     * @var CollectionFactory
      */
-    private $tokenModel;
+    private $tokenCollection;
 
     /**
      * @var UserModel
@@ -54,19 +55,21 @@ public function setUp()
         $this->customerAccountManagement = Bootstrap::getObjectManager()->get(
             'Magento\Customer\Api\AccountManagementInterface'
         );
-        $this->tokenModel = Bootstrap::getObjectManager()->get('Magento\Integration\Model\Oauth\Token');
+        $tokenCollectionFactory = Bootstrap::getObjectManager()->get(
+            'Magento\Integration\Model\Resource\Oauth\Token\CollectionFactory'
+        );
+        $this->tokenCollection = $tokenCollectionFactory->create();
         $this->userModel = Bootstrap::getObjectManager()->get('Magento\User\Model\User');
     }
 
     /**
      * @magentoApiDataFixture Magento/Customer/_files/customer.php
-     * @magentoAppIsolation enabled
-     * @magentoDbIsolation enabled
      */
     public function testCreateCustomerAccessToken()
     {
         $customerUserName = 'customer@example.com';
         $password = 'password';
+        $isTokenCorrect = false;
 
         $serviceInfo = [
             'rest' => [
@@ -78,9 +81,17 @@ public function testCreateCustomerAccessToken()
         $accessToken = $this->_webApiCall($serviceInfo, $requestData);
 
         $customerData = $this->customerAccountManagement->authenticate($customerUserName, $password);
-        /** @var $token TokenModel */
-        $token = $this->tokenModel->loadByCustomerId($customerData->getId())->getToken();
-        $this->assertEquals($accessToken, $token);
+
+        /** @var $this->tokenCollection \Magento\Integration\Model\Resource\Oauth\Token\Collection */
+        $this->tokenCollection->addFilterByCustomerId($customerData->getId());
+
+        foreach ($this->tokenCollection->getItems() as $item) {
+            /** @var $item TokenModel */
+            if ($item->getToken() == $accessToken) {
+                $isTokenCorrect = true;
+            }
+        }
+        $this->assertTrue($isTokenCorrect);
     }
 
     /**