package middleware import ( "net/http" "net/http/httptest" "testing" "github.com/stretchr/testify/require" ) func TestReqTypeDetermination(t *testing.T) { bkt, obj, domain := "test-bucket", "test-object", "domain" for _, tc := range []struct { name string target string host string domains []string expectedType ReqType expectedBktName string expectedObjName string }{ { name: "bucket request, path-style", target: "/" + bkt, expectedType: bucketType, expectedBktName: bkt, }, { name: "bucket request with slash, path-style", target: "/" + bkt + "/", expectedType: bucketType, expectedBktName: bkt, }, { name: "object request, path-style", target: "/" + bkt + "/" + obj, expectedType: objectType, expectedBktName: bkt, expectedObjName: obj, }, { name: "object request with slash, path-style", target: "/" + bkt + "/" + obj + "/", expectedType: objectType, expectedBktName: bkt, expectedObjName: obj + "/", }, { name: "none type request", target: "/", expectedType: noneType, }, { name: "bucket request, virtual-hosted style", target: "/", host: bkt + "." + domain, domains: []string{"some-domain", domain}, expectedType: bucketType, expectedBktName: bkt, }, { name: "object request, virtual-hosted style", target: "/" + obj, host: bkt + "." + domain, domains: []string{"some-domain", domain}, expectedType: objectType, expectedBktName: bkt, expectedObjName: obj, }, } { t.Run(tc.name, func(t *testing.T) { r := httptest.NewRequest(http.MethodPut, tc.target, nil) r.Host = tc.host reqType, bktName, objName := getBucketObject(r, tc.domains) require.Equal(t, tc.expectedType, reqType) require.Equal(t, tc.expectedBktName, bktName) require.Equal(t, tc.expectedObjName, objName) }) } } func TestDetermineBucketOperation(t *testing.T) { const defaultValue = "value" for _, tc := range []struct { name string method string queryParam map[string]string expected string }{ { name: "OptionsOperation", method: http.MethodOptions, expected: OptionsOperation, }, { name: "HeadBucketOperation", method: http.MethodHead, expected: HeadBucketOperation, }, { name: "ListMultipartUploadsOperation", method: http.MethodGet, queryParam: map[string]string{UploadsQuery: defaultValue}, expected: ListMultipartUploadsOperation, }, { name: "GetBucketLocationOperation", method: http.MethodGet, queryParam: map[string]string{LocationQuery: defaultValue}, expected: GetBucketLocationOperation, }, { name: "GetBucketPolicyOperation", method: http.MethodGet, queryParam: map[string]string{PolicyQuery: defaultValue}, expected: GetBucketPolicyOperation, }, { name: "GetBucketLifecycleOperation", method: http.MethodGet, queryParam: map[string]string{LifecycleQuery: defaultValue}, expected: GetBucketLifecycleOperation, }, { name: "GetBucketEncryptionOperation", method: http.MethodGet, queryParam: map[string]string{EncryptionQuery: defaultValue}, expected: GetBucketEncryptionOperation, }, { name: "GetBucketCorsOperation", method: http.MethodGet, queryParam: map[string]string{CorsQuery: defaultValue}, expected: GetBucketCorsOperation, }, { name: "GetBucketACLOperation", method: http.MethodGet, queryParam: map[string]string{ACLQuery: defaultValue}, expected: GetBucketACLOperation, }, { name: "GetBucketWebsiteOperation", method: http.MethodGet, queryParam: map[string]string{WebsiteQuery: defaultValue}, expected: GetBucketWebsiteOperation, }, { name: "GetBucketAccelerateOperation", method: http.MethodGet, queryParam: map[string]string{AccelerateQuery: defaultValue}, expected: GetBucketAccelerateOperation, }, { name: "GetBucketRequestPaymentOperation", method: http.MethodGet, queryParam: map[string]string{RequestPaymentQuery: defaultValue}, expected: GetBucketRequestPaymentOperation, }, { name: "GetBucketLoggingOperation", method: http.MethodGet, queryParam: map[string]string{LoggingQuery: defaultValue}, expected: GetBucketLoggingOperation, }, { name: "GetBucketReplicationOperation", method: http.MethodGet, queryParam: map[string]string{ReplicationQuery: defaultValue}, expected: GetBucketReplicationOperation, }, { name: "GetBucketTaggingOperation", method: http.MethodGet, queryParam: map[string]string{TaggingQuery: defaultValue}, expected: GetBucketTaggingOperation, }, { name: "GetBucketObjectLockConfigOperation", method: http.MethodGet, queryParam: map[string]string{ObjectLockQuery: defaultValue}, expected: GetBucketObjectLockConfigOperation, }, { name: "GetBucketVersioningOperation", method: http.MethodGet, queryParam: map[string]string{VersioningQuery: defaultValue}, expected: GetBucketVersioningOperation, }, { name: "GetBucketNotificationOperation", method: http.MethodGet, queryParam: map[string]string{NotificationQuery: defaultValue}, expected: GetBucketNotificationOperation, }, { name: "ListenBucketNotificationOperation", method: http.MethodGet, queryParam: map[string]string{EventsQuery: defaultValue}, expected: ListenBucketNotificationOperation, }, { name: "ListBucketObjectVersionsOperation", method: http.MethodGet, queryParam: map[string]string{VersionsQuery: defaultValue}, expected: ListBucketObjectVersionsOperation, }, { name: "ListObjectsV2MOperation", method: http.MethodGet, queryParam: map[string]string{ListTypeQuery: "2", MetadataQuery: "true"}, expected: ListObjectsV2MOperation, }, { name: "ListObjectsV2Operation", method: http.MethodGet, queryParam: map[string]string{ListTypeQuery: "2"}, expected: ListObjectsV2Operation, }, { name: "ListObjectsV1Operation", method: http.MethodGet, expected: ListObjectsV1Operation, }, { name: "PutBucketCorsOperation", method: http.MethodPut, queryParam: map[string]string{CorsQuery: defaultValue}, expected: PutBucketCorsOperation, }, { name: "PutBucketACLOperation", method: http.MethodPut, queryParam: map[string]string{ACLQuery: defaultValue}, expected: PutBucketACLOperation, }, { name: "PutBucketLifecycleOperation", method: http.MethodPut, queryParam: map[string]string{LifecycleQuery: defaultValue}, expected: PutBucketLifecycleOperation, }, { name: "PutBucketEncryptionOperation", method: http.MethodPut, queryParam: map[string]string{EncryptionQuery: defaultValue}, expected: PutBucketEncryptionOperation, }, { name: "PutBucketPolicyOperation", method: http.MethodPut, queryParam: map[string]string{PolicyQuery: defaultValue}, expected: PutBucketPolicyOperation, }, { name: "PutBucketObjectLockConfigOperation", method: http.MethodPut, queryParam: map[string]string{ObjectLockQuery: defaultValue}, expected: PutBucketObjectLockConfigOperation, }, { name: "PutBucketTaggingOperation", method: http.MethodPut, queryParam: map[string]string{TaggingQuery: defaultValue}, expected: PutBucketTaggingOperation, }, { name: "PutBucketVersioningOperation", method: http.MethodPut, queryParam: map[string]string{VersioningQuery: defaultValue}, expected: PutBucketVersioningOperation, }, { name: "PutBucketNotificationOperation", method: http.MethodPut, queryParam: map[string]string{NotificationQuery: defaultValue}, expected: PutBucketNotificationOperation, }, { name: "CreateBucketOperation", method: http.MethodPut, expected: CreateBucketOperation, }, { name: "DeleteMultipleObjectsOperation", method: http.MethodPost, queryParam: map[string]string{DeleteQuery: defaultValue}, expected: DeleteMultipleObjectsOperation, }, { name: "PostObjectOperation", method: http.MethodPost, expected: PostObjectOperation, }, { name: "DeleteBucketCorsOperation", method: http.MethodDelete, queryParam: map[string]string{CorsQuery: defaultValue}, expected: DeleteBucketCorsOperation, }, { name: "DeleteBucketWebsiteOperation", method: http.MethodDelete, queryParam: map[string]string{WebsiteQuery: defaultValue}, expected: DeleteBucketWebsiteOperation, }, { name: "DeleteBucketTaggingOperation", method: http.MethodDelete, queryParam: map[string]string{TaggingQuery: defaultValue}, expected: DeleteBucketTaggingOperation, }, { name: "DeleteBucketPolicyOperation", method: http.MethodDelete, queryParam: map[string]string{PolicyQuery: defaultValue}, expected: DeleteBucketPolicyOperation, }, { name: "DeleteBucketLifecycleOperation", method: http.MethodDelete, queryParam: map[string]string{LifecycleQuery: defaultValue}, expected: DeleteBucketLifecycleOperation, }, { name: "DeleteBucketEncryptionOperation", method: http.MethodDelete, queryParam: map[string]string{EncryptionQuery: defaultValue}, expected: DeleteBucketEncryptionOperation, }, { name: "DeleteBucketOperation", method: http.MethodDelete, expected: DeleteBucketOperation, }, { name: "UnmatchedBucketOperation", method: "invalid-method", expected: "UnmatchedBucketOperation", }, } { t.Run(tc.name, func(t *testing.T) { req := httptest.NewRequest(tc.method, "/test", nil) if tc.queryParam != nil { addQueryParams(req, tc.queryParam) } actual := determineBucketOperation(req) require.Equal(t, tc.expected, actual) }) } } func TestDetermineObjectOperation(t *testing.T) { const ( amzCopySource = "X-Amz-Copy-Source" defaultValue = "value" ) for _, tc := range []struct { name string method string queryParam map[string]string headerKeys []string expected string }{ { name: "HeadObjectOperation", method: http.MethodHead, expected: HeadObjectOperation, }, { name: "ListPartsOperation", method: http.MethodGet, queryParam: map[string]string{UploadIDQuery: defaultValue}, expected: ListPartsOperation, }, { name: "GetObjectACLOperation", method: http.MethodGet, queryParam: map[string]string{ACLQuery: defaultValue}, expected: GetObjectACLOperation, }, { name: "GetObjectTaggingOperation", method: http.MethodGet, queryParam: map[string]string{TaggingQuery: defaultValue}, expected: GetObjectTaggingOperation, }, { name: "GetObjectRetentionOperation", method: http.MethodGet, queryParam: map[string]string{RetentionQuery: defaultValue}, expected: GetObjectRetentionOperation, }, { name: "GetObjectLegalHoldOperation", method: http.MethodGet, queryParam: map[string]string{LegalQuery: defaultValue}, expected: GetObjectLegalHoldOperation, }, { name: "GetObjectAttributesOperation", method: http.MethodGet, queryParam: map[string]string{AttributesQuery: defaultValue}, expected: GetObjectAttributesOperation, }, { name: "GetObjectOperation", method: http.MethodGet, expected: GetObjectOperation, }, { name: "UploadPartCopyOperation", method: http.MethodPut, queryParam: map[string]string{PartNumberQuery: defaultValue, UploadIDQuery: defaultValue}, headerKeys: []string{amzCopySource}, expected: UploadPartCopyOperation, }, { name: "UploadPartOperation", method: http.MethodPut, queryParam: map[string]string{PartNumberQuery: defaultValue, UploadIDQuery: defaultValue}, expected: UploadPartOperation, }, { name: "PutObjectACLOperation", method: http.MethodPut, queryParam: map[string]string{ACLQuery: defaultValue}, expected: PutObjectACLOperation, }, { name: "PutObjectTaggingOperation", method: http.MethodPut, queryParam: map[string]string{TaggingQuery: defaultValue}, expected: PutObjectTaggingOperation, }, { name: "CopyObjectOperation", method: http.MethodPut, headerKeys: []string{amzCopySource}, expected: CopyObjectOperation, }, { name: "PutObjectRetentionOperation", method: http.MethodPut, queryParam: map[string]string{RetentionQuery: defaultValue}, expected: PutObjectRetentionOperation, }, { name: "PutObjectLegalHoldOperation", method: http.MethodPut, queryParam: map[string]string{LegalHoldQuery: defaultValue}, expected: PutObjectLegalHoldOperation, }, { name: "PutObjectOperation", method: http.MethodPut, expected: PutObjectOperation, }, { name: "CompleteMultipartUploadOperation", method: http.MethodPost, queryParam: map[string]string{UploadIDQuery: defaultValue}, expected: CompleteMultipartUploadOperation, }, { name: "CreateMultipartUploadOperation", method: http.MethodPost, queryParam: map[string]string{UploadsQuery: defaultValue}, expected: CreateMultipartUploadOperation, }, { name: "SelectObjectContentOperation", method: http.MethodPost, expected: SelectObjectContentOperation, }, { name: "AbortMultipartUploadOperation", method: http.MethodDelete, queryParam: map[string]string{UploadIDQuery: defaultValue}, expected: AbortMultipartUploadOperation, }, { name: "DeleteObjectTaggingOperation", method: http.MethodDelete, queryParam: map[string]string{TaggingQuery: defaultValue}, expected: DeleteObjectTaggingOperation, }, { name: "DeleteObjectOperation", method: http.MethodDelete, expected: DeleteObjectOperation, }, { name: "UnmatchedObjectOperation", method: "invalid-method", expected: "UnmatchedObjectOperation", }, } { t.Run(tc.name, func(t *testing.T) { req := httptest.NewRequest(tc.method, "/test", nil) if tc.queryParam != nil { addQueryParams(req, tc.queryParam) } if tc.headerKeys != nil { addHeaderParams(req, tc.headerKeys) } actual := determineObjectOperation(req) require.Equal(t, tc.expected, actual) }) } } func addQueryParams(req *http.Request, pairs map[string]string) { values := req.URL.Query() for key, val := range pairs { values.Add(key, val) } req.URL.RawQuery = values.Encode() } func addHeaderParams(req *http.Request, keys []string) { for _, key := range keys { req.Header.Set(key, "val") } } func TestDetermineGeneralOperation(t *testing.T) { req := httptest.NewRequest(http.MethodGet, "/test", nil) actual := determineGeneralOperation(req) require.Equal(t, ListBucketsOperation, actual) req = httptest.NewRequest(http.MethodPost, "/test", nil) actual = determineGeneralOperation(req) require.Equal(t, "UnmatchedOperation", actual) }