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,180 @@
+package gosqs
+
+import (
+	"fmt"
+	"net/http"
+	"testing"
+
+	"shared-sqs/app/test"
+
+	"github.com/mitchellh/copystructure"
+
+	"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 TestGetQueueAttributesV1_success_all(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.GetQueueAttributesRequest)
+		*v = fixtures.GetQueueAttributesRequest
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, response := GetQueueAttributesV1(r)
+
+	assert.Equal(t, http.StatusOK, code)
+	assert.Equal(t, fixtures.GetQueueAttributesResponse, response)
+}
+
+func TestGetQueueAttributesV1_success_no_request_attrs_returns_all(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.GetQueueAttributesRequest)
+		*v = models.GetQueueAttributesRequest{
+			QueueUrl: "unit-queue1",
+		}
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, response := GetQueueAttributesV1(r)
+
+	assert.Equal(t, http.StatusOK, code)
+	assert.Equal(t, fixtures.GetQueueAttributesResponse, response)
+}
+
+func TestGetQueueAttributesV1_success_all_with_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.GetQueueAttributesRequest)
+		*v = models.GetQueueAttributesRequest{
+			QueueUrl:       "unit-queue2",
+			AttributeNames: []string{"All"},
+		}
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, response := GetQueueAttributesV1(r)
+
+	dupe, _ := copystructure.Copy(fixtures.GetQueueAttributesResponse)
+	expectedResponse, _ := dupe.(models.GetQueueAttributesResponse)
+	expectedResponse.Result.Attrs[9].Value = fmt.Sprintf("%s:%s", fixtures.BASE_SQS_ARN, "unit-queue2")
+	expectedResponse.Result.Attrs = append(expectedResponse.Result.Attrs,
+		models.Attribute{
+			Name:  "RedrivePolicy",
+			Value: fmt.Sprintf(`{"maxReceiveCount":"1", "deadLetterTargetArn":"%s:%s"}`, fixtures.BASE_SQS_ARN, "dead-letter-queue1"),
+		},
+	)
+
+	assert.Equal(t, http.StatusOK, code)
+	assert.Equal(t, expectedResponse, response)
+}
+
+func TestGetQueueAttributesV1_success_specific_fields(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.GetQueueAttributesRequest)
+		*v = models.GetQueueAttributesRequest{
+			QueueUrl:       fmt.Sprintf("%s/unit-queue1", fixtures.BASE_URL),
+			AttributeNames: []string{"DelaySeconds"},
+		}
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, response := GetQueueAttributesV1(r)
+
+	expectedResponse := models.GetQueueAttributesResponse{
+		Xmlns: models.BaseXmlns,
+		Result: models.GetQueueAttributesResult{Attrs: []models.Attribute{
+			models.Attribute{
+				Name:  "DelaySeconds",
+				Value: "0",
+			},
+		}},
+		Metadata: models.BaseResponseMetadata,
+	}
+
+	assert.Equal(t, http.StatusOK, code)
+	assert.Equal(t, expectedResponse, response)
+}
+
+func TestGetQueueAttributesV1_request_transformer_error(t *testing.T) {
+	defer func() {
+		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, _ := GetQueueAttributesV1(r)
+
+	assert.Equal(t, http.StatusBadRequest, code)
+}
+
+func TestGetQueueAttributesV1_missing_queue_url_in_request_returns_error(t *testing.T) {
+	defer func() {
+		utils.REQUEST_TRANSFORMER = utils.TransformRequest
+	}()
+
+	utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request, emptyRequestValid bool) (success bool) {
+		v := resultingStruct.(*models.GetQueueAttributesRequest)
+		*v = models.GetQueueAttributesRequest{
+			QueueUrl:       "",
+			AttributeNames: []string{},
+		}
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, _ := GetQueueAttributesV1(r)
+
+	assert.Equal(t, http.StatusBadRequest, code)
+}
+
+func TestGetQueueAttributesV1_missing_queue_returns_error(t *testing.T) {
+	defer func() {
+		utils.REQUEST_TRANSFORMER = utils.TransformRequest
+	}()
+
+	utils.REQUEST_TRANSFORMER = func(resultingStruct interfaces.AbstractRequestBody, req *http.Request, emptyRequestValid bool) (success bool) {
+		v := resultingStruct.(*models.GetQueueAttributesRequest)
+		*v = fixtures.GetQueueAttributesRequest
+		return true
+	}
+
+	_, r := test.GenerateRequestInfo("POST", "/", nil, true)
+	code, _ := GetQueueAttributesV1(r)
+
+	assert.Equal(t, http.StatusBadRequest, code)
+}