Improve coverage for FailedJobsListener

Author: ajibarra

tests/TestCase/Listener/FailedJobsListenerTest.php

@@ -18,8 +18,12 @@
 
 use Cake\Event\Event;
 use Cake\Event\EventManager;
+use Cake\ORM\Entity;
+use Cake\ORM\Exception\PersistenceFailedException;
+use Cake\ORM\Locator\TableLocator;
 use Cake\Queue\Job\Message;
 use Cake\Queue\Listener\FailedJobsListener;
+use Cake\Queue\Model\Table\FailedJobsTable;
 use Cake\Queue\QueueManager;
 use Cake\TestSuite\TestCase;
 use Enqueue\Null\NullConnectionFactory;
@@ -92,4 +96,75 @@ public function testFailedJobIsAddedWhenEventIsFired()
         $this->assertSame('example_queue', $failedJob->queue);
         $this->assertStringContainsString('some message', $failedJob->exception);
     }
+
+    /**
+     * Data provider for testStoreFailedJobException
+     *
+     * @return array[]
+     */
+    public function storeFailedJobExceptionDataProvider()
+    {
+        return [
+            [['exception' => 'some message'], '`logger` was not defined on Consumption.LimitAttemptsExtension.failed event'],
+            [['exception' => 'some message', 'logger' => new \stdClass()], '`logger` is not an instance of `LoggerInterface` on Consumption.LimitAttemptsExtension.failed event.'],
+        ];
+    }
+
+    /**
+     * @dataProvider storeFailedJobExceptionDataProvider
+     * @return void
+     */
+    public function testStoreFailedJobException($eventData, $exceptionMessage)
+    {
+        $tableLocator = $this
+            ->getMockBuilder(TableLocator::class)
+            ->onlyMethods(['get'])
+            ->getMock();
+        $failedJobsTable = $this
+            ->getMockBuilder(FailedJobsTable::class)
+            ->onlyMethods(['saveOrFail'])
+            ->getMock();
+        $failedJobsListener = $this
+            ->getMockBuilder(FailedJobsListener::class)
+            ->onlyMethods(['getTableLocator'])
+            ->getMock();
+        $failedJobsTable->expects($this->once())
+            ->method('saveOrFail')
+            ->willThrowException(new PersistenceFailedException(new Entity(), 'Persistence Failed'));
+        $tableLocator->expects($this->once())
+            ->method('get')
+            ->with('Cake/Queue.FailedJobs')
+            ->willReturn($failedJobsTable);
+        $failedJobsListener->expects($this->once())
+            ->method('getTableLocator')
+            ->willReturn($tableLocator);
+
+        $parsedBody = [
+            'class' => [LogToDebugJob::class, 'execute'],
+            'data' => ['example_key' => 'example_value'],
+            'requeueOptions' => [
+                'config' => 'example_config',
+                'priority' => 'example_priority',
+                'queue' => 'example_queue',
+            ],
+        ];
+        $messageBody = json_encode($parsedBody);
+        $connectionFactory = new NullConnectionFactory();
+
+        $context = $connectionFactory->createContext();
+        $originalMessage = new NullMessage($messageBody);
+        $message = new Message($originalMessage, $context);
+
+        $event = new Event(
+            'Consumption.LimitAttemptsExtension.failed',
+            $message,
+            $eventData
+        );
+
+        $this->expectException(\RuntimeException::class);
+        $this->expectExceptionMessage($exceptionMessage);
+
+        EventManager::instance()->on($failedJobsListener);
+        EventManager::instance()->dispatch($event);
+    }
 }