func init() {
Before("", func() {
conn = s3.NewMockS3()
out = bytes.Buffer{}
tempDir, _ = ioutil.TempDir("", "")
os.Chdir(tempDir)
})
After("", func() {
// Integration tests are mostly run against mock S3, but can be run
// against a real account, so we should cleanup.
for _, bucket := range testBuckets {
cleanupBucket(bucket)
}
// Cleanup temp dir
if tempDir != "" {
os.RemoveAll(tempDir)
}
})
Given(`^I have bucket "(.+?)"$`, func(bucket string) {
input := awss3.CreateBucketInput{
Bucket: aws.String(bucket),
}
conn.CreateBucket(&input)
})
Given(`^bucket "(.+?)" key "(.+?)" contains "(.+?)"$`, func(bucket string, key string, content string) {
body := bytes.NewReader([]byte(content))
input := awss3.PutObjectInput{
Bucket: aws.String(bucket),
Key: aws.String(key),
Body: body,
}
conn.PutObject(&input)
})
Given(`^local file "(.+?)" contains "(.+?)"$`, func(filename string, content string) {
// create containing directory if necessary
dirname := path.Dir(filename)
if dirname != "" {
if _, err := os.Stat(dirname); os.IsNotExist(err) {
err := os.MkdirAll(dirname, 0755)
if err != nil {
T.Errorf("Couldn't create directory: %s\n%s", dirname, err)
return
}
}
}
file, err := os.Create(filename)
if err != nil {
T.Errorf("Couldn't create file: %s\n%s", filename, err)
return
}
defer file.Close()
file.WriteString(content)
})
When(`^I run "(.+?)"$`, func(s1 string) {
args := strings.Split(s1, " ")
o := threadSafeWriter{&out, sync.Mutex{}}
lastExitCode = s3.Main(conn, args, &o)
})
Then(`^local file "(.+?)" has contents "(.+?)"$`, func(filename string, exp string) {
file, err := os.Open(filename)
if err != nil {
T.Errorf("Local file error:\n%s", err)
return
}
defer file.Close()
content, err := ioutil.ReadAll(file)
if err != nil {
T.Errorf("Local file error:\n%s", err)
return
}
act := string(content)
if act != exp {
T.Errorf("%s contents expected:\n%s\ngot:\n%s", filename, exp, act)
}
})
Then(`^the output is "(.*?)"$`, func(exp string) {
// replace newlines
exp = replacer.Replace(exp)
act := string(out.Bytes())
if act != exp {
T.Errorf("Output expected:\n%s\ngot:\n%s", exp, act)
}
})
Then(`^the output contains "(.*?)"$`, func(exp string) {
// replace newlines
exp = replacer.Replace(exp)
act := string(out.Bytes())
if !strings.Contains(act, exp) {
T.Errorf("Output does not contain:\n%s\ngot:\n%s", exp, act)
}
})
Then(`^the exit code is (\d+?)$`, func(code int) {
if code != lastExitCode {
T.Errorf("Exit code expected:\n%d\ngot:\n%d", code, lastExitCode)
}
})
Then(`^the bucket "(.+?)" exists$`, func(bucket string) {
if !bucketExists(bucket) {
T.Errorf("Bucket %s does not exist", bucket)
}
})
Then(`^the bucket "(.+?)" does not exist$`, func(bucket string) {
if bucketExists(bucket) {
T.Errorf("Bucket %s exists", bucket)
}
})
Then(`^bucket "(.+?)" has key "(.+?)" with contents "(.+?)"$`, func(bucket string, key string, exp string) {
input := awss3.GetObjectInput{
Bucket: aws.String(bucket),
Key: aws.String(key),
}
output, err := conn.GetObject(&input)
if err != nil {
T.Errorf("Bucket %s Key %s error:\n%s", bucket, key, err)
return
}
content, err := ioutil.ReadAll(output.Body)
if err != nil {
T.Errorf("Bucket %s Key %s error:\n%s", bucket, key, err)
return
}
act := string(content)
if act != exp {
T.Errorf("%s Key %s contents expected:\n%s\ngot:\n%s", bucket, key, exp, act)
return
}
})
Then(`^bucket "(.+?)" key "(.+?)" exists$`, func(bucket string, key string) {
input := awss3.GetObjectInput{
Bucket: aws.String(bucket),
Key: aws.String(key),
}
_, err := conn.GetObject(&input)
if err != nil {
T.Errorf("Bucket %s Key %s does not exist", bucket, key)
}
})
Then(`^bucket "(.+?)" key "(.+?)" does not exist$`, func(bucket string, key string) {
input := awss3.GetObjectInput{
Bucket: aws.String(bucket),
Key: aws.String(key),
}
_, err := conn.GetObject(&input)
if err == nil {
T.Errorf("Bucket %s Key %s exists", bucket, key)
}
})
}
func main() {
exitCode := s3.Main(nil, os.Args, os.Stdout)
os.Exit(exitCode)
}