9df6344a07
License: MIT Signed-off-by: Hector Sanjuan <hector@protocol.ai>
84 lines
1.6 KiB
Go
84 lines
1.6 KiB
Go
package raft
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/ipfs/ipfs-cluster/api"
|
|
"github.com/ipfs/ipfs-cluster/state/mapstate"
|
|
"github.com/ipfs/ipfs-cluster/test"
|
|
)
|
|
|
|
func TestApplyToPin(t *testing.T) {
|
|
ctx := context.Background()
|
|
cc := testingConsensus(t, 1)
|
|
op := &LogOp{
|
|
Cid: api.PinCid(test.TestCid1),
|
|
Type: LogOpPin,
|
|
consensus: cc,
|
|
}
|
|
defer cleanRaft(1)
|
|
defer cc.Shutdown(ctx)
|
|
|
|
st := mapstate.NewMapState()
|
|
op.ApplyTo(st)
|
|
pins := st.List(ctx)
|
|
if len(pins) != 1 || !pins[0].Cid.Equals(test.TestCid1) {
|
|
t.Error("the state was not modified correctly")
|
|
}
|
|
}
|
|
|
|
func TestApplyToUnpin(t *testing.T) {
|
|
ctx := context.Background()
|
|
cc := testingConsensus(t, 1)
|
|
op := &LogOp{
|
|
Cid: api.PinCid(test.TestCid1),
|
|
Type: LogOpUnpin,
|
|
consensus: cc,
|
|
}
|
|
defer cleanRaft(1)
|
|
defer cc.Shutdown(ctx)
|
|
|
|
st := mapstate.NewMapState()
|
|
st.Add(ctx, testPin(test.TestCid1))
|
|
op.ApplyTo(st)
|
|
pins := st.List(ctx)
|
|
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.PinCid(test.TestCid1),
|
|
Type: LogOpUnpin,
|
|
}
|
|
|
|
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.PinSerial{Cid: "agadfaegf"},
|
|
// Type: LogOpPin,
|
|
// ctx: context.Background(),
|
|
// rpcClient: test.NewMockRPCClient(t),
|
|
// }
|
|
|
|
// st := mapstate.NewMapState()
|
|
// op.ApplyTo(st)
|
|
// }
|