Add unit tests

Author: andrewmy

pkg/sns/Tests/Spec/SnsContextTest.php

@@ -2,20 +2,49 @@
 
 namespace Enqueue\Sns\Tests\Spec;
 
+use Aws\Result;
 use Enqueue\Sns\SnsClient;
 use Enqueue\Sns\SnsContext;
+use Enqueue\Sns\SnsDestination;
+use Enqueue\Sns\SnsSubscribe;
 use Interop\Queue\Spec\ContextSpec;
 
 class SnsContextTest extends ContextSpec
 {
-    public function testShouldCreateConsumerOnCreateConsumerMethodCall()
+    public function testShouldCreateConsumerOnCreateConsumerMethodCall(): void
     {
         $this->expectException(\LogicException::class);
         $this->expectExceptionMessage('SNS transport does not support consumption. You should consider using SQS instead.');
 
         parent::testShouldCreateConsumerOnCreateConsumerMethodCall();
     }
 
+    public function testSetsSubscriptionAttributes(): void
+    {
+        $client = $this->createMock(SnsClient::class);
+        $client->expects($this->once())
+            ->method('listSubscriptionsByTopic')
+            ->willReturn(new Result(['Subscriptions' => [
+                ['SubscriptionArn' => 'arn1'],
+                ['SubscriptionArn' => 'arn2'],
+            ]]));
+        $client->expects($this->exactly(2))
+            ->method('setSubscriptionAttributes')
+            ->withConsecutive(
+                [$this->equalTo(['attr1' => 'value1', 'SubscriptionArn' => 'arn1'])],
+                [$this->equalTo(['attr1' => 'value1', 'SubscriptionArn' => 'arn2'])],
+            );
+
+        $context = new SnsContext($client, ['topic_arns' => ['topic1' => 'topicArn1']]);
+        $context->setSubscriptionAttributes(new SnsSubscribe(
+            new SnsDestination('topic1'),
+            'endpoint1',
+            'protocol1',
+            false,
+            ['attr1' => 'value1'],
+        ));
+    }
+
     protected function createContext()
     {
         $client = $this->createMock(SnsClient::class);

pkg/snsqs/Tests/Spec/SnsQsContextTest.php

@@ -3,13 +3,47 @@
 namespace Enqueue\SnsQs\Tests\Spec;
 
 use Enqueue\Sns\SnsContext;
+use Enqueue\Sns\SnsSubscribe;
 use Enqueue\SnsQs\SnsQsContext;
+use Enqueue\SnsQs\SnsQsQueue;
+use Enqueue\SnsQs\SnsQsTopic;
 use Enqueue\Sqs\SqsConsumer;
 use Enqueue\Sqs\SqsContext;
 use Interop\Queue\Spec\ContextSpec;
 
 class SnsQsContextTest extends ContextSpec
 {
+    public function testSetsSubscriptionAttributes(): void
+    {
+        $topic = new SnsQsTopic('topic1');
+
+        $snsContext = $this->createMock(SnsContext::class);
+        $snsContext->expects($this->once())
+            ->method('setSubscriptionAttributes')
+            ->with($this->equalTo(new SnsSubscribe(
+                $topic,
+                'queueArn1',
+                'sqs',
+                false,
+                ['attr1' => 'value1'],
+            )));
+
+        $sqsContext = $this->createMock(SqsContext::class);
+        $sqsContext->expects($this->any())
+            ->method('createConsumer')
+            ->willReturn($this->createMock(SqsConsumer::class));
+        $sqsContext->expects($this->any())
+            ->method('getQueueArn')
+            ->willReturn('queueArn1');
+
+        $context = new SnsQsContext($snsContext, $sqsContext);
+        $context->setSubscriptionAttributes(
+            $topic,
+            new SnsQsQueue('queue1'),
+            ['attr1' => 'value1'],
+        );
+    }
+
     protected function createContext()
     {
         $sqsContext = $this->createMock(SqsContext::class);