Integration Test for CC Number Encryption Updater

Author: pmclain

dev/tests/integration/testsuite/Magento/Sales/Model/ResourceModel/Order/Payment/EncryptionUpdateTest.php

@@ -0,0 +1,50 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+namespace Magento\Sales\Model\ResourceModel\Order\Payment;
+
+use Magento\Framework\Encryption\Encryptor;
+use Magento\TestFramework\Helper\Bootstrap;
+
+class EncryptionUpdateTest extends \PHPUnit\Framework\TestCase
+{
+    const TEST_CC_NUMBER = '4111111111111111';
+
+    /**
+     * Tests re-encryption of credit card numbers
+     *
+     * @magentoDataFixture Magento/Sales/_files/payment_enc_cc.php
+     */
+    public function testReEncryptCreditCardNumbers()
+    {
+        $objectManager = Bootstrap::getObjectManager();
+
+        /** @var \Magento\Framework\Encryption\EncryptorInterface $encyptor */
+        $encyptor = $objectManager->get(\Magento\Framework\Encryption\EncryptorInterface::class);
+
+        /** @var \Magento\Sales\Model\ResourceModel\Order\Payment\EncryptionUpdate $resource */
+        $resource = $objectManager->create(\Magento\Sales\Model\ResourceModel\Order\Payment\EncryptionUpdate::class);
+        $resource->reEncryptCreditCardNumbers();
+
+        /** @var \Magento\Sales\Model\ResourceModel\Order\Payment\Collection $collection */
+        $collection = $objectManager->create(\Magento\Sales\Model\ResourceModel\Order\Payment\Collection::class);
+        $collection->addFieldToFilter('cc_number_enc', ['notnull' => true]);
+
+        $this->assertGreaterThan(0, $collection->getTotalCount());
+
+        /** @var \Magento\Sales\Model\Order\Payment $payment */
+        foreach ($collection->getItems() as $payment) {
+            $this->assertEquals(
+                static::TEST_CC_NUMBER,
+                $encyptor->decrypt($payment->getCcNumberEnc())
+            );
+            
+            $this->assertStringStartsWith('0:' . Encryptor::CIPHER_LATEST . ':', $payment->getCcNumberEnc());
+        }
+    }
+}

dev/tests/integration/testsuite/Magento/Sales/_files/payment_enc_cc.php

@@ -0,0 +1,53 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+
+declare(strict_types=1);
+
+use Magento\Framework\Api\SearchCriteriaBuilder;
+use Magento\Framework\Api\SearchCriteria;
+use Magento\Sales\Model\ResourceModel\Order\Payment\EncryptionUpdateTest;
+use Magento\Framework\App\DeploymentConfig;
+
+require 'order.php';
+
+$objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager();
+
+/** @var DeploymentConfig $deployConfig */
+$deployConfig = $objectManager->get(DeploymentConfig::class);
+
+/**
+ * Creates an encrypted card number with the current crypt key using
+ * a legacy cipher.
+ */
+// @codingStandardIgnoreStart
+$handle = @mcrypt_module_open(MCRYPT_RIJNDAEL_256, '', MCRYPT_MODE_CBC, '');
+$initVectorSize = @mcrypt_enc_get_iv_size($handle);
+$initVector = str_repeat("\0", $initVectorSize);
+@mcrypt_generic_init($handle, $deployConfig->get('crypt/key'), $initVector);
+
+$encCcNumber = @mcrypt_generic($handle, EncryptionUpdateTest::TEST_CC_NUMBER);
+
+@mcrypt_generic_deinit($handle);
+@mcrypt_module_close($handle);
+// @codingStandardIgnoreEnd
+
+/** @var SearchCriteria $searchCriteria */
+$searchCriteria = $objectManager->get(SearchCriteriaBuilder::class)
+    ->addFilter('increment_id', '100000001')
+    ->create();
+
+$orders = $orderRepository->getList($searchCriteria)->getItems();
+$order = array_pop($orders);
+
+/** @var \Magento\Sales\Model\ResourceModel\Order\Payment $resource */
+$resource = $objectManager->create(\Magento\Sales\Model\ResourceModel\Order\Payment::class);
+$resource->getConnection()->insert(
+    $resource->getMainTable(),
+    [
+        'parent_id' => $order->getId(),
+        'cc_number_enc' => '0:2:' . base64_encode($encCcNumber),
+    ]
+);