|
|
|
@ -39,48 +39,50 @@ func TestIsValidLocationContraint(t *testing.T) { |
|
|
|
|
t.Fatal(err) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// Test with corrupted XML
|
|
|
|
|
// Corrupted XML
|
|
|
|
|
malformedReq := &http.Request{ |
|
|
|
|
Body: ioutil.NopCloser(bytes.NewBuffer([]byte("<>"))), |
|
|
|
|
ContentLength: int64(len("<>")), |
|
|
|
|
} |
|
|
|
|
if _, err := parseLocationConstraint(malformedReq); err != ErrMalformedXML { |
|
|
|
|
t.Fatal("Unexpected error: ", err) |
|
|
|
|
|
|
|
|
|
// Not an XML
|
|
|
|
|
badRequest := &http.Request{ |
|
|
|
|
Body: ioutil.NopCloser(bytes.NewReader([]byte("garbage"))), |
|
|
|
|
ContentLength: int64(len("garbage")), |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// generates the input request with XML bucket configuration set to the request body.
|
|
|
|
|
createExpectedRequest := func(req *http.Request, location string) (*http.Request, error) { |
|
|
|
|
createExpectedRequest := func(req *http.Request, location string) *http.Request { |
|
|
|
|
createBucketConfig := createBucketLocationConfiguration{} |
|
|
|
|
createBucketConfig.Location = location |
|
|
|
|
var createBucketConfigBytes []byte |
|
|
|
|
createBucketConfigBytes, e := xml.Marshal(createBucketConfig) |
|
|
|
|
if e != nil { |
|
|
|
|
return nil, e |
|
|
|
|
} |
|
|
|
|
createBucketConfigBytes, _ := xml.Marshal(createBucketConfig) |
|
|
|
|
createBucketConfigBuffer := bytes.NewBuffer(createBucketConfigBytes) |
|
|
|
|
req.Body = ioutil.NopCloser(createBucketConfigBuffer) |
|
|
|
|
req.ContentLength = int64(createBucketConfigBuffer.Len()) |
|
|
|
|
return req, nil |
|
|
|
|
return req |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
testCases := []struct { |
|
|
|
|
locationForInputRequest string |
|
|
|
|
request *http.Request |
|
|
|
|
serverConfigRegion string |
|
|
|
|
expectedCode APIErrorCode |
|
|
|
|
}{ |
|
|
|
|
// Test case - 1.
|
|
|
|
|
{globalMinioDefaultRegion, globalMinioDefaultRegion, ErrNone}, |
|
|
|
|
{createExpectedRequest(&http.Request{}, "eu-central-1"), globalMinioDefaultRegion, ErrNone}, |
|
|
|
|
// Test case - 2.
|
|
|
|
|
// In case of empty request body ErrNone is returned.
|
|
|
|
|
{"", globalMinioDefaultRegion, ErrNone}, |
|
|
|
|
{createExpectedRequest(&http.Request{}, ""), globalMinioDefaultRegion, ErrNone}, |
|
|
|
|
// Test case - 3
|
|
|
|
|
// In case of garbage request body ErrMalformedXML is returned.
|
|
|
|
|
{badRequest, globalMinioDefaultRegion, ErrMalformedXML}, |
|
|
|
|
// Test case - 4
|
|
|
|
|
// In case of invalid XML request body ErrMalformedXML is returned.
|
|
|
|
|
{malformedReq, globalMinioDefaultRegion, ErrMalformedXML}, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
for i, testCase := range testCases { |
|
|
|
|
inputRequest, e := createExpectedRequest(&http.Request{}, testCase.locationForInputRequest) |
|
|
|
|
if e != nil { |
|
|
|
|
t.Fatalf("Test %d: Failed to Marshal bucket configuration", i+1) |
|
|
|
|
} |
|
|
|
|
globalServerConfig.SetRegion(testCase.serverConfigRegion) |
|
|
|
|
_, actualCode := parseLocationConstraint(inputRequest) |
|
|
|
|
_, actualCode := parseLocationConstraint(testCase.request) |
|
|
|
|
if testCase.expectedCode != actualCode { |
|
|
|
|
t.Errorf("Test %d: Expected the APIErrCode to be %d, but instead found %d", i+1, testCase.expectedCode, actualCode) |
|
|
|
|
} |
|
|
|
|