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
140 lines
3.4 KiB
Go
140 lines
3.4 KiB
Go
package utils
|
|
|
|
import (
|
|
"net/url"
|
|
"testing"
|
|
|
|
"shared-sqs/app/models"
|
|
|
|
"shared-sqs/app/test"
|
|
|
|
"shared-sqs/app/fixtures"
|
|
"shared-sqs/app/mocks"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestTransformRequest_success_json(t *testing.T) {
|
|
_, r := test.GenerateRequestInfo("POST", "url", fixtures.JSONRequestBody, true)
|
|
|
|
mock := &mocks.MockRequestBody{}
|
|
|
|
ok := TransformRequest(mock, r, false)
|
|
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "mock-value", mock.RequestFieldStr)
|
|
assert.False(t, mock.SetAttributesFromFormCalled)
|
|
}
|
|
|
|
func TestTransformRequest_success_json_empty_request_accepted(t *testing.T) {
|
|
_, r := test.GenerateRequestInfo("POST", "url", nil, true)
|
|
|
|
mock := &mocks.MockRequestBody{}
|
|
|
|
ok := TransformRequest(mock, r, true)
|
|
|
|
assert.True(t, ok)
|
|
//assert.Equal(t, "mock-value", mock.RequestFieldStr)
|
|
assert.False(t, mock.SetAttributesFromFormCalled)
|
|
}
|
|
|
|
func TestTransformRequest_success_xml(t *testing.T) {
|
|
_, r := test.GenerateRequestInfo("POST", "url", nil, false)
|
|
form := url.Values{}
|
|
form.Add("Action", "CreateQueue")
|
|
form.Add("QueueName", "UnitTestQueue1")
|
|
form.Add("Attribute.1.Name", "VisibilityTimeout")
|
|
form.Add("Attribute.1.Value", "60")
|
|
form.Add("Attribute.2.Name", "MaximumMessageSize")
|
|
form.Add("Attribute.2.Value", "2048")
|
|
r.PostForm = form
|
|
|
|
mock := &mocks.MockRequestBody{}
|
|
|
|
ok := TransformRequest(mock, r, false)
|
|
|
|
assert.True(t, ok)
|
|
assert.True(t, mock.SetAttributesFromFormCalled)
|
|
assert.Equal(t, []interface{}{form}, mock.SetAttributesFromFormCalledWith)
|
|
}
|
|
|
|
func TestTransformRequest_error_invalid_request_body_json(t *testing.T) {
|
|
_, r := test.GenerateRequestInfo("POST", "url", "\"I-am-garbage", true)
|
|
|
|
mock := &mocks.MockRequestBody{}
|
|
|
|
ok := TransformRequest(mock, r, false)
|
|
|
|
assert.False(t, ok)
|
|
assert.Equal(t, "", mock.RequestFieldStr)
|
|
assert.False(t, mock.SetAttributesFromFormCalled)
|
|
}
|
|
|
|
func TestTransformRequest_error_failure_to_parse_form_xml(t *testing.T) {
|
|
_, r := test.GenerateRequestInfo("POST", "url", nil, false)
|
|
|
|
mock := &mocks.MockRequestBody{}
|
|
|
|
ok := TransformRequest(mock, r, false)
|
|
|
|
assert.False(t, ok)
|
|
assert.False(t, mock.SetAttributesFromFormCalled)
|
|
}
|
|
|
|
func TestTransformRequest_error_invalid_request_body_xml(t *testing.T) {
|
|
_, r := test.GenerateRequestInfo("POST", "url", nil, false)
|
|
|
|
form := url.Values{}
|
|
form.Add("intField", "\"I-am-garbage")
|
|
r.PostForm = form
|
|
|
|
mock := &mocks.MockRequestBody{}
|
|
|
|
ok := TransformRequest(mock, r, false)
|
|
|
|
assert.False(t, ok)
|
|
assert.False(t, mock.SetAttributesFromFormCalled)
|
|
}
|
|
|
|
func TestExtractQueueAttributes_success(t *testing.T) {
|
|
u := url.Values{}
|
|
u.Add("Attribute.1.Name", "DelaySeconds")
|
|
u.Add("Attribute.1.Value", "20")
|
|
u.Add("Attribute.2.Name", "VisibilityTimeout")
|
|
u.Add("Attribute.2.Value", "30")
|
|
u.Add("Attribute.3.Name", "Policy")
|
|
|
|
attr := ExtractQueueAttributes(u)
|
|
expected := map[string]string{
|
|
"DelaySeconds": "20",
|
|
"VisibilityTimeout": "30",
|
|
}
|
|
|
|
assert.Equal(t, expected, attr)
|
|
}
|
|
|
|
func TestGetMD5Hash(t *testing.T) {
|
|
hash1 := GetMD5Hash("This is a test")
|
|
hash2 := GetMD5Hash("This is a test")
|
|
if hash1 != hash2 {
|
|
t.Errorf("hashs and hash2 should be the same, but were not")
|
|
}
|
|
|
|
hash1 = GetMD5Hash("This is a test")
|
|
hash2 = GetMD5Hash("This is a tfst")
|
|
if hash1 == hash2 {
|
|
t.Errorf("hashs and hash2 are the same, but should not be")
|
|
}
|
|
}
|
|
|
|
func TestSortedKeys(t *testing.T) {
|
|
attributes := map[string]models.MessageAttribute{
|
|
"b": {},
|
|
"a": {},
|
|
}
|
|
|
|
keys := sortedKeys(attributes)
|
|
assert.Equal(t, "a", keys[0])
|
|
assert.Equal(t, "b", keys[1])
|
|
}
|