ipfs-cluster/rpc_api_test.go

package ipfscluster

import (
	"errors"
	"testing"
	"time"

	rpc "github.com/hsanjuan/go-libp2p-gorpc"
	cid "github.com/ipfs/go-cid"
	crypto "github.com/libp2p/go-libp2p-crypto"
	peer "github.com/libp2p/go-libp2p-peer"
)

var errBadCid = errors.New("this is an expected error when using errorCid")

type mockService struct{}

func mockRPCClient(t *testing.T) *rpc.Client {
	s := rpc.NewServer(nil, "mock")
	c := rpc.NewClientWithServer(nil, "mock", s)
	err := s.RegisterName("Cluster", &mockService{})
	if err != nil {
		t.Fatal(err)
	}
	return c
}

func (mock *mockService) Pin(in *CidArg, out *struct{}) error {
	if in.Cid == errorCid {
		return errBadCid
	}
	return nil
}

func (mock *mockService) Unpin(in *CidArg, out *struct{}) error {
	if in.Cid == errorCid {
		return errBadCid
	}
	return nil
}

func (mock *mockService) PinList(in struct{}, out *[]string) error {
	*out = []string{testCid, testCid2, testCid3}
	return nil
}

func (mock *mockService) ID(in struct{}, out *IDSerial) error {
	_, pubkey, _ := crypto.GenerateKeyPair(
		DefaultConfigCrypto,
		DefaultConfigKeyLength)
	*out = ID{
		ID:        testPeerID,
		PublicKey: pubkey,
		Version:   "0.0.mock",
		IPFS: IPFSID{
			ID: testPeerID,
		},
	}.ToSerial()
	return nil
}

func (mock *mockService) Version(in struct{}, out *string) error {
	*out = "0.0.mock"
	return nil
}

func (mock *mockService) Peers(in struct{}, out *[]IDSerial) error {
	id := IDSerial{}
	mock.ID(in, &id)

	*out = []IDSerial{id}
	return nil
}

func (mock *mockService) PeerAdd(in MultiaddrSerial, out *IDSerial) error {
	id := IDSerial{}
	mock.ID(struct{}{}, &id)
	*out = id
	return nil
}

func (mock *mockService) PeerRemove(in peer.ID, out *struct{}) error {
	return nil
}

func (mock *mockService) StatusAll(in struct{}, out *[]GlobalPinInfo) error {
	c1, _ := cid.Decode(testCid1)
	c2, _ := cid.Decode(testCid2)
	c3, _ := cid.Decode(testCid3)
	*out = []GlobalPinInfo{
		{
			Cid: c1,
			PeerMap: map[peer.ID]PinInfo{
				testPeerID: {
					CidStr: testCid1,
					Peer:   testPeerID,
					Status: TrackerStatusPinned,
					TS:     time.Now(),
				},
			},
		},
		{
			Cid: c2,
			PeerMap: map[peer.ID]PinInfo{
				testPeerID: {
					CidStr: testCid2,
					Peer:   testPeerID,
					Status: TrackerStatusPinning,
					TS:     time.Now(),
				},
			},
		},
		{
			Cid: c3,
			PeerMap: map[peer.ID]PinInfo{
				testPeerID: {
					CidStr: testCid3,
					Peer:   testPeerID,
					Status: TrackerStatusPinError,
					TS:     time.Now(),
				},
			},
		},
	}
	return nil
}

func (mock *mockService) Status(in *CidArg, out *GlobalPinInfo) error {
	if in.Cid == errorCid {
		return errBadCid
	}
	c1, _ := cid.Decode(testCid1)
	*out = GlobalPinInfo{
		Cid: c1,
		PeerMap: map[peer.ID]PinInfo{
			testPeerID: {
				CidStr: testCid1,
				Peer:   testPeerID,
				Status: TrackerStatusPinned,
				TS:     time.Now(),
			},
		},
	}
	return nil
}

func (mock *mockService) SyncAll(in struct{}, out *[]GlobalPinInfo) error {
	return mock.StatusAll(in, out)
}

func (mock *mockService) Sync(in *CidArg, out *GlobalPinInfo) error {
	return mock.Status(in, out)
}

func (mock *mockService) StateSync(in struct{}, out *[]PinInfo) error {
	*out = []PinInfo{}
	return nil
}

func (mock *mockService) Recover(in *CidArg, out *GlobalPinInfo) error {
	return mock.Status(in, out)
}

func (mock *mockService) Track(in *CidArg, out *struct{}) error {
	return nil
}

func (mock *mockService) Untrack(in *CidArg, out *struct{}) error {
	return nil
}