shared-sqs: Этап 1 — клон GoAWS, удаление SNS, go build OK

2026-04-09 12:25:02 +03:00
parent d1d1bffd7c
commit f4352a17b1
58 changed files with 9688 additions and 0 deletions
@@ -0,0 +1,169 @@
+package gosqs
+
+import (
+	"fmt"
+	"net/http"
+	"testing"
+
+	"shared-sqs/app/test"
+
+	"shared-sqs/app/conf"
+	"shared-sqs/app/fixtures"
+	"shared-sqs/app/interfaces"
+	"shared-sqs/app/models"
+	"shared-sqs/app/utils"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSetQueueAttributesV1_success_multiple_attributes(t *testing.T) {
+	conf.LoadYamlConfig("../conf/mock-data/mock-config.yaml", "BaseUnitTests")
+	defer func() {
+		models.ResetApp()
+		utils.REQUEST_TRANSFORMER = utils.TransformRequest
+	}()
+
+	utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request, emptyRequestValid bool) (success bool) {
+		v := resultingStruct.(*models.SetQueueAttributesRequest)
+		*v = fixtures.SetQueueAttributesRequest
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, response := SetQueueAttributesV1(r)
+
+	expectedResponse := models.SetQueueAttributesResponse{
+		Xmlns:    models.BaseXmlns,
+		Metadata: models.BaseResponseMetadata,
+	}
+	assert.Equal(t, http.StatusOK, code)
+	assert.Equal(t, expectedResponse, response)
+
+	actualQueue := models.SyncQueues.Queues["unit-queue1"]
+	assert.Equal(t, 5, actualQueue.VisibilityTimeout)
+	assert.Equal(t, 4, actualQueue.ReceiveMessageWaitTimeSeconds)
+	assert.Equal(t, 1, actualQueue.DelaySeconds)
+	assert.Equal(t, 2, actualQueue.MaximumMessageSize)
+	assert.Equal(t, 3, actualQueue.MessageRetentionPeriod)
+}
+
+func TestSetQueueAttributesV1_success_single_attribute(t *testing.T) {
+	conf.LoadYamlConfig("../conf/mock-data/mock-config.yaml", "BaseUnitTests")
+	defer func() {
+		models.ResetApp()
+		utils.REQUEST_TRANSFORMER = utils.TransformRequest
+	}()
+
+	utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request, emptyRequestValid bool) (success bool) {
+		v := resultingStruct.(*models.SetQueueAttributesRequest)
+		*v = models.SetQueueAttributesRequest{
+			QueueUrl: fmt.Sprintf("%s/%s", fixtures.BASE_URL, "unit-queue1"),
+			Attributes: models.QueueAttributes{
+				VisibilityTimeout: 5,
+			},
+		}
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, response := SetQueueAttributesV1(r)
+
+	expectedResponse := models.SetQueueAttributesResponse{
+		Xmlns:    models.BaseXmlns,
+		Metadata: models.BaseResponseMetadata,
+	}
+	assert.Equal(t, http.StatusOK, code)
+	assert.Equal(t, expectedResponse, response)
+
+	actualQueue := models.SyncQueues.Queues["unit-queue1"]
+	assert.Equal(t, 5, actualQueue.VisibilityTimeout)
+	assert.Equal(t, 0, actualQueue.ReceiveMessageWaitTimeSeconds)
+	assert.Equal(t, 0, actualQueue.DelaySeconds)
+	assert.Equal(t, 0, actualQueue.MaximumMessageSize)
+	assert.Equal(t, 345600, actualQueue.MessageRetentionPeriod)
+}
+
+func TestSetQueueAttributesV1_invalid_request_body(t *testing.T) {
+	conf.LoadYamlConfig("../conf/mock-data/mock-config.yaml", "BaseUnitTests")
+	defer func() {
+		models.ResetApp()
+		utils.REQUEST_TRANSFORMER = utils.TransformRequest
+	}()
+
+	utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request, emptyRequestValid bool) (success bool) {
+		return false
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, _ := SetQueueAttributesV1(r)
+
+	assert.Equal(t, http.StatusBadRequest, code)
+}
+
+func TestSetQueueAttributesV1_missing_queue_url(t *testing.T) {
+	conf.LoadYamlConfig("../conf/mock-data/mock-config.yaml", "BaseUnitTests")
+	defer func() {
+		models.ResetApp()
+		utils.REQUEST_TRANSFORMER = utils.TransformRequest
+	}()
+
+	utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request, emptyRequestValid bool) (success bool) {
+		v := resultingStruct.(*models.SetQueueAttributesRequest)
+		*v = models.SetQueueAttributesRequest{
+			Attributes: models.QueueAttributes{},
+		}
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, _ := SetQueueAttributesV1(r)
+
+	assert.Equal(t, http.StatusBadRequest, code)
+}
+
+func TestSetQueueAttributesV1_missing_expected_queue(t *testing.T) {
+	conf.LoadYamlConfig("../conf/mock-data/mock-config.yaml", "BaseUnitTests")
+	defer func() {
+		models.ResetApp()
+		utils.REQUEST_TRANSFORMER = utils.TransformRequest
+	}()
+
+	utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request, emptyRequestValid bool) (success bool) {
+		v := resultingStruct.(*models.SetQueueAttributesRequest)
+		*v = models.SetQueueAttributesRequest{
+			QueueUrl: "garbage",
+		}
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, _ := SetQueueAttributesV1(r)
+
+	assert.Equal(t, http.StatusBadRequest, code)
+}
+
+func TestSetQueueAttributesV1_invalid_redrive_queue(t *testing.T) {
+	conf.LoadYamlConfig("../conf/mock-data/mock-config.yaml", "BaseUnitTests")
+	defer func() {
+		models.ResetApp()
+		utils.REQUEST_TRANSFORMER = utils.TransformRequest
+	}()
+
+	utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request, emptyRequestValid bool) (success bool) {
+		v := resultingStruct.(*models.SetQueueAttributesRequest)
+		*v = models.SetQueueAttributesRequest{
+			QueueUrl: fmt.Sprintf("%s/%s", fixtures.BASE_URL, "unit-queue1"),
+			Attributes: models.QueueAttributes{
+				RedrivePolicy: models.RedrivePolicy{
+					MaxReceiveCount:     100,
+					DeadLetterTargetArn: fmt.Sprintf("arn:aws:sqs:us-east-1:100010001000:%s", "garbage"),
+				},
+			},
+		}
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, _ := SetQueueAttributesV1(r)
+
+	assert.Equal(t, http.StatusBadRequest, code)
+}