Naeel
c3ba2dcae4
- Standalone SQS-service repository - Multi-tenant message queue service, AWS SQS compatible - Based on GoAws, with mutable tenants, auth, WebUI, Redis persistence - Ready for independent development and deployment - See doc/ and README.md for architecture and usage
117 lines
3.7 KiB
Go
117 lines
3.7 KiB
Go
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 TestListQueuesV1_success(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.ListQueueRequest)
|
|
*v = models.ListQueueRequest{}
|
|
return true
|
|
}
|
|
|
|
_, r := test.GenerateRequestInfo("POST", "/", nil, true)
|
|
code, response := ListQueuesV1(r)
|
|
r1 := response.(models.ListQueuesResponse)
|
|
|
|
assert.Equal(t, http.StatusOK, code)
|
|
assert.Contains(t, r1.Result.QueueUrls, fmt.Sprintf("%s/%s", fixtures.BASE_URL, "unit-queue1"))
|
|
assert.Contains(t, r1.Result.QueueUrls, fmt.Sprintf("%s/%s", fixtures.BASE_URL, "unit-queue2"))
|
|
assert.Contains(t, r1.Result.QueueUrls, fmt.Sprintf("%s/%s", fixtures.BASE_URL, "dead-letter-queue1"))
|
|
}
|
|
|
|
func TestListQueuesV1_success_no_queues(t *testing.T) {
|
|
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.ListQueueRequest)
|
|
*v = models.ListQueueRequest{}
|
|
return true
|
|
}
|
|
|
|
_, r := test.GenerateRequestInfo("POST", "/", nil, true)
|
|
code, response := ListQueuesV1(r)
|
|
r1 := response.(models.ListQueuesResponse)
|
|
|
|
assert.Equal(t, http.StatusOK, code)
|
|
assert.Equal(t, r1.Result.QueueUrls, []string{})
|
|
}
|
|
|
|
func TestListQueuesV1_success_with_queue_name_prefix(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.ListQueueRequest)
|
|
*v = models.ListQueueRequest{QueueNamePrefix: "dead-letter"}
|
|
return true
|
|
}
|
|
|
|
_, r := test.GenerateRequestInfo("POST", "/", nil, true)
|
|
code, response := ListQueuesV1(r)
|
|
r1 := response.(models.ListQueuesResponse)
|
|
|
|
assert.Equal(t, http.StatusOK, code)
|
|
assert.Equal(t, []string{fmt.Sprintf("%s/%s", fixtures.BASE_URL, "dead-letter-queue1")}, r1.Result.QueueUrls)
|
|
}
|
|
|
|
func TestListQueuesV1_success_with_queue_name_prefix_no_matching_queues(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.ListQueueRequest)
|
|
*v = models.ListQueueRequest{QueueNamePrefix: "garbage"}
|
|
return true
|
|
}
|
|
|
|
_, r := test.GenerateRequestInfo("POST", "/", nil, true)
|
|
code, response := ListQueuesV1(r)
|
|
r1 := response.(models.ListQueuesResponse)
|
|
|
|
assert.Equal(t, http.StatusOK, code)
|
|
assert.Equal(t, []string{}, r1.Result.QueueUrls)
|
|
}
|
|
|
|
func TestListQueuesV1_request_transformer_error(t *testing.T) {
|
|
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, _ := ListQueuesV1(r)
|
|
|
|
assert.Equal(t, http.StatusBadRequest, code)
|
|
}
|