Merge pull request #75 from ahmetalpbalkan/large-stream
Modify TestWriteReadLargeStreams to write in one shot
This commit is contained in:
commit
a1547ab751
1 changed files with 29 additions and 9 deletions
|
@ -259,16 +259,13 @@ func (suite *DriverSuite) TestWriteReadLargeStreams(c *check.C) {
|
|||
defer suite.StorageDriver.Delete(firstPart(filename))
|
||||
|
||||
checksum := sha1.New()
|
||||
var offset int64
|
||||
var chunkSize int64 = 1024 * 1024
|
||||
var fileSize int64 = 5 * 1024 * 1024 * 1024
|
||||
|
||||
contents := newRandReader(fileSize)
|
||||
written, err := suite.StorageDriver.WriteStream(filename, 0, io.TeeReader(contents, checksum))
|
||||
c.Assert(err, check.IsNil)
|
||||
c.Assert(written, check.Equals, fileSize)
|
||||
|
||||
for i := 0; i < 5*1024; i++ {
|
||||
contents := randomContents(chunkSize)
|
||||
written, err := suite.StorageDriver.WriteStream(filename, offset, io.TeeReader(bytes.NewReader(contents), checksum))
|
||||
c.Assert(err, check.IsNil)
|
||||
c.Assert(written, check.Equals, chunkSize)
|
||||
offset += chunkSize
|
||||
}
|
||||
reader, err := suite.StorageDriver.ReadStream(filename, 0)
|
||||
c.Assert(err, check.IsNil)
|
||||
|
||||
|
@ -1088,6 +1085,29 @@ func randomContents(length int64) []byte {
|
|||
return b
|
||||
}
|
||||
|
||||
type randReader struct {
|
||||
r int64
|
||||
m sync.Mutex
|
||||
}
|
||||
|
||||
func (rr *randReader) Read(p []byte) (n int, err error) {
|
||||
rr.m.Lock()
|
||||
defer rr.m.Unlock()
|
||||
for i := 0; i < len(p) && rr.r > 0; i++ {
|
||||
p[i] = byte(rand.Intn(255))
|
||||
n++
|
||||
rr.r--
|
||||
}
|
||||
if rr.r == 0 {
|
||||
err = io.EOF
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func newRandReader(n int64) *randReader {
|
||||
return &randReader{r: n}
|
||||
}
|
||||
|
||||
func firstPart(filePath string) string {
|
||||
if filePath == "" {
|
||||
return "/"
|
||||
|
|
Loading…
Reference in a new issue