ipfs-cluster/log_op_test.go

package ipfscluster

import (
	"context"
	"testing"

	cid "github.com/ipfs/go-cid"

	"github.com/ipfs/ipfs-cluster/api"
	"github.com/ipfs/ipfs-cluster/state/mapstate"
	"github.com/ipfs/ipfs-cluster/test"
)

func TestApplyToPin(t *testing.T) {
	op := &LogOp{
		Cid:       api.CidArgSerial{Cid: test.TestCid1},
		Type:      LogOpPin,
		ctx:       context.Background(),
		rpcClient: test.NewMockRPCClient(t),
	}

	st := mapstate.NewMapState()
	op.ApplyTo(st)
	pins := st.List()
	if len(pins) != 1 || pins[0].Cid.String() != test.TestCid1 {
		t.Error("the state was not modified correctly")
	}
}

func TestApplyToUnpin(t *testing.T) {
	op := &LogOp{
		Cid:       api.CidArgSerial{Cid: test.TestCid1},
		Type:      LogOpUnpin,
		ctx:       context.Background(),
		rpcClient: test.NewMockRPCClient(t),
	}

	st := mapstate.NewMapState()
	c, _ := cid.Decode(test.TestCid1)
	st.Add(api.CidArg{Cid: c, Everywhere: true})
	op.ApplyTo(st)
	pins := st.List()
	if len(pins) != 0 {
		t.Error("the state was not modified correctly")
	}
}

func TestApplyToBadState(t *testing.T) {
	defer func() {
		if r := recover(); r == nil {
			t.Error("should have recovered an error")
		}
	}()

	op := &LogOp{
		Cid:       api.CidArgSerial{Cid: test.TestCid1},
		Type:      LogOpUnpin,
		ctx:       context.Background(),
		rpcClient: test.NewMockRPCClient(t),
	}

	var st interface{}
	op.ApplyTo(st)
}

// func TestApplyToBadCid(t *testing.T) {
// 	defer func() {
// 		if r := recover(); r == nil {
// 			t.Error("should have recovered an error")
// 		}
// 	}()

// 	op := &LogOp{
// 		Cid:       api.CidArgSerial{Cid: "agadfaegf"},
// 		Type:      LogOpPin,
// 		ctx:       context.Background(),
// 		rpcClient: test.NewMockRPCClient(t),
// 	}

// 	st := mapstate.NewMapState()
// 	op.ApplyTo(st)
// }