From a03956b35e856089ae2782940b0500a70bdc2162 Mon Sep 17 00:00:00 2001
From: Vincent Batts <vbatts@hashbangbash.com>
Date: Sun, 16 Aug 2015 19:02:13 -0400
Subject: [PATCH] stream_test: remove buffer use

---
 stream_test.go | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/stream_test.go b/stream_test.go
index 6afffa6..6c55c98 100644
--- a/stream_test.go
+++ b/stream_test.go
@@ -1,7 +1,6 @@
 package merkle
 
 import (
-	"bytes"
 	"fmt"
 	"io"
 	"io/ioutil"
@@ -59,15 +58,15 @@ func TestMerkleHashWriterLargeChunk(t *testing.T) {
 }
 
 func TestMerkleHashWriter(t *testing.T) {
-	msg := "the quick brown fox jumps over the lazy dog"
+	msg := []byte("the quick brown fox jumps over the lazy dog")
 	expectedSum := "48940c1c72636648ad40aa59c162f2208e835b38"
 
 	h := NewHash(DefaultHashMaker, 10)
-	i, err := io.Copy(h, bytes.NewBufferString(msg))
+	i, err := h.Write(msg)
 	if err != nil {
 		t.Fatal(err)
 	}
-	if i != int64(len(msg)) {
+	if i != len(msg) {
 		t.Fatalf("expected to write %d, only wrote %d", len(msg), i)
 	}
 
@@ -103,11 +102,11 @@ func TestMerkleHashWriter(t *testing.T) {
 	}
 
 	// write our msg again and get the same sum
-	i, err = io.Copy(h, bytes.NewBufferString(msg))
+	i, err = h.Write(msg)
 	if err != nil {
 		t.Fatal(err)
 	}
-	if i != int64(len(msg)) {
+	if i != len(msg) {
 		t.Fatalf("expected to write %d, only wrote %d", len(msg), i)
 	}
 	// Test Sum(), ensure same sum
@@ -117,11 +116,11 @@ func TestMerkleHashWriter(t *testing.T) {
 	}
 
 	// Write more. This should pop the last node, and use the lastBlock.
-	i, err = io.Copy(h, bytes.NewBufferString(msg))
+	i, err = h.Write(msg)
 	if err != nil {
 		t.Fatal(err)
 	}
-	if i != int64(len(msg)) {
+	if i != len(msg) {
 		t.Fatalf("expected to write %d, only wrote %d", len(msg), i)
 	}
 	expectedNum = 9