2016-12-23 18:35:37 +00:00
|
|
|
package ipfscluster
|
|
|
|
|
|
|
|
import (
|
|
|
|
"errors"
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
2017-01-25 11:14:39 +00:00
|
|
|
rpc "github.com/hsanjuan/go-libp2p-gorpc"
|
2016-12-23 18:35:37 +00:00
|
|
|
cid "github.com/ipfs/go-cid"
|
2017-01-24 15:19:23 +00:00
|
|
|
crypto "github.com/libp2p/go-libp2p-crypto"
|
2016-12-23 18:35:37 +00:00
|
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
|
|
)
|
|
|
|
|
2017-01-23 11:08:04 +00:00
|
|
|
var errBadCid = errors.New("this is an expected error when using errorCid")
|
2016-12-23 18:35:37 +00:00
|
|
|
|
|
|
|
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 {
|
2016-12-28 15:25:24 +00:00
|
|
|
return errBadCid
|
2016-12-23 18:35:37 +00:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (mock *mockService) Unpin(in *CidArg, out *struct{}) error {
|
|
|
|
if in.Cid == errorCid {
|
2016-12-28 15:25:24 +00:00
|
|
|
return errBadCid
|
2016-12-23 18:35:37 +00:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (mock *mockService) PinList(in struct{}, out *[]string) error {
|
|
|
|
*out = []string{testCid, testCid2, testCid3}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-01-26 18:59:31 +00:00
|
|
|
func (mock *mockService) ID(in struct{}, out *IDSerial) error {
|
2017-01-24 15:19:23 +00:00
|
|
|
_, pubkey, _ := crypto.GenerateKeyPair(
|
|
|
|
DefaultConfigCrypto,
|
|
|
|
DefaultConfigKeyLength)
|
|
|
|
*out = ID{
|
|
|
|
ID: testPeerID,
|
|
|
|
PublicKey: pubkey,
|
|
|
|
Version: "0.0.mock",
|
2017-01-26 18:59:31 +00:00
|
|
|
IPFS: IPFSID{
|
|
|
|
ID: testPeerID,
|
|
|
|
},
|
|
|
|
}.ToSerial()
|
2017-01-24 15:19:23 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2016-12-23 18:35:37 +00:00
|
|
|
func (mock *mockService) Version(in struct{}, out *string) error {
|
|
|
|
*out = "0.0.mock"
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-01-26 18:59:31 +00:00
|
|
|
func (mock *mockService) Peers(in struct{}, out *[]IDSerial) error {
|
|
|
|
id := IDSerial{}
|
|
|
|
mock.ID(in, &id)
|
|
|
|
|
|
|
|
*out = []IDSerial{id}
|
2016-12-23 18:35:37 +00:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-01-30 12:12:25 +00:00
|
|
|
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
|
|
|
|
}
|
|
|
|
|
2017-01-25 18:38:23 +00:00
|
|
|
func (mock *mockService) StatusAll(in struct{}, out *[]GlobalPinInfo) error {
|
2016-12-23 18:35:37 +00:00
|
|
|
c1, _ := cid.Decode(testCid1)
|
|
|
|
c2, _ := cid.Decode(testCid2)
|
|
|
|
c3, _ := cid.Decode(testCid3)
|
|
|
|
*out = []GlobalPinInfo{
|
2017-01-24 11:39:08 +00:00
|
|
|
{
|
2016-12-23 18:35:37 +00:00
|
|
|
Cid: c1,
|
2017-01-25 17:07:19 +00:00
|
|
|
PeerMap: map[peer.ID]PinInfo{
|
2017-01-24 11:39:08 +00:00
|
|
|
testPeerID: {
|
2016-12-23 18:35:37 +00:00
|
|
|
CidStr: testCid1,
|
|
|
|
Peer: testPeerID,
|
2017-01-25 17:07:19 +00:00
|
|
|
Status: TrackerStatusPinned,
|
2016-12-23 18:35:37 +00:00
|
|
|
TS: time.Now(),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2017-01-24 11:39:08 +00:00
|
|
|
{
|
2016-12-23 18:35:37 +00:00
|
|
|
Cid: c2,
|
2017-01-25 17:07:19 +00:00
|
|
|
PeerMap: map[peer.ID]PinInfo{
|
2017-01-24 11:39:08 +00:00
|
|
|
testPeerID: {
|
2016-12-23 18:35:37 +00:00
|
|
|
CidStr: testCid2,
|
|
|
|
Peer: testPeerID,
|
2017-01-25 17:07:19 +00:00
|
|
|
Status: TrackerStatusPinning,
|
2016-12-23 18:35:37 +00:00
|
|
|
TS: time.Now(),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
2017-01-24 11:39:08 +00:00
|
|
|
{
|
2016-12-23 18:35:37 +00:00
|
|
|
Cid: c3,
|
2017-01-25 17:07:19 +00:00
|
|
|
PeerMap: map[peer.ID]PinInfo{
|
2017-01-24 11:39:08 +00:00
|
|
|
testPeerID: {
|
2016-12-23 18:35:37 +00:00
|
|
|
CidStr: testCid3,
|
|
|
|
Peer: testPeerID,
|
2017-01-25 17:07:19 +00:00
|
|
|
Status: TrackerStatusPinError,
|
2016-12-23 18:35:37 +00:00
|
|
|
TS: time.Now(),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-01-25 18:38:23 +00:00
|
|
|
func (mock *mockService) Status(in *CidArg, out *GlobalPinInfo) error {
|
2016-12-23 18:35:37 +00:00
|
|
|
if in.Cid == errorCid {
|
2016-12-28 15:25:24 +00:00
|
|
|
return errBadCid
|
2016-12-23 18:35:37 +00:00
|
|
|
}
|
|
|
|
c1, _ := cid.Decode(testCid1)
|
|
|
|
*out = GlobalPinInfo{
|
|
|
|
Cid: c1,
|
2017-01-25 17:07:19 +00:00
|
|
|
PeerMap: map[peer.ID]PinInfo{
|
2017-01-24 11:39:08 +00:00
|
|
|
testPeerID: {
|
2016-12-23 18:35:37 +00:00
|
|
|
CidStr: testCid1,
|
|
|
|
Peer: testPeerID,
|
2017-01-25 17:07:19 +00:00
|
|
|
Status: TrackerStatusPinned,
|
2016-12-23 18:35:37 +00:00
|
|
|
TS: time.Now(),
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-01-25 18:38:23 +00:00
|
|
|
func (mock *mockService) SyncAll(in struct{}, out *[]GlobalPinInfo) error {
|
|
|
|
return mock.StatusAll(in, out)
|
2016-12-23 18:35:37 +00:00
|
|
|
}
|
|
|
|
|
2017-01-25 18:38:23 +00:00
|
|
|
func (mock *mockService) Sync(in *CidArg, out *GlobalPinInfo) error {
|
|
|
|
return mock.Status(in, out)
|
2016-12-23 18:35:37 +00:00
|
|
|
}
|
2017-01-23 11:08:04 +00:00
|
|
|
|
|
|
|
func (mock *mockService) StateSync(in struct{}, out *[]PinInfo) error {
|
|
|
|
*out = []PinInfo{}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-01-25 18:38:23 +00:00
|
|
|
func (mock *mockService) Recover(in *CidArg, out *GlobalPinInfo) error {
|
|
|
|
return mock.Status(in, out)
|
|
|
|
}
|
|
|
|
|
2017-01-23 11:08:04 +00:00
|
|
|
func (mock *mockService) Track(in *CidArg, out *struct{}) error {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (mock *mockService) Untrack(in *CidArg, out *struct{}) error {
|
|
|
|
return nil
|
|
|
|
}
|