TEST_F(QueueOperationTest, TestQueueAttributes) { CreateQueueRequest createQueueRequest; createQueueRequest.SetQueueName(BuildResourceName(BASE_ATTRIBUTES_QUEUE_NAME)); createQueueRequest.AddAttributes(QueueAttributeName::DelaySeconds, "45"); CreateQueueOutcome createQueueOutcome = sqsClient->CreateQueue(createQueueRequest); ASSERT_TRUE(createQueueOutcome.IsSuccess()); Aws::String queueUrl = createQueueOutcome.GetResult().GetQueueUrl(); ASSERT_TRUE(queueUrl.find(createQueueRequest.GetQueueName()) != Aws::String::npos); GetQueueAttributesRequest queueAttributesRequest; queueAttributesRequest.AddAttributeNames(QueueAttributeName::DelaySeconds).WithQueueUrl(queueUrl); GetQueueAttributesOutcome queueAttributesOutcome = sqsClient->GetQueueAttributes(queueAttributesRequest); ASSERT_TRUE(queueAttributesOutcome.IsSuccess()); EXPECT_EQ("45", queueAttributesOutcome.GetResult().GetAttributes().find(QueueAttributeName::DelaySeconds)->second); SetQueueAttributesRequest setQueueAttributesRequest; setQueueAttributesRequest.AddAttributes(QueueAttributeName::VisibilityTimeout, "42").WithQueueUrl(queueUrl); SetQueueAttributesOutcome setQueueAttributesOutcome = sqsClient->SetQueueAttributes(setQueueAttributesRequest); ASSERT_TRUE(setQueueAttributesOutcome.IsSuccess()); queueAttributesRequest.AddAttributeNames(QueueAttributeName::VisibilityTimeout).WithQueueUrl(queueUrl); queueAttributesOutcome = sqsClient->GetQueueAttributes(queueAttributesRequest); ASSERT_TRUE(queueAttributesOutcome.IsSuccess()); EXPECT_EQ("45", queueAttributesOutcome.GetResult().GetAttributes().find(QueueAttributeName::DelaySeconds)->second); EXPECT_EQ("42", queueAttributesOutcome.GetResult().GetAttributes().find(QueueAttributeName::VisibilityTimeout)->second); DeleteQueueRequest deleteQueueRequest; deleteQueueRequest.WithQueueUrl(queueUrl); DeleteQueueOutcome deleteQueueOutcome = sqsClient->DeleteQueue(deleteQueueRequest); ASSERT_TRUE(deleteQueueOutcome.IsSuccess()); }
TEST_F(QueueOperationTest, TestCreateAndDeleteQueue) { CreateQueueRequest createQueueRequest; createQueueRequest.SetQueueName(BuildResourceName(BASE_SIMPLE_QUEUE_NAME)); CreateQueueOutcome createQueueOutcome; bool shouldContinue = true; while (shouldContinue) { createQueueOutcome = sqsClient->CreateQueue(createQueueRequest); if (createQueueOutcome.IsSuccess()) break; if (createQueueOutcome.GetError().GetErrorType() == SQSErrors::QUEUE_DELETED_RECENTLY) { std::this_thread::sleep_for(std::chrono::seconds(10)); } else { FAIL() << "Unexpected error response: " << createQueueOutcome.GetError().GetMessage(); } } Aws::String queueUrl = createQueueOutcome.GetResult().GetQueueUrl(); ASSERT_TRUE(queueUrl.find(createQueueRequest.GetQueueName()) != Aws::String::npos); createQueueRequest.AddAttributes(QueueAttributeName::VisibilityTimeout, "50"); createQueueOutcome = sqsClient->CreateQueue(createQueueRequest); ASSERT_FALSE(createQueueOutcome.IsSuccess()); SQSErrors error = createQueueOutcome.GetError().GetErrorType(); EXPECT_TRUE(SQSErrors::QUEUE_NAME_EXISTS == error || SQSErrors::QUEUE_DELETED_RECENTLY == error); // This call in eventually consistent (sometimes over 1 min), so try it a few times for (int attempt = 0; ; attempt++) { ListQueuesRequest listQueueRequest; listQueueRequest.WithQueueNamePrefix(BuildResourcePrefix()); ListQueuesOutcome listQueuesOutcome = sqsClient->ListQueues(listQueueRequest); if (listQueuesOutcome.IsSuccess()) { ListQueuesResult listQueuesResult = listQueuesOutcome.GetResult(); if (listQueuesResult.GetQueueUrls().size() == 1) { EXPECT_EQ(queueUrl, listQueuesResult.GetQueueUrls()[0]); EXPECT_TRUE(listQueuesResult.GetResponseMetadata().GetRequestId().length() > 0); break; // success! } } if (attempt >= 10) FAIL(); std::this_thread::sleep_for(std::chrono::seconds(10)); } DeleteQueueRequest deleteQueueRequest; deleteQueueRequest.WithQueueUrl(queueUrl); DeleteQueueOutcome deleteQueueOutcome = sqsClient->DeleteQueue(deleteQueueRequest); ASSERT_TRUE(deleteQueueOutcome.IsSuccess()); }
TEST_F(QueueOperationTest, TestListDeadLetterSourceQueues) { CreateQueueRequest createQueueRequest; createQueueRequest.SetQueueName(DEAD_LETTER_SOURCE_QUEUE_NAME); CreateQueueOutcome createQueueOutcome = sqsClient->CreateQueue(createQueueRequest); ASSERT_TRUE(createQueueOutcome.IsSuccess()); Aws::String queueUrl = createQueueOutcome.GetResult().GetQueueUrl(); createQueueRequest.SetQueueName(DEAD_LETTER_QUEUE_NAME); createQueueOutcome = sqsClient->CreateQueue(createQueueRequest); ASSERT_TRUE(createQueueOutcome.IsSuccess()); Aws::String deadLetterQueueUrl = createQueueOutcome.GetResult().GetQueueUrl(); GetQueueAttributesRequest queueAttributesRequest; queueAttributesRequest.AddAttributeNames(QueueAttributeName::QueueArn).WithQueueUrl(deadLetterQueueUrl); GetQueueAttributesOutcome queueAttributesOutcome = sqsClient->GetQueueAttributes(queueAttributesRequest); ASSERT_TRUE(queueAttributesOutcome.IsSuccess()); Aws::String redrivePolicy = "{\"maxReceiveCount\":\"5\", \"deadLetterTargetArn\":\"" + queueAttributesOutcome.GetResult().GetAttributes().find(QueueAttributeName::QueueArn)->second + "\"}"; SetQueueAttributesRequest setQueueAttributesRequest; setQueueAttributesRequest.AddAttributes(QueueAttributeName::RedrivePolicy, redrivePolicy).WithQueueUrl(queueUrl); SetQueueAttributesOutcome setQueueAttributesOutcome = sqsClient->SetQueueAttributes(setQueueAttributesRequest); ASSERT_TRUE(setQueueAttributesOutcome.IsSuccess()); ListDeadLetterSourceQueuesRequest listDeadLetterQueuesRequest; listDeadLetterQueuesRequest.WithQueueUrl(deadLetterQueueUrl); ListDeadLetterSourceQueuesOutcome listDeadLetterQueuesOutcome = sqsClient->ListDeadLetterSourceQueues(listDeadLetterQueuesRequest); ASSERT_TRUE(listDeadLetterQueuesOutcome.IsSuccess()); //deadletter queue stuff is eventually consistent, let's try for 100 seconds or so. unsigned count = 0; bool found = listDeadLetterQueuesOutcome.GetResult().GetQueueUrls().size() == 1uL; while (listDeadLetterQueuesOutcome.IsSuccess() && !found && count++ < 100) { if (listDeadLetterQueuesOutcome.GetResult().GetQueueUrls().size() == 1uL) { found = true; } std::this_thread::sleep_for(std::chrono::seconds(1)); listDeadLetterQueuesOutcome = sqsClient->ListDeadLetterSourceQueues(listDeadLetterQueuesRequest); } ASSERT_TRUE(found); EXPECT_EQ(queueUrl, listDeadLetterQueuesOutcome.GetResult().GetQueueUrls()[0]); DeleteQueueRequest deleteQueueRequest; deleteQueueRequest.WithQueueUrl(queueUrl); DeleteQueueOutcome deleteQueueOutcome = sqsClient->DeleteQueue(deleteQueueRequest); ASSERT_TRUE(deleteQueueOutcome.IsSuccess()); deleteQueueRequest.WithQueueUrl(deadLetterQueueUrl); deleteQueueOutcome = sqsClient->DeleteQueue(deleteQueueRequest); ASSERT_TRUE(deleteQueueOutcome.IsSuccess()); }
TEST_F(QueueOperationTest, TestPermissions) { Aws::String queueName = BuildResourceName(BASE_PERMISSIONS_QUEUE_NAME); Aws::String queueUrl = CreateDefaultQueue(queueName); ASSERT_TRUE(queueUrl.find(queueName) != Aws::String::npos); AddPermissionRequest addPermissionRequest; addPermissionRequest.AddAWSAccountIds(m_accountId).AddActions("ReceiveMessage").WithLabel("Test").WithQueueUrl( queueUrl); AddPermissionOutcome permissionOutcome = sqsClient->AddPermission(addPermissionRequest); ASSERT_TRUE(permissionOutcome.IsSuccess()); GetQueueAttributesRequest queueAttributesRequest; queueAttributesRequest.AddAttributeNames(QueueAttributeName::Policy).WithQueueUrl(queueUrl); GetQueueAttributesOutcome queueAttributesOutcome = sqsClient->GetQueueAttributes(queueAttributesRequest); ASSERT_TRUE(queueAttributesOutcome.IsSuccess()); Aws::String policyString = queueAttributesOutcome.GetResult().GetAttributes().find(QueueAttributeName::Policy)->second; EXPECT_TRUE(policyString.length() > 0); JsonValue policy(policyString); EXPECT_EQ(addPermissionRequest.GetLabel(), policy.GetArray("Statement")[0].GetString("Sid")); Aws::StringStream expectedValue; expectedValue << "arn:aws:iam::" << m_accountId << ":root"; EXPECT_EQ(expectedValue.str(), policy.GetArray("Statement")[0].GetObject("Principal").GetString("AWS")); EXPECT_EQ("SQS:ReceiveMessage", policy.GetArray("Statement")[0].GetString("Action")); RemovePermissionRequest removePermissionRequest; removePermissionRequest.WithLabel("Test").WithQueueUrl(queueUrl); RemovePermissionOutcome removePermissionOutcome = sqsClient->RemovePermission(removePermissionRequest); ASSERT_TRUE(removePermissionOutcome.IsSuccess()); queueAttributesOutcome = sqsClient->GetQueueAttributes(queueAttributesRequest); ASSERT_TRUE(queueAttributesOutcome.IsSuccess()); EXPECT_TRUE(queueAttributesOutcome.GetResult().GetAttributes().find(QueueAttributeName::Policy) == queueAttributesOutcome.GetResult().GetAttributes().end()); DeleteQueueRequest deleteQueueRequest; deleteQueueRequest.WithQueueUrl(queueUrl); DeleteQueueOutcome deleteQueueOutcome = sqsClient->DeleteQueue(deleteQueueRequest); ASSERT_TRUE(deleteQueueOutcome.IsSuccess()); }
TEST_F(QueueOperationTest, TestSendReceiveDelete) { Aws::String queueName = BuildResourceName(BASE_SEND_RECEIVE_QUEUE_NAME); Aws::String queueUrl = CreateDefaultQueue(queueName); ASSERT_TRUE(queueUrl.find(queueName) != Aws::String::npos); SendMessageRequest sendMessageRequest; sendMessageRequest.SetMessageBody("TestMessageBody"); MessageAttributeValue stringAttributeValue; stringAttributeValue.SetStringValue("TestString"); stringAttributeValue.SetDataType("String"); sendMessageRequest.AddMessageAttributes("TestStringAttribute", stringAttributeValue); MessageAttributeValue binaryAttributeValue; Aws::Utils::ByteBuffer byteBuffer(10); for(unsigned i = 0; i < 10; ++i) { byteBuffer[i] = (unsigned char)i; } binaryAttributeValue.SetBinaryValue(byteBuffer); binaryAttributeValue.SetDataType("Binary"); sendMessageRequest.AddMessageAttributes("TestBinaryAttribute", binaryAttributeValue); sendMessageRequest.SetQueueUrl(queueUrl); SendMessageOutcome sendMessageOutcome = sqsClient->SendMessage(sendMessageRequest); ASSERT_TRUE(sendMessageOutcome.IsSuccess()); EXPECT_TRUE(sendMessageOutcome.GetResult().GetMessageId().length() > 0); ReceiveMessageRequest receiveMessageRequest; receiveMessageRequest.SetMaxNumberOfMessages(1); receiveMessageRequest.SetQueueUrl(queueUrl); receiveMessageRequest.AddMessageAttributeNames("All"); ReceiveMessageOutcome receiveMessageOutcome = sqsClient->ReceiveMessage(receiveMessageRequest); ASSERT_TRUE(receiveMessageOutcome.IsSuccess()); ReceiveMessageResult receiveMessageResult = receiveMessageOutcome.GetResult(); ASSERT_EQ(1uL, receiveMessageResult.GetMessages().size()); EXPECT_EQ("TestMessageBody", receiveMessageResult.GetMessages()[0].GetBody()); Aws::Map<Aws::String, MessageAttributeValue> messageAttributes = receiveMessageResult.GetMessages()[0].GetMessageAttributes(); ASSERT_TRUE(messageAttributes.find("TestStringAttribute") != messageAttributes.end()); EXPECT_EQ(stringAttributeValue.GetStringValue(), messageAttributes["TestStringAttribute"].GetStringValue()); ASSERT_TRUE(messageAttributes.find("TestBinaryAttribute") != messageAttributes.end()); EXPECT_EQ(byteBuffer, messageAttributes["TestBinaryAttribute"].GetBinaryValue()); DeleteMessageRequest deleteMessageRequest; deleteMessageRequest.SetQueueUrl(queueUrl); deleteMessageRequest.SetReceiptHandle(receiveMessageResult.GetMessages()[0].GetReceiptHandle()); DeleteMessageOutcome deleteMessageOutcome = sqsClient->DeleteMessage(deleteMessageRequest); ASSERT_TRUE(deleteMessageOutcome.IsSuccess()); receiveMessageOutcome = sqsClient->ReceiveMessage(receiveMessageRequest); EXPECT_EQ(0uL, receiveMessageOutcome.GetResult().GetMessages().size()); DeleteQueueRequest deleteQueueRequest; deleteQueueRequest.WithQueueUrl(queueUrl); DeleteQueueOutcome deleteQueueOutcome = sqsClient->DeleteQueue(deleteQueueRequest); ASSERT_TRUE(deleteQueueOutcome.IsSuccess()); }