[consumption] Add test for Extension::onResult method.

php-enqueue · Apr 7, 2017 · 0bc22c2 · 0bc22c2
1 parent da4dfcd
commit 0bc22c2
Show file tree

Hide file tree

Showing 4 changed files with 166 additions and 3 deletions.
diff --git a/pkg/enqueue/Consumption/ChainExtension.php b/pkg/enqueue/Consumption/ChainExtension.php
@@ -49,6 +49,16 @@ public function onPreReceived(Context $context)
         }
     }
 
+    /**
+     * @param Context $context
+     */
+    public function onResult(Context $context)
+    {
+        foreach ($this->extensions as $extension) {
+            $extension->onResult($context);
+        }
+    }
+
     /**
      * @param Context $context
      */

diff --git a/...ueue/Tests/Consumption/ExtensionsTest.php → .../Tests/Consumption/ChainExtensionTest.php b/...ueue/Tests/Consumption/ExtensionsTest.php → .../Tests/Consumption/ChainExtensionTest.php
@@ -7,7 +7,7 @@
 use Enqueue\Consumption\ExtensionInterface;
 use Enqueue\Test\ClassExtensionTrait;
 
-class ExtensionsTest extends \PHPUnit_Framework_TestCase
+class ChainExtensionTest extends \PHPUnit_Framework_TestCase
 {
     use ClassExtensionTrait;
 
@@ -87,6 +87,28 @@ public function testShouldProxyOnPreReceiveToAllInternalExtensions()
         $extensions->onPreReceived($context);
     }
 
+    public function testShouldProxyOnResultToAllInternalExtensions()
+    {
+        $context = $this->createContextMock();
+
+        $fooExtension = $this->createExtension();
+        $fooExtension
+            ->expects($this->once())
+            ->method('onResult')
+            ->with($this->identicalTo($context))
+        ;
+        $barExtension = $this->createExtension();
+        $barExtension
+            ->expects($this->once())
+            ->method('onResult')
+            ->with($this->identicalTo($context))
+        ;
+
+        $extensions = new ChainExtension([$fooExtension, $barExtension]);
+
+        $extensions->onResult($context);
+    }
+
     public function testShouldProxyOnPostReceiveToAllInternalExtensions()
     {
         $context = $this->createContextMock();

diff --git a/pkg/enqueue/Tests/Consumption/EmptyExtensionTraitTest.php b/pkg/enqueue/Tests/Consumption/EmptyExtensionTraitTest.php
@@ -0,0 +1,19 @@
+<?php
+namespace Enqueue\Tests\Consumption;
+
+use Enqueue\Consumption\EmptyExtensionTrait;
+use Enqueue\Consumption\ExtensionInterface;
+use PHPUnit\Framework\TestCase;
+
+class EmptyExtensionTraitTest extends TestCase
+{
+    public function testTraitMustImplementOrExtensionMethods()
+    {
+        new EmptyExtension();
+    }
+}
+
+class EmptyExtension implements ExtensionInterface
+{
+    use EmptyExtensionTrait;
+}
diff --git a/pkg/enqueue/Tests/Consumption/QueueConsumerTest.php b/pkg/enqueue/Tests/Consumption/QueueConsumerTest.php
@@ -467,7 +467,7 @@ public function testShouldCallOnBeforeReceiveExtensionMethod()
         $queueConsumer->consume();
     }
 
-    public function testShouldCallOnPreReceivedAndPostReceivedExtensionMethods()
+    public function testShouldCallOnPreReceivedExtensionMethodWithExpectedContext()
     {
         $expectedMessage = $this->createMessageMock();
         $messageConsumerStub = $this->createMessageConsumerStub($expectedMessage);
@@ -497,6 +497,62 @@ public function testShouldCallOnPreReceivedAndPostReceivedExtensionMethods()
                 $this->assertFalse($context->isExecutionInterrupted());
             })
         ;
