MAGETWO-56126: Login failed after new custom attribute was added

- unit test for CustomerAuthUpdate

Author: cpartica

app/code/Magento/Customer/Model/CustomerAuthUpdate.php

@@ -45,11 +45,11 @@ public function saveAuth($customerId)
 
         $this->customerResourceModel->getConnection()->update(
             $this->customerResourceModel->getTable('customer_entity'),
-            array(
+            [
                 'failures_num' => $customerSecure->getData('failures_num'),
                 'first_failure' => $customerSecure->getData('first_failure'),
                 'lock_expires' => $customerSecure->getData('lock_expires'),
-            ),
+            ],
             $this->customerResourceModel->getConnection()->quoteInto('entity_id = ?', $customerId)
         );
 

app/code/Magento/Customer/Test/Unit/Model/CustomerAuthUpdateTest.php

@@ -0,0 +1,115 @@
+<?php
+/**
+ * Copyright © 2016 Magento. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+namespace Magento\Test\Unit\Customer\Model;
+
+use Magento\Customer\Model\CustomerAuthUpdate;
+
+/**
+ * Class CustomerAuthUpdateTest
+ */
+class CustomerAuthUpdateTest extends \PHPUnit_Framework_TestCase
+{
+    /**
+     * @var CustomerAuthUpdate
+     */
+    protected $model;
+    /**
+     * @var \Magento\Customer\Model\CustomerRegistry|\PHPUnit_Framework_MockObject_MockObject
+     */
+    protected $customerRegistry;
+
+    /**
+     * @var \Magento\Customer\Model\ResourceModel\Customer|\PHPUnit_Framework_MockObject_MockObject
+     */
+    protected $customerResourceModel;
+
+    /**
+     * @var \Magento\Framework\TestFramework\Unit\Helper\ObjectManager
+     */
+    protected $objectManager;
+
+    /**
+     * Setup the test
+     */
+    protected function setUp()
+    {
+        $this->objectManager = new \Magento\Framework\TestFramework\Unit\Helper\ObjectManager($this);
+
+        $className = '\Magento\Customer\Model\CustomerRegistry';
+        $this->customerRegistry = $this->getMock($className, [], [], '', false);
+
+        $className = '\Magento\Customer\Model\ResourceModel\Customer';
+        $this->customerResourceModel = $this->getMock($className, [], [], '', false);
+
+        $this->model = $this->objectManager->getObject(
+            '\Magento\Customer\Model\CustomerAuthUpdate',
+            [
+                'customerRegistry' => $this->customerRegistry,
+                'customerResourceModel' => $this->customerResourceModel,
+            ]
+        );
+    }
+
+    /**
+     * test SaveAuth
+     */
+    public function testSaveAuth()
+    {
+        $customerId = 1;
+
+        $customerSecureMock = $this->getMock(
+            \Magento\Customer\Model\Data\CustomerSecure::class,
+            [],
+            [],
+            '',
+            false
+        );
+
+        $dbAdapter = $this->getMock(
+            \Magento\Framework\DB\Adapter\AdapterInterface::class,
+            [],
+            [],
+            '',
+            false
+        );
+
+        $this->customerRegistry->expects($this->once())
+            ->method('retrieveSecureData')
+            ->willReturn($customerSecureMock);
+
+        $customerSecureMock->expects($this->exactly(3))
+            ->method('getData')
+            ->willReturn(1);
+
+        $this->customerResourceModel->expects($this->any())
+            ->method('getConnection')
+            ->willReturn($dbAdapter);
+
+        $this->customerResourceModel->expects($this->any())
+            ->method('getTable')
+            ->willReturn('customer_entity');
+
+        $dbAdapter->expects($this->any())
+            ->method('update')
+            ->with(
+                'customer_entity',
+                [
+                    'failures_num' => 1,
+                    'first_failure' => 1,
+                    'lock_expires' => 1
+                ]
+            );
+
+        $dbAdapter->expects($this->any())
+            ->method('quoteInto')
+            ->with(
+                'entity_id = ?',
+                $customerId
+            );
+
+        $this->model->saveAuth($customerId);
+    }
+}