From 10e16a5baa78fa6e990919f35e594964a155e07c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=81ukasz=20Magiera?= <magik6k@gmail.com>
Date: Thu, 11 Jan 2018 23:34:30 +0100
Subject: [PATCH] coreapi: pin tests
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

License: MIT
Signed-off-by: Łukasz Magiera <magik6k@gmail.com>
---
 core/coreapi/pin.go      |   2 +-
 core/coreapi/pin_test.go | 209 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 210 insertions(+), 1 deletion(-)
 create mode 100644 core/coreapi/pin_test.go

diff --git a/core/coreapi/pin.go b/core/coreapi/pin.go
index bcfba1ff88d..75374371183 100644
--- a/core/coreapi/pin.go
+++ b/core/coreapi/pin.go
@@ -80,7 +80,7 @@ type badNode struct {
 }
 
 func (s *pinStatus) Ok() bool {
-	return s.Ok()
+	return s.ok
 }
 
 func (s *pinStatus) BadNodes() []coreiface.BadPinNode {
diff --git a/core/coreapi/pin_test.go b/core/coreapi/pin_test.go
new file mode 100644
index 00000000000..6ab0e5350bc
--- /dev/null
+++ b/core/coreapi/pin_test.go
@@ -0,0 +1,209 @@
+package coreapi_test
+
+import (
+	"context"
+	"strings"
+	"testing"
+)
+
+func TestPinAdd(t *testing.T) {
+	ctx := context.Background()
+	_, api, err := makeAPI(ctx)
+	if err != nil {
+		t.Error(err)
+	}
+
+	p, err := api.Unixfs().Add(ctx, strings.NewReader("foo"))
+	if err != nil {
+		t.Error(err)
+	}
+
+	err = api.Pin().Add(ctx, p)
+	if err != nil {
+		t.Error(err)
+	}
+}
+
+func TestPinSimple(t *testing.T) {
+	ctx := context.Background()
+	_, api, err := makeAPI(ctx)
+	if err != nil {
+		t.Error(err)
+	}
+
+	p, err := api.Unixfs().Add(ctx, strings.NewReader("foo"))
+	if err != nil {
+		t.Error(err)
+	}
+
+	err = api.Pin().Add(ctx, p)
+	if err != nil {
+		t.Error(err)
+	}
+
+	list, err := api.Pin().Ls(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if len(list) != 1 {
+		t.Errorf("unexpected pin list len: %d", len(list))
+	}
+
+	if list[0].Path().String() != p.String() {
+		t.Error("paths don't match")
+	}
+
+	if list[0].Type() != "recursive" {
+		t.Error("unexpected pin type")
+	}
+
+	err = api.Pin().Rm(ctx, p)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	list, err = api.Pin().Ls(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if len(list) != 0 {
+		t.Errorf("unexpected pin list len: %d", len(list))
+	}
+}
+
+func TestPinRecursive(t *testing.T) {
+	ctx := context.Background()
+	nd, api, err := makeAPI(ctx)
+	if err != nil {
+		t.Error(err)
+	}
+
+	p0, err := api.Unixfs().Add(ctx, strings.NewReader("foo"))
+	if err != nil {
+		t.Error(err)
+	}
+
+	p1, err := api.Unixfs().Add(ctx, strings.NewReader("bar"))
+	if err != nil {
+		t.Error(err)
+	}
+
+	p2, err := api.Dag().Put(ctx, strings.NewReader(`{"lnk": {"/": "`+p0.Cid().String()+`"}}`))
+	if err != nil {
+		t.Error(err)
+	}
+
+	p3, err := api.Dag().Put(ctx, strings.NewReader(`{"lnk": {"/": "`+p1.Cid().String()+`"}}`))
+	if err != nil {
+		t.Error(err)
+	}
+
+	err = api.Pin().Add(ctx, p2)
+	if err != nil {
+		t.Error(err)
+	}
+
+	err = api.Pin().Add(ctx, p3, api.Pin().WithRecursive(false))
+	if err != nil {
+		t.Error(err)
+	}
+
+	list, err := api.Pin().Ls(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if len(list) != 3 {
+		t.Errorf("unexpected pin list len: %d", len(list))
+	}
+
+	list, err = api.Pin().Ls(ctx, api.Pin().WithType("direct"))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if len(list) != 1 {
+		t.Errorf("unexpected pin list len: %d", len(list))
+	}
+
+	if list[0].Path().String() != p3.String() {
+		t.Error("unexpected path")
+	}
+
+	list, err = api.Pin().Ls(ctx, api.Pin().WithType("recursive"))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if len(list) != 1 {
+		t.Errorf("unexpected pin list len: %d", len(list))
+	}
+
+	if list[0].Path().String() != p2.String() {
+		t.Error("unexpected path")
+	}
+
+	list, err = api.Pin().Ls(ctx, api.Pin().WithType("indirect"))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if len(list) != 1 {
+		t.Errorf("unexpected pin list len: %d", len(list))
+	}
+
+	if list[0].Path().String() != p0.String() {
+		t.Error("unexpected path")
+	}
+
+	res, err := api.Pin().Verify(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+	n := 0
+	for r := range res {
+		if !r.Ok() {
+			t.Error("expected pin to be ok")
+		}
+		n++
+	}
+
+	if n != 1 {
+		t.Errorf("unexpected verify result count: %d", n)
+	}
+
+	err = nd.Blockstore.DeleteBlock(p0.Cid())
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	res, err = api.Pin().Verify(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+	n = 0
+	for r := range res {
+		if r.Ok() {
+			t.Error("expected pin to not be ok")
+		}
+
+		if len(r.BadNodes()) != 1 {
+			t.Fatalf("unexpected badNodes len")
+		}
+
+		if r.BadNodes()[0].Path().String() != p0.String() {
+			t.Error("unexpected badNode path")
+		}
+
+		if r.BadNodes()[0].Err().Error() != "merkledag: not found" {
+			t.Errorf("unexpected badNode error: %s", r.BadNodes()[0].Err().Error())
+		}
+		n++
+	}
+
+	if n != 1 {
+		t.Errorf("unexpected verify result count: %d", n)
+	}
+}