Updating unit test

Test/Unit/Helper/PaymentResponseHandlerTest.php

@@ -10,8 +10,6 @@
  */
 namespace Adyen\Payment\Test\Unit\Helper;
 
-namespace Adyen\Payment\Test\Unit\Helper;
-
 use Adyen\Payment\Helper\PaymentResponseHandler;
 use Adyen\Payment\Logger\AdyenLogger;
 use Adyen\Payment\Helper\Vault;
@@ -29,9 +27,9 @@
 use Magento\Sales\Model\OrderRepository;
 use Magento\Sales\Model\Order\Status\HistoryFactory;
 use Adyen\Payment\Helper\StateData;
+use Adyen\Payment\Model\ResourceModel\PaymentResponse\Collection;
 use Adyen\Payment\Model\ResourceModel\PaymentResponse\CollectionFactory;
 use Adyen\Payment\Helper\Config as Config;
-use \Magento\Framework\Data\Collection\AbstractDb;
 
 class PaymentResponseHandlerTest extends AbstractAdyenTestCase
 {
@@ -63,26 +61,23 @@ protected function setUp(): void
             'create'
         ]);
         $this->stateDataHelperMock = $this->createMock(StateData::class);
-        $this->paymentResponseCollectionFactoryMock = $this->createMock(CollectionFactory::class);
         $this->configHelperMock = $this->createMock(Config::class);
 
-        // Mock for PaymentResponseCollection
-        $this->paymentResponseCollectionMock = $this->createMock(AbstractDb::class);
-
+        $this->paymentResponseMockForFactory = $this->createMock(Collection::class);
         // Mock PaymentResponseCollectionFactory to return the mocked collection
-        $this->paymentResponseCollectionFactoryMock->method('create')
-            ->willReturn($this->paymentResponseCollectionMock);
 
+        $this->paymentResponseCollectionFactoryMock = $this->createGeneratedMock(CollectionFactory::class, ['create']);
+        $this->paymentResponseCollectionFactoryMock->method('create')->willReturn($this->paymentResponseMockForFactory);
         // Mock addFieldToFilter behavior
-        $this->paymentResponseCollectionMock->method('addFieldToFilter')
+        $this->paymentResponseMockForFactory->method('addFieldToFilter')
             ->willReturnSelf();
 
         // Mock getSize to return a desired value
-        $this->paymentResponseCollectionMock->method('getSize')
+        $this->paymentResponseMockForFactory->method('getSize')
             ->willReturn(1); // Adjust based on your test case logic
 
         // Mock getData to return some dummy data
-        $this->paymentResponseCollectionMock->method('getData')
+        $this->paymentResponseMockForFactory->method('getData')
             ->willReturn([['field' => 'value']]);
 
         $orderHistory = $this->createMock(History::class);