octoleo/restic
2
2
Fork 0
mirror of https://github.com/octoleo/restic.git synced 2024-11-10 15:21:03 +00:00
Code Issues Releases Wiki Activity

Modify test suite to use s3 test server.

Browse Source 
Fix s3 backend to return error on attempting to write to already
existing key.
This commit is contained in:
Chris Howey 2015-05-16 06:48:55 -05:00
parent 4aa75994b0
commit 2350419f59
2 changed files with 35 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/s3/s3.go
View File

@ -87,8 +87,15 @@ func (bb *s3Blob) Finalize(t backend.Type, name string) error {
bb.final = true bb.final = true
path := s3path(t, name) path := s3path(t, name)
// Check key does not already exist
key, err := bb.b.bucket.GetKey(path)
if err == nil && key.Key == path {
return errors.New("key already exists!")
}
bb.b.mput.Lock() bb.b.mput.Lock()
err := bb.b.bucket.Put(path, bb.buf.Bytes(), "binary/octet-stream", "private") err = bb.b.bucket.Put(path, bb.buf.Bytes(), "binary/octet-stream", "private")
bb.b.mput.Unlock() bb.b.mput.Unlock()
bb.buf.Reset() bb.buf.Reset()
return err return err

44
backend/s3_test.go
View File

@ -5,39 +5,49 @@ import (
"github.com/mitchellh/goamz/aws" "github.com/mitchellh/goamz/aws"
"github.com/mitchellh/goamz/s3" "github.com/mitchellh/goamz/s3"
"github.com/mitchellh/goamz/testutil" "github.com/mitchellh/goamz/s3/s3test"
bes3 "github.com/restic/restic/backend/s3" bes3 "github.com/restic/restic/backend/s3"
. "github.com/restic/restic/test" . "github.com/restic/restic/test"
) )
var testServer = testutil.NewHTTPServer() type LocalServer struct {
auth aws.Auth
region aws.Region
srv *s3test.Server
config *s3test.Config
}
var s LocalServer
func setupS3Backend(t *testing.T) *bes3.S3 { func setupS3Backend(t *testing.T) *bes3.S3 {
testServer.Start() s.config = &s3test.Config{
auth := aws.Auth{"abc", "123", ""} Send409Conflict: true,
service := s3.New(auth, aws.Region{Name: "faux-region-1", S3Endpoint: testServer.URL}) }
srv, err := s3test.NewServer(s.config)
OK(t, err)
s.srv = srv
s.region = aws.Region{
Name: "faux-region-1",
S3Endpoint: srv.URL(),
S3LocationConstraint: true, // s3test server requires a LocationConstraint
}
s.auth = aws.Auth{"abc", "123", ""}
service := s3.New(s.auth, s.region)
bucket := service.Bucket("testbucket") bucket := service.Bucket("testbucket")
err := bucket.PutBucket("private") err = bucket.PutBucket("private")
OK(t, err) OK(t, err)
t.Logf("created s3 backend locally at %s", testServer.URL) t.Logf("created s3 backend locally")
return bes3.OpenS3Bucket(bucket, "testbucket") return bes3.OpenS3Bucket(bucket, "testbucket")
} }
func teardownS3Backend(t *testing.T, b *bes3.S3) {
if !*TestCleanup {
t.Logf("leaving backend at %s\n", b.Location())
return
}
testServer.Flush()
}
func TestS3Backend(t *testing.T) { func TestS3Backend(t *testing.T) {
s := setupS3Backend(t) s := setupS3Backend(t)
defer teardownS3Backend(t, s)
testBackend(s, t) testBackend(s, t)
} }