OXIESEC PANEL

	Name	Size	Modified	Perms
	📁 ..	-	11/12/2024 10:59:38 AM	rwxr-xr-x
	📄 SqsClientTest.php	6.19 KB	07/30/2024 12:09:12 PM	rw-r--r--

Editing: SqsClientTest.php

<?php
namespace Aws\Test\Sqs;

use Aws\Middleware;
use Aws\Result;
use Aws\Sqs\SqsClient;
use Aws\Test\UsesServiceTrait;
use Yoast\PHPUnitPolyfills\TestCases\TestCase;

/**
 * @covers Aws\Sqs\SqsClient
 */
class SqsClientTest extends TestCase
{
    use UsesServiceTrait;

public function testGetQueueArn()
    {
        $url = 'https://sqs.us-east-1.amazonaws.com/057737625318/php-integ-sqs-queue-1359765974';
        $arn = 'arn:aws:sqs:us-east-1:057737625318:php-integ-sqs-queue-1359765974';
        $sqs = new SqsClient([
            'region'  => 'us-east-1',
            'version' => 'latest'
        ]);
        $this->assertSame($arn, $sqs->getQueueArn($url));
    }

public function testFifoQueueArn()
    {
        $url = 'https://sqs.us-east-1.amazonaws.com/057737625318/php-integ-sqs-queue-1359765974.fifo';
        $arn = 'arn:aws:sqs:us-east-1:057737625318:php-integ-sqs-queue-1359765974.fifo';
        $sqs = new SqsClient([
            'region'  => 'us-east-1',
            'version' => 'latest'
        ]);
        $this->assertSame($arn, $sqs->getQueueArn($url));
    }

public function testValidatesMd5OfBody()
    {
        $this->expectExceptionMessage("MD5 mismatch. Expected foo, found ddc35f88fa71b6ef142ae61f35364653");
        $this->expectException(\Aws\Sqs\Exception\SqsException::class);
        $client = new SqsClient([
            'region'  => 'us-west-2',
            'version' => 'latest'
        ]);

$mock = new Result(['Messages' => [['MD5OfBody' => 'foo', 'Body' => 'Bar']]]);
        $this->addMockResults($client, [$mock]);
        $client->receiveMessage(['QueueUrl' => 'http://foo.com']);
    }

public function testValidatesMd5OfMessageAttributes()
    {
        $this->expectExceptionMessage("Attribute MD5 mismatch. Expected foo, found ee5a4b60facbcc4723c1b5b8baca2593");
        $this->expectException(\Aws\Sqs\Exception\SqsException::class);
        $client = new SqsClient([
            'region'  => 'us-west-2',
            'version' => 'latest'
        ]);

$mock = new Result(['Messages' => [[
            'Body' => 'Test',
            'MD5OfMessageAttributes' => 'foo',
            'MessageAttributes' => [
                'Reference' => [
                    'BinaryValue' => 'ID-1234',
                    'DataType' => 'Binary'
                ],
                'Name' => [
                    'StringValue' => 'Bob',
                    'DataType' => 'String'
                ],
                'LastName' => [
                    'StringValue' => 'Smith',
                    'DataType' => 'String.LN'
                ],
                'Id' => [
                    'StringValue' => '3.14',
                    'DataType' => 'Number'
                ],
                'Test' => [
                    'StringValue' => '短发',
                    'DataType' => 'String'
                ],
                'Test2' => [
                    'StringValue' => 'true',
                    'DataType' => 'String.短发'
                ],
                'name' => [
                    'StringValue' => 'bob',
                    'DataType' => 'String'
                ],
                'Named' => [
                    'StringValue' => 'true',
                    'DataType' => 'String'
                ],
            ]
        ]]]);
        $this->addMockResults($client, [$mock]);
        $client->receiveMessage([
            'QueueUrl' => 'http://foo.com',
            'MessageAttributeNames' => [
                'All'
            ],
        ]);
    }

public function testValidatesMd5OfMessageAttributesExists()
    {
        $this->expectExceptionMessage("No Attribute MD5 found. Expected 0408bb33aa149494a6a4683d58a7133f");
        $this->expectException(\Aws\Sqs\Exception\SqsException::class);
        $client = new SqsClient([
            'region'  => 'us-west-2',
            'version' => 'latest'
        ]);

$mock = new Result(['Messages' => [[
            'Body' => 'Test',
            'MessageAttributes' => [
                'Name' => [
                    'StringValue' => 'Bob',
                    'DataType' => 'String'
                ]
            ]
        ]]]);
        $this->addMockResults($client, [$mock]);
        $client->receiveMessage([
            'QueueUrl' => 'http://foo.com',
            'MessageAttributeNames' => [
                'All'
            ],
        ]);
    }

public function testValidatesMessageAttributesExistWithMd5()
    {
        $this->expectExceptionMessage("Attribute MD5 mismatch. Expected foo, found No Attributes");
        $this->expectException(\Aws\Sqs\Exception\SqsException::class);
        $client = new SqsClient([
            'region'  => 'us-west-2',
            'version' => 'latest'
        ]);

$mock = new Result(['Messages' => [[
            'Body' => 'Test',
            'MD5OfMessageAttributes' => 'foo',
        ]]]);
        $this->addMockResults($client, [$mock]);
        $client->receiveMessage([
            'QueueUrl' => 'http://foo.com',
            'MessageAttributeNames' => [
                'All'
            ],
        ]);
    }

/** @doesNotPerformAssertions */
    public function testSkipsCommandsThatAreNotReceiveMessage()
    {
        $client = new SqsClient([
            'region'  => 'us-west-2',
            'version' => 'latest'
        ]);
        $this->addMockResults($client, [new Result()]);
        $client->listQueues();
    }

public function testDoesNotValidateEmptyMessageAttributes()
    {
        $client = new SqsClient([
            'region'  => 'us-west-2',
            'version' => 'latest'
        ]);

$mock = new Result([
            'Messages' => [
                [
                    'Body' => 'Test',
                    'MessageAttributes' => []
                ]
            ]
        ]);

$this->addMockResults($client, [$mock]);
        $response = $client->receiveMessage([
            'QueueUrl' => 'http://foo.com',
            'MessageAttributeNames' => [
                'All'
            ],
        ]);

$this->assertEmpty($response->get('MessageAttributes'));
    }
}