add package

[bytom/vapor.git] / vendor / github.com / multiformats / go-multihash / io_test.go

diff --git a/vendor/github.com/multiformats/go-multihash/io_test.go b/vendor/github.com/multiformats/go-multihash/io_test.go
new file mode 100644 (file)
index 0000000..d9ac4e5
--- /dev/null
+++ b/vendor/github.com/multiformats/go-multihash/io_test.go
@@ -0,0 +1,102 @@
+package multihash
+
+import (
+       "bytes"
+       "io"
+       "testing"
+)
+
+type evilReader struct {
+       buffer []byte
+}
+
+func (er *evilReader) Read(buf []byte) (int, error) {
+       n := copy(buf, er.buffer)
+       er.buffer = er.buffer[n:]
+       var err error
+       if len(er.buffer) == 0 {
+               err = io.EOF
+       }
+       return n, err
+}
+
+func TestEvilReader(t *testing.T) {
+       emptyHash, err := Sum(nil, ID, 0)
+       if err != nil {
+               t.Fatal(err)
+       }
+       r := NewReader(&evilReader{emptyHash})
+       h, err := r.ReadMultihash()
+       if err != nil {
+               t.Fatal(err)
+       }
+       if !bytes.Equal(h, []byte(emptyHash)) {
+               t.Fatal(err)
+       }
+       h, err = r.ReadMultihash()
+       if len([]byte(h)) > 0 || err != io.EOF {
+               t.Fatal("expected end of file")
+       }
+}
+
+func TestReader(t *testing.T) {
+
+       var buf bytes.Buffer
+
+       for _, tc := range testCases {
+               m, err := tc.Multihash()
+               if err != nil {
+                       t.Fatal(err)
+               }
+
+               buf.Write([]byte(m))
+       }
+
+       r := NewReader(&buf)
+
+       for _, tc := range testCases {
+               h, err := tc.Multihash()
+               if err != nil {
+                       t.Fatal(err)
+               }
+
+               h2, err := r.ReadMultihash()
+               if err != nil {
+                       t.Error(err)
+                       continue
+               }
+
+               if !bytes.Equal(h, h2) {
+                       t.Error("h and h2 should be equal")
+               }
+       }
+}
+
+func TestWriter(t *testing.T) {
+
+       var buf bytes.Buffer
+       w := NewWriter(&buf)
+
+       for _, tc := range testCases {
+               m, err := tc.Multihash()
+               if err != nil {
+                       t.Error(err)
+                       continue
+               }
+
+               if err := w.WriteMultihash(m); err != nil {
+                       t.Error(err)
+                       continue
+               }
+
+               buf2 := make([]byte, len(m))
+               if _, err := io.ReadFull(&buf, buf2); err != nil {
+                       t.Error(err)
+                       continue
+               }
+
+               if !bytes.Equal(m, buf2) {
+                       t.Error("m and buf2 should be equal")
+               }
+       }
+}