test: add unixfs/reader tests for WriteTo and size
License: MIT Signed-off-by: Jakub Sztandera <kubuxu@protonmail.ch>
Jakub Sztandera committed
Sep 8, 2016 at 13:28 UTC
a547d88094c0878da384509f1b7421a65ec55923
1 file changed
+39
unixfs/io/dagreader_test.go
+39
@@ -1,6 +1,7 @@
1
package io
2
3
import (
4
+ "bytes"
5
"io/ioutil"
6
"os"
7
"strings"
@@ -197,6 +198,44 @@ func TestMetadataNode(t *testing.T) {
198
}
199
}
200
201
+func TestWriteTo(t *testing.T) {
202
+ dserv := testu.GetDAGServ()
203
+ inbuf, node := testu.GetRandomNode(t, dserv, 1024)
204
+ ctx, closer := context.WithCancel(context.Background())
205
+ defer closer()
206
+
207
+ reader, err := NewDagReader(ctx, node, dserv)
208
+ if err != nil {
209
+ t.Fatal(err)
210
+ }
211
+
212
+ outbuf := new(bytes.Buffer)
213
+ reader.WriteTo(outbuf)
214
+
215
+ err = testu.ArrComp(inbuf, outbuf.Bytes())
216
+ if err != nil {
217
+ t.Fatal(err)
218
+ }
219
+
220
+}
221
+
222
+func TestReaderSzie(t *testing.T) {
223
+ dserv := testu.GetDAGServ()
224
+ size := int64(1024)
225
+ _, node := testu.GetRandomNode(t, dserv, size)
226
+ ctx, closer := context.WithCancel(context.Background())
227
+ defer closer()
228
+
229
+ reader, err := NewDagReader(ctx, node, dserv)
230
+ if err != nil {
231
+ t.Fatal(err)
232
+ }
233
+
234
+ if reader.Size() != uint64(size) {
235
+ t.Fatal("wrong reader size")
236
+ }
237
+}
238
+
239
func readByte(t testing.TB, reader *DagReader) byte {
240
out := make([]byte, 1)
241
c, err := reader.Read(out)