+
+        $chainExtensions = new ChainExtension([$extension, new BreakCycleExtension(1)]);
+        $queueConsumer = new QueueConsumer($contextStub, $chainExtensions, 0);
+        $queueConsumer->bind(new NullQueue('aQueueName'), $processorMock);
+
+        $queueConsumer->consume();
+    }
+
+    public function testShouldCallOnResultExtensionMethodWithExpectedContext()
+    {
+        $expectedMessage = $this->createMessageMock();
+        $messageConsumerStub = $this->createMessageConsumerStub($expectedMessage);
+
+        $contextStub = $this->createPsrContextStub($messageConsumerStub);
+
+        $processorMock = $this->createProcessorStub();
+
+        $extension = $this->createExtension();
+        $extension
+            ->expects($this->once())
+            ->method('onResult')
+            ->with($this->isInstanceOf(Context::class))
+            ->willReturnCallback(function (Context $context) use (
+                $contextStub,
+                $messageConsumerStub,
+                $processorMock,
+                $expectedMessage
+            ) {
+                $this->assertSame($contextStub, $context->getPsrContext());
+                $this->assertSame($messageConsumerStub, $context->getPsrConsumer());
+                $this->assertSame($processorMock, $context->getPsrProcessor());
+                $this->assertSame($expectedMessage, $context->getPsrMessage());
+                $this->assertInstanceOf(NullLogger::class, $context->getLogger());
+                $this->assertNull($context->getException());
+                $this->assertSame(Result::ACK, $context->getResult());
+                $this->assertFalse($context->isExecutionInterrupted());
+            })
+        ;
+
+        $chainExtensions = new ChainExtension([$extension, new BreakCycleExtension(1)]);
+        $queueConsumer = new QueueConsumer($contextStub, $chainExtensions, 0);
+        $queueConsumer->bind(new NullQueue('aQueueName'), $processorMock);
+
+        $queueConsumer->consume();
+    }
+
+    public function testShouldCallOnPostReceivedExtensionMethodWithExpectedContext()
+    {
+        $expectedMessage = $this->createMessageMock();
+        $messageConsumerStub = $this->createMessageConsumerStub($expectedMessage);
+
+        $contextStub = $this->createPsrContextStub($messageConsumerStub);
+
+        $processorMock = $this->createProcessorStub();
+
+        $extension = $this->createExtension();
         $extension
             ->expects($this->once())
             ->method('onPostReceived')
@@ -722,6 +778,57 @@ public function testShouldAllowInterruptConsumingOnPreReceiveButProcessCurrentMe
         $queueConsumer->consume();
     }
 
+    public function testShouldAllowInterruptConsumingOnResult()
+    {
+        $expectedMessage = $this->createMessageMock();
+        $messageConsumerStub = $this->createMessageConsumerStub($expectedMessage);
+
+        $contextStub = $this->createPsrContextStub($messageConsumerStub);
+
+        $processorMock = $this->createProcessorMock();
+        $processorMock
+            ->expects($this->once())
+            ->method('process')
+            ->willReturn(Result::ACK)
+        ;
+
+        $extension = $this->createExtension();
+        $extension
+            ->expects($this->once())
+            ->method('onResult')
+            ->with($this->isInstanceOf(Context::class))
+            ->willReturnCallback(function (Context $context) {
+                $context->setExecutionInterrupted(true);
+            })
+        ;
+        $extension
+            ->expects($this->atLeastOnce())
+            ->method('onInterrupted')
+            ->with($this->isInstanceOf(Context::class))
+            ->willReturnCallback(function (Context $context) use (
+                $contextStub,
+                $messageConsumerStub,
+                $processorMock,
+                $expectedMessage
+            ) {
+                $this->assertSame($contextStub, $context->getPsrContext());
+                $this->assertSame($messageConsumerStub, $context->getPsrConsumer());
+                $this->assertSame($processorMock, $context->getPsrProcessor());
+                $this->assertSame($expectedMessage, $context->getPsrMessage());
+                $this->assertInstanceOf(NullLogger::class, $context->getLogger());
+                $this->assertNull($context->getException());
+                $this->assertSame(Result::ACK, $context->getResult());
+                $this->assertTrue($context->isExecutionInterrupted());
+            })
+        ;
+
+        $chainExtensions = new ChainExtension([$extension, new BreakCycleExtension(1)]);
+        $queueConsumer = new QueueConsumer($contextStub, $chainExtensions, 0);
+        $queueConsumer->bind(new NullQueue('aQueueName'), $processorMock);
+
+        $queueConsumer->consume();
+    }
+
     public function testShouldAllowInterruptConsumingOnPostReceive()
     {
         $expectedMessage = $this->createMessageMock();
@@ -850,6 +957,11 @@ public function testShouldCallExtensionPassedOnRuntime()
             ->method('onPreReceived')
             ->with($this->isInstanceOf(Context::class))
         ;
+        $runtimeExtension
+            ->expects($this->once())
+            ->method('onResult')
+            ->with($this->isInstanceOf(Context::class))
+        ;
         $runtimeExtension
             ->expects($this->once())
             ->method('onPostReceived')
@@ -936,7 +1048,7 @@ public function testShouldCallEachQueueOneByOne()
             })
         ;
         $extension
-            ->expects($this->at(4))
+            ->expects($this->at(5))
             ->method('onBeforeReceive')
             ->with($this->isInstanceOf(Context::class))
             ->willReturnCallback(function (Context $context) use ($anotherProcessorMock, $queue2) {