105 lines
3.5 KiB
Go
105 lines
3.5 KiB
Go
// +-------------------------------------------------------------------------
|
|
// | Copyright (C) 2016 Yunify, Inc.
|
|
// +-------------------------------------------------------------------------
|
|
// | Licensed under the Apache License, Version 2.0 (the "License");
|
|
// | you may not use this work except in compliance with the License.
|
|
// | You may obtain a copy of the License in the LICENSE file, or at:
|
|
// |
|
|
// | http://www.apache.org/licenses/LICENSE-2.0
|
|
// |
|
|
// | Unless required by applicable law or agreed to in writing, software
|
|
// | distributed under the License is distributed on an "AS IS" BASIS,
|
|
// | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// | See the License for the specific language governing permissions and
|
|
// | limitations under the License.
|
|
// +-------------------------------------------------------------------------
|
|
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/DATA-DOG/godog"
|
|
"github.com/DATA-DOG/godog/gherkin"
|
|
|
|
qs "github.com/yunify/qingstor-sdk-go/service"
|
|
)
|
|
|
|
// BucketPolicyFeatureContext provides feature context for bucket policy.
|
|
func BucketPolicyFeatureContext(s *godog.Suite) {
|
|
s.Step(`^put bucket policy:$`, putBucketPolicy)
|
|
s.Step(`^put bucket policy status code is (\d+)$`, putBucketPolicyStatusCodeIs)
|
|
|
|
s.Step(`^get bucket policy$`, getBucketPolicy)
|
|
s.Step(`^get bucket policy status code is (\d+)$`, getBucketPolicyStatusCodeIs)
|
|
s.Step(`^get bucket policy should have Referer "([^"]*)"$`, getBucketPolicyShouldHaveReferer)
|
|
|
|
s.Step(`^delete bucket policy$`, deleteBucketPolicy)
|
|
s.Step(`^delete bucket policy status code is (\d+)$`, deleteBucketPolicyStatusCodeIs)
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
var putBucketPolicyOutput *qs.PutBucketPolicyOutput
|
|
|
|
func putBucketPolicy(PolicyJSONText *gherkin.DocString) error {
|
|
putBucketPolicyInput := &qs.PutBucketPolicyInput{}
|
|
err = json.Unmarshal([]byte(PolicyJSONText.Content), putBucketPolicyInput)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(putBucketPolicyInput.Statement) == 1 {
|
|
putBucketPolicyInput.Statement[0].Resource = qs.StringSlice([]string{tc.BucketName + "/*"})
|
|
}
|
|
|
|
putBucketPolicyOutput, err = bucket.PutPolicy(putBucketPolicyInput)
|
|
return err
|
|
}
|
|
|
|
func putBucketPolicyStatusCodeIs(statusCode int) error {
|
|
return checkEqual(qs.IntValue(putBucketPolicyOutput.StatusCode), statusCode)
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
var getBucketPolicyOutput *qs.GetBucketPolicyOutput
|
|
|
|
func getBucketPolicy() error {
|
|
getBucketPolicyOutput, err = bucket.GetPolicy()
|
|
return err
|
|
}
|
|
|
|
func getBucketPolicyStatusCodeIs(statusCode int) error {
|
|
return checkEqual(qs.IntValue(getBucketPolicyOutput.StatusCode), statusCode)
|
|
}
|
|
|
|
func getBucketPolicyShouldHaveReferer(compare string) error {
|
|
for _, statement := range getBucketPolicyOutput.Statement {
|
|
if statement.Condition != nil &&
|
|
statement.Condition.StringLike != nil {
|
|
|
|
for _, referer := range statement.Condition.StringLike.Referer {
|
|
if qs.StringValue(referer) == compare {
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return fmt.Errorf("Referer \"%s\" not found in bucket policy statement", compare)
|
|
}
|
|
|
|
// --------------------------------------------------------------------------
|
|
|
|
var deleteBucketPolicyOutput *qs.DeleteBucketPolicyOutput
|
|
|
|
func deleteBucketPolicy() error {
|
|
deleteBucketPolicyOutput, err = bucket.DeletePolicy()
|
|
return err
|
|
}
|
|
|
|
func deleteBucketPolicyStatusCodeIs(statusCode int) error {
|
|
return checkEqual(qs.IntValue(deleteBucketPolicyOutput.StatusCode), statusCode)
|
|
}
|