maxmustermann/minio
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Donut now tests amount written should match expected amount to write

Browse source
This commit is contained in:
Frederick F. Kautz IV 2015-02-28 16:09:52 -08:00
parent d3d1205eb1
commit ddc7cf835e
2 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pkg/storage/donut/v1/donut.go
View file

@ -19,6 +19,7 @@ package v1
import (
"bytes"
"encoding/binary"
"errors"
"io"
"github.com/minio-io/minio/pkg/utils/checksum/crc32c"
@ -101,10 +102,13 @@ func Write(target io.Writer, reader io.Reader, length uint64) error {
checksumChannel := make(chan checksumValue)
go generateChecksum(sumReader, checksumChannel)
teeReader := io.TeeReader(reader, sumWriter)
_, err = io.Copy(target, teeReader)
dataLength, err := io.Copy(target, teeReader)
if err != nil {
return err
}
if uint64(dataLength) != length {
return errors.New("Specified data length and amount written mismatched")
}
sumWriter.Close()
dataChecksum := <-checksumChannel
if dataChecksum.err != nil {

6
pkg/storage/donut/v1/donut_test.go
View file

@ -124,3 +124,9 @@ func (s *MySuite) TestSingleWrite(c *C) {
// ensure no extra data is in the file
c.Assert(testBuffer.Len(), Equals, 0)
}
func (s *MySuite) TestLengthMismatchInWrite(c *C) {
var testData bytes.Buffer
err := Write(&testData, bytes.NewBufferString("hello, world"), 5)
c.Assert(err, Not(IsNil))
}