test capture exit status

greblov · greblov · commit b8c5901e8c17 · 2019-02-14T11:24:18.000+02:00
diff --git a/pkg/enqueue/Consumption/Extension/CaptureExitStatusExtension.php b/pkg/enqueue/Consumption/Extension/CaptureExitStatusExtension.php
@@ -3,9 +3,9 @@
 namespace Enqueue\Consumption\Extension;
 
 use Enqueue\Consumption\Context\End;
-use Enqueue\Consumption\EndExtensionInterface;
+use Enqueue\Consumption\ExtensionInterface;
 
-class CaptureExitStatusExtension implements EndExtensionInterface
+class CaptureExitStatusExtension implements ExtensionInterface
 {
     /**
      * @var int
@@ -32,4 +32,40 @@ public function isExitStatusCaptured(): bool
     {
         return $this->isExitStatusCaptured;
     }
+
+    public function onInitLogger(\Enqueue\Consumption\Context\InitLogger $context): void
+    {
+    }
+
+    public function onMessageReceived(\Enqueue\Consumption\Context\MessageReceived $context): void
+    {
+    }
+
+    public function onPostConsume(\Enqueue\Consumption\Context\PostConsume $context): void
+    {
+    }
+
+    public function onPostMessageReceived(\Enqueue\Consumption\Context\PostMessageReceived $context): void
+    {
+    }
+
+    public function onPreConsume(\Enqueue\Consumption\Context\PreConsume $context): void
+    {
+    }
+
+    public function onPreSubscribe(\Enqueue\Consumption\Context\PreSubscribe $context): void
+    {
+    }
+
+    public function onProcessorException(\Enqueue\Consumption\Context\ProcessorException $context): void
+    {
+    }
+
+    public function onResult(\Enqueue\Consumption\Context\MessageResult $context): void
+    {
+    }
+
+    public function onStart(\Enqueue\Consumption\Context\Start $context): void
+    {
+    }
 }
diff --git a/pkg/enqueue/Tests/Consumption/QueueConsumerTest.php b/pkg/enqueue/Tests/Consumption/QueueConsumerTest.php
@@ -16,6 +16,7 @@
 use Enqueue\Consumption\Context\ProcessorException;
 use Enqueue\Consumption\Context\Start;
 use Enqueue\Consumption\Exception\InvalidArgumentException;
+use Enqueue\Consumption\Extension\CaptureExitStatusExtension;
 use Enqueue\Consumption\ExtensionInterface;
 use Enqueue\Consumption\QueueConsumer;
 use Enqueue\Consumption\Result;
@@ -1429,6 +1430,30 @@ public function testShouldCallProcessorAsMessageComeAlong()
         $this->assertSame($fooConsumerStub, $actualContexts[2]->getConsumer());
     }
 
+    public function testCaptureExitStatus()
+    {
+        $testExitCode = 5;
+
+        $stubExtension = $this->createExtension();
+
+        $stubExtension
+            ->expects($this->once())
+            ->method('onStart')
+            ->with($this->isInstanceOf(Start::class))
+            ->willReturnCallback(function (Start $context) use ($testExitCode) {
+                $context->interruptExecution($testExitCode);
+            })
+        ;
+
+        $exitExtension = new CaptureExitStatusExtension();
+
+        $consumer = new QueueConsumer($this->createContextStub(), $stubExtension);
+        $consumer->consume($exitExtension);
+
+        $this->assertEquals($testExitCode, $exitExtension->getExitStatus());
+        $this->assertTrue($exitExtension->isExitStatusCaptured());
+    }
+
     /**
      * @return \PHPUnit_Framework_MockObject_MockObject
      */
