From 9b1274cce5c9563b75c929b49cd5a455df60ab0f Mon Sep 17 00:00:00 2001
From: rliukshyn <rliukshyn@magento.com>
Date: Tue, 15 Nov 2016 11:40:22 +0200
Subject: [PATCH] MTA-3790: Add variation with PayPal Payflow Pro method to One
 Page Checkout test

---
 .../Vault/Test/Block/Onepage/Payment/Method/Vault.php       | 4 ++--
 .../Vault/Test/TestStep/CheckSaveCreditCardOptionStep.php   | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/dev/tests/functional/tests/app/Magento/Vault/Test/Block/Onepage/Payment/Method/Vault.php b/dev/tests/functional/tests/app/Magento/Vault/Test/Block/Onepage/Payment/Method/Vault.php
index 75257b37a83..c416381a0b2 100644
--- a/dev/tests/functional/tests/app/Magento/Vault/Test/Block/Onepage/Payment/Method/Vault.php
+++ b/dev/tests/functional/tests/app/Magento/Vault/Test/Block/Onepage/Payment/Method/Vault.php
@@ -38,7 +38,7 @@ class Vault extends Method
     public function saveCreditCard($paymentMethod, $creditCardSave)
     {
         $saveCard = sprintf($this->vaultCheckbox, $paymentMethod);
-        $this->_rootElement->find($saveCard, 'checkbox')->setValue($creditCardSave);
+        $this->_rootElement->find($saveCard, Locator::SELECTOR_CSS, 'checkbox')->setValue($creditCardSave);
     }
 
     /**
@@ -50,7 +50,7 @@ class Vault extends Method
     public function isVaultVisible($paymentMethod)
     {
         $saveCard = sprintf($this->vaultCheckbox, $paymentMethod);
-        return $this->_rootElement->find($saveCard, 'checkbox')->isVisible();
+        return $this->_rootElement->find($saveCard, Locator::SELECTOR_CSS, 'checkbox')->isVisible();
     }
 
     /**
diff --git a/dev/tests/functional/tests/app/Magento/Vault/Test/TestStep/CheckSaveCreditCardOptionStep.php b/dev/tests/functional/tests/app/Magento/Vault/Test/TestStep/CheckSaveCreditCardOptionStep.php
index 168f7da10a5..7949cd0295a 100644
--- a/dev/tests/functional/tests/app/Magento/Vault/Test/TestStep/CheckSaveCreditCardOptionStep.php
+++ b/dev/tests/functional/tests/app/Magento/Vault/Test/TestStep/CheckSaveCreditCardOptionStep.php
@@ -38,7 +38,7 @@ class checkSaveCreditCardOptionStep implements TestStepInterface
     /**
      * If vault is enabled for payment method.
      *
-     * @var bool
+     * @var null|bool
      */
     private $isVaultEnabled;
 
@@ -46,13 +46,13 @@ class checkSaveCreditCardOptionStep implements TestStepInterface
      * @param CheckoutOnepage $checkoutOnepage
      * @param AssertSaveCreditCardOptionNotPresent $assertSaveCreditCardOptionNotPresent
      * @param array $payment
-     * @param bool $isVaultEnabled
+     * @param null|bool $isVaultEnabled
      */
     public function __construct(
         CheckoutOnepage $checkoutOnepage,
         AssertSaveCreditCardOptionNotPresent $assertSaveCreditCardOptionNotPresent,
         array $payment,
-        $isVaultEnabled
+        $isVaultEnabled = null
     ) {
         $this->checkoutOnepage = $checkoutOnepage;
         $this->assertSaveCreditCardOptionNotPresent = $assertSaveCreditCardOptionNotPresent;
-- 
GitLab