Add a test about message size exceeding

goddenrich · May 8, 2024 · 29dc396 · 29dc396
1 parent 9083ad0
commit 29dc396
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 2 deletions.
diff --git a/app/gosqs/gosqs.go b/app/gosqs/gosqs.go
@@ -42,7 +42,7 @@ func init() {
 	app.SqsErrors["InvalidVisibilityTimeout"] = err8
 	err9 := app.SqsErrorType{HttpError: http.StatusBadRequest, Type: "MessageNotInFlight", Code: "AWS.SimpleQueueService.MessageNotInFlight", Message: "The message referred to isn't in flight."}
 	app.SqsErrors["MessageNotInFlight"] = err9
-	err10 := app.SqsErrorType{HttpError: http.StatusBadRequest, Type: "MessageTooBig", Code: "InvalidMessageContents", Message: "The message size exceeds the limit."}
+	err10 := app.SqsErrorType{HttpError: http.StatusBadRequest, Type: "MessageTooBig", Code: "InvalidParameterValue", Message: "The message size exceeds the limit."}
 	app.SqsErrors["MessageTooBig"] = err10
 	app.SqsErrors[ErrInvalidParameterValue.Type] = *ErrInvalidParameterValue
 	app.SqsErrors[ErrInvalidAttributeValue.Type] = *ErrInvalidAttributeValue

diff --git a/smoke_tests/sqs_send_message_test.go b/smoke_tests/sqs_send_message_test.go
@@ -168,7 +168,32 @@ func Test_SendMessageV1_json_with_attributes(t *testing.T) {
 }
 
 func Test_SendMessageV1_json_MaximumMessageSize_TooBig(t *testing.T) {
-	// TODO
+	server := generateServer()
+	defer func() {
+		server.Close()
+		utils.ResetResources()
+	}()
+
+	sdkConfig, _ := config.LoadDefaultConfig(context.TODO())
+	sdkConfig.BaseEndpoint = aws.String(server.URL)
+	sqsClient := sqs.NewFromConfig(sdkConfig)
+	sdkResponse, _ := sqsClient.CreateQueue(context.TODO(), &sqs.CreateQueueInput{
+		QueueName: &af.QueueName,
+		Attributes: map[string]string{
+			"MaximumMessageSize": "1",
+		},
+	})
+	targetQueueUrl := sdkResponse.QueueUrl
+
+	// Target test: send a message that is bigger than MaximumMessageSize
+	targetMessageBody := "Test_SendMessageV1_json_no_attributes"
+	sendMessageOutput, err := sqsClient.SendMessage(context.TODO(), &sqs.SendMessageInput{
+		QueueUrl:    targetQueueUrl,
+		MessageBody: &targetMessageBody,
+	})
+	assert.Contains(t, err.Error(), "400")
+	assert.Contains(t, err.Error(), "InvalidParameterValue")
+	assert.Nil(t, sendMessageOutput)
 }
 
 func Test_SendMessageV1_json_QueueNotExistant(t *testing.T) {
@@ -190,5 +215,6 @@ func Test_SendMessageV1_json_QueueNotExistant(t *testing.T) {
 		MessageBody: &targetMessageBody,
 	})
 	assert.Contains(t, err.Error(), "400")
+	assert.Contains(t, err.Error(), "AWS.SimpleQueueService.NonExistentQueue")
 	assert.Nil(t, sendMessageOutput)
 }