Merge pull request #142 from nan0tube/bytestreamproducer-readcloser

Close ByteStreamProducer data implementing ReadCloser

Author: casualjim

bytestream.go

@@ -109,6 +109,10 @@ func ByteStreamProducer(opts ...byteStreamOpt) Producer {
 		}
 		defer close()
 
+		if rc, ok := data.(io.ReadCloser); ok {
+			defer rc.Close()
+		}
+
 		if rdr, ok := data.(io.Reader); ok {
 			_, err := io.Copy(writer, rdr)
 			return err

bytestream_test.go

@@ -217,4 +217,14 @@ func TestBytestreamProducer_Close(t *testing.T) {
 		assert.Equal(t, expected, r.String())
 		assert.EqualValues(t, 0, r.calledClose)
 	}
+
+	cons = ByteStreamProducer()
+	r = &closingWriter{}
+	data := &closingReader{b: bytes.NewBufferString(expected)}
+	// can produce using a readcloser
+	if assert.NoError(t, cons.Produce(r, data)) {
+		assert.Equal(t, expected, r.String())
+		assert.EqualValues(t, 0, r.calledClose)
+		assert.EqualValues(t, 1, data.calledClose)
+	}
 }