diff options
Diffstat (limited to 'storage/memory/object_test.go')
-rw-r--r-- | storage/memory/object_test.go | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/storage/memory/object_test.go b/storage/memory/object_test.go index f9dd25d..f2873fa 100644 --- a/storage/memory/object_test.go +++ b/storage/memory/object_test.go @@ -51,7 +51,11 @@ func (s *ObjectSuite) TestSize(c *C) { func (s *ObjectSuite) TestReader(c *C) { o := &Object{content: []byte("foo")} - b, err := ioutil.ReadAll(o.Reader()) + reader, err := o.Reader() + c.Assert(err, IsNil) + defer func() { c.Assert(reader.Close(), IsNil) }() + + b, err := ioutil.ReadAll(reader) c.Assert(err, IsNil) c.Assert(b, DeepEquals, []byte("foo")) } @@ -59,7 +63,11 @@ func (s *ObjectSuite) TestReader(c *C) { func (s *ObjectSuite) TestWriter(c *C) { o := &Object{} - n, err := o.Writer().Write([]byte("foo")) + writer, err := o.Writer() + c.Assert(err, IsNil) + defer func() { c.Assert(writer.Close(), IsNil) }() + + n, err := writer.Write([]byte("foo")) c.Assert(err, IsNil) c.Assert(n, Equals, 3) |