MAGETWO-62556: Subscription Request

Author: fascinosum

app/code/Magento/Analytics/Test/Unit/Model/IntegrationManagerTest.php

@@ -70,6 +70,11 @@ public function setUp()
         );
     }
 
+    /**
+     * @param string $status
+     *
+     * @return array
+     */
     private function getIntegrationUserData($status)
     {
         return [
@@ -83,6 +88,9 @@ private function getIntegrationUserData($status)
         ];
     }
 
+    /**
+     * @return void
+     */
     public function testActivateIntegrationSuccess()
     {
         $this->integrationServiceMock->expects($this->once())
@@ -125,7 +133,13 @@ public function testActivateIntegrationFailureNoSuchEntity()
         $this->integrationManager->activateIntegration();
     }
 
-    public function testGetTokenNewIntegration()
+    /**
+     * @dataProvider integrationIdDataProvider
+     *
+     * @param int|null $integrationId If null integration is absent.
+     * @return void
+     */
+    public function testGetTokenNewIntegration($integrationId)
     {
         $this->configMock->expects($this->atLeastOnce())
             ->method('getConfigDataValue')
@@ -140,7 +154,14 @@ public function testGetTokenNewIntegration()
             ->willReturn(100500);
         $this->integrationMock->expects($this->once())
             ->method('getId')
-            ->willReturn(1);
+            ->willReturn($integrationId);
+        if (!$integrationId) {
+            $this->integrationServiceMock
+                ->expects($this->once())
+                ->method('create')
+                ->with($this->getIntegrationUserData(Integration::STATUS_INACTIVE))
+                ->willReturn($this->integrationMock);
+        }
         $this->oauthServiceMock->expects($this->at(0))
             ->method('getAccessToken')
             ->with(100500)
@@ -156,7 +177,13 @@ public function testGetTokenNewIntegration()
         $this->assertEquals('IntegrationToken', $this->integrationManager->generateToken());
     }
 
-    public function testGetTokenExistingIntegration()
+    /**
+     * @dataProvider integrationIdDataProvider
+     *
+     * @param int|null $integrationId If null integration is absent.
+     * @return void
+     */
+    public function testGetTokenExistingIntegration($integrationId)
     {
         $this->configMock->expects($this->atLeastOnce())
             ->method('getConfigDataValue')
@@ -171,7 +198,14 @@ public function testGetTokenExistingIntegration()
             ->willReturn(100500);
         $this->integrationMock->expects($this->once())
             ->method('getId')
-            ->willReturn(1);
+            ->willReturn($integrationId);
+        if (!$integrationId) {
+            $this->integrationServiceMock
+                ->expects($this->once())
+                ->method('create')
+                ->with($this->getIntegrationUserData(Integration::STATUS_INACTIVE))
+                ->willReturn($this->integrationMock);
+        }
         $this->oauthServiceMock->expects($this->once())
             ->method('getAccessToken')
             ->with(100500)
@@ -180,4 +214,15 @@ public function testGetTokenExistingIntegration()
             ->method('createAccessToken');
         $this->assertEquals('IntegrationToken', $this->integrationManager->generateToken());
     }
+
+    /**
+     * @return array
+     */
+    public function integrationIdDataProvider()
+    {
+        return [
+            [1],
+            [null],
+        ];
+    }
 }