octoleo/restic
2
2
Fork 0
mirror of https://github.com/octoleo/restic.git synced 2024-11-26 14:56:29 +00:00
Code Issues Releases Wiki Activity

backend tests load: Use reader with Size() method

Browse Source
This commit is contained in:
Alexander Neumann 2017-05-13 19:56:11 +02:00
parent c7209ef231
commit 4ac0d3ad40
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/restic/backend/test/tests.go
View File

@ -277,7 +277,8 @@ func BackendTestLoad(t testing.TB, s *Suite) {
type errorCloser struct { type errorCloser struct {
io.Reader io.Reader
t testing.TB size int64
t testing.TB
} }
func (ec errorCloser) Close() error { func (ec errorCloser) Close() error {
@ -285,6 +286,10 @@ func (ec errorCloser) Close() error {
return errors.New("forbidden method close was called") return errors.New("forbidden method close was called")
} }
func (ec errorCloser) Size() int64 {
return ec.size
}
// BackendTestSave tests saving data in the backend. // BackendTestSave tests saving data in the backend.
func BackendTestSave(t testing.TB, s *Suite) { func BackendTestSave(t testing.TB, s *Suite) {
b := s.open(t) b := s.open(t)
@ -354,7 +359,7 @@ func BackendTestSave(t testing.TB, s *Suite) {
// wrap the tempfile in an errorCloser, so we can detect if the backend // wrap the tempfile in an errorCloser, so we can detect if the backend
// closes the reader // closes the reader
err = b.Save(h, errorCloser{t: t, Reader: tmpfile}) err = b.Save(h, errorCloser{t: t, size: int64(length), Reader: tmpfile})
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }