MC-39109: Vault throw The requested Payment Method is not available. error

Author: rostyslav-hymon

app/code/Magento/Vault/Plugin/PaymentVaultInformationManagement.php

@@ -10,7 +10,7 @@
 use Magento\Checkout\Api\PaymentInformationManagementInterface;
 use Magento\Quote\Api\Data\AddressInterface;
 use Magento\Quote\Api\Data\PaymentInterface;
-use Magento\Vault\Model\PaymentMethodList;
+use Magento\Vault\Api\PaymentMethodListInterface;
 use Magento\Store\Model\StoreManagerInterface;
 
 /**
@@ -19,7 +19,7 @@
 class PaymentVaultInformationManagement
 {
     /**
-     * @var PaymentMethodList
+     * @var PaymentMethodListInterface
      */
     private $vaultPaymentMethodList;
 
@@ -31,11 +31,11 @@ class PaymentVaultInformationManagement
     /**
      * PaymentVaultInformationManagement constructor.
      *
-     * @param PaymentMethodList $vaultPaymentMethodList
+     * @param PaymentMethodListInterface $vaultPaymentMethodList
      * @param StoreManagerInterface $storeManager
      */
     public function __construct(
-        PaymentMethodList $vaultPaymentMethodList,
+        PaymentMethodListInterface $vaultPaymentMethodList,
         StoreManagerInterface $storeManager
     ) {
         $this->vaultPaymentMethodList = $vaultPaymentMethodList;

app/code/Magento/Vault/Test/Unit/Plugin/PaymentVaultInformationManagementTest.php

@@ -0,0 +1,130 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\Vault\Test\Unit\Plugin;
+
+use Magento\Checkout\Api\PaymentInformationManagementInterface;
+use Magento\Store\Api\Data\StoreInterface;
+use Magento\Store\Model\StoreManagerInterface;
+use Magento\Vault\Api\PaymentMethodListInterface;
+use Magento\Vault\Plugin\PaymentVaultInformationManagement;
+use Magento\Quote\Api\Data\PaymentInterface;
+use PHPUnit\Framework\MockObject\MockObject;
+use PHPUnit\Framework\TestCase;
+
+/**
+ * Test for payment vault information management plugin
+ */
+class PaymentVaultInformationManagementTest extends TestCase
+{
+    /**
+     * @var StoreManagerInterface|MockObject
+     */
+    private $storeManager;
+
+    /**
+     * @var StoreInterface|MockObject
+     */
+    private $store;
+
+    /**
+     * @var PaymentMethodListInterface|MockObject
+     */
+    private $paymentMethodList;
+
+    /**
+     * @var PaymentVaultInformationManagement
+     */
+    private $plugin;
+
+    /**
+     * @var PaymentInformationManagementInterface|MockObject
+     */
+    private $paymentInformationManagement;
+
+    /**
+     * @var PaymentInterface|MockObject
+     */
+    private $payment;
+
+    /**
+     * @inheritDoc
+     */
+    protected function setUp(): void
+    {
+        $this->storeManager = $this->getMockBuilder(StoreManagerInterface::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['getStore'])
+            ->getMockForAbstractClass();
+        $this->store = $this->getMockBuilder(StoreInterface::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['getId'])
+            ->getMockForAbstractClass();
+        $this->paymentMethodList = $this->getMockBuilder(PaymentMethodListInterface::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['getActiveList'])
+            ->getMockForAbstractClass();
+        $this->paymentInformationManagement = $this
+            ->getMockBuilder(PaymentInformationManagementInterface::class)
+            ->disableOriginalConstructor()
+            ->getMockForAbstractClass();
+        $this->payment = $this->getMockBuilder(PaymentInterface::class)
+            ->onlyMethods(['setMethod'])
+            ->disableOriginalConstructor()
+            ->getMockForAbstractClass();
+        $this->plugin = new PaymentVaultInformationManagement($this->paymentMethodList, $this->storeManager);
+    }
+
+    /**
+     * Test payment method for vault before saving payment information
+     *
+     * @param string $requestPaymentMethodCode
+     * @param string $methodCode
+     * @dataProvider vaultPaymentMethodDataProvider
+     *
+     * @return void
+     */
+    public function testBeforeSavePaymentInformation($requestPaymentMethodCode, $methodCode): void
+    {
+        $this->store->method('getId')
+            ->willReturn(1);
+        $this->storeManager->method('getStore')
+            ->willReturn($this->store);
+        $activeVaultMethod = $this->getMockBuilder(PaymentInterface::class)
+            ->disableOriginalConstructor()
+            ->addMethods(['getCode', 'getProviderCode'])
+            ->getMockForAbstractClass();
+        $activeVaultMethod->method('getCode')
+            ->willReturn($methodCode);
+        $this->paymentMethodList->method('getActiveList')
+            ->willReturn([$activeVaultMethod]);
+        $this->payment->method('getMethod')
+            ->willReturn($requestPaymentMethodCode);
+        $this->payment->expects($this->once())
+            ->method('setMethod')
+            ->with($methodCode);
+
+        $this->plugin->beforeSavePaymentInformation(
+            $this->paymentInformationManagement,
+            '1',
+            $this->payment,
+            null
+        );
+    }
+
+    /**
+     * Data provider for BeforeSavePaymentInformation.
+     *
+     * @return array
+     */
+    public function vaultPaymentMethodDataProvider(): array
+    {
+        return [
+            ['braintree_cc_vault_01', 'braintree_cc_vault'],
+        ];
+    }
+}