etcdserver: add TestRecvSnapshot
parent
df70f653a4
commit
44ab66d858
|
@ -472,9 +472,6 @@ func TestSnapshot(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Applied > SnapCount should trigger a SaveSnap event
|
// Applied > SnapCount should trigger a SaveSnap event
|
||||||
// TODO: receive a snapshot from raft leader should also be able
|
|
||||||
// to trigger snapSave and also trigger a store.Recover.
|
|
||||||
// We need fake node!
|
|
||||||
func TestTriggerSnap(t *testing.T) {
|
func TestTriggerSnap(t *testing.T) {
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
n := raft.StartNode(0xBAD0, []int64{0xBAD0}, 10, 1)
|
n := raft.StartNode(0xBAD0, []int64{0xBAD0}, 10, 1)
|
||||||
|
@ -507,6 +504,34 @@ func TestTriggerSnap(t *testing.T) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestRecvSnapshot tests when it receives a snapshot from raft leader,
|
||||||
|
// it should trigger storage.SaveSnap and also store.Recover.
|
||||||
|
func TestRecvSnapshot(t *testing.T) {
|
||||||
|
n := newReadyNode(raft.Ready{Snapshot: raftpb.Snapshot{Index: 1}})
|
||||||
|
st := &storeRecorder{}
|
||||||
|
p := &storageRecorder{}
|
||||||
|
s := &EtcdServer{
|
||||||
|
Store: st,
|
||||||
|
Send: func(_ []raftpb.Message) {},
|
||||||
|
Storage: p,
|
||||||
|
Node: n,
|
||||||
|
}
|
||||||
|
|
||||||
|
s.Start()
|
||||||
|
// make goroutines move forward to receive snapshot
|
||||||
|
forceGosched()
|
||||||
|
s.Stop()
|
||||||
|
|
||||||
|
waction := []string{"Recovery"}
|
||||||
|
if g := st.Action(); !reflect.DeepEqual(g, waction) {
|
||||||
|
t.Errorf("store action = %v, want %v", g, waction)
|
||||||
|
}
|
||||||
|
waction = []string{"Save", "SaveSnap"}
|
||||||
|
if g := p.Action(); !reflect.DeepEqual(g, waction) {
|
||||||
|
t.Errorf("storage action = %v, want %v", g, waction)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// TODO: test wait trigger correctness in multi-server case
|
// TODO: test wait trigger correctness in multi-server case
|
||||||
|
|
||||||
func TestGetBool(t *testing.T) {
|
func TestGetBool(t *testing.T) {
|
||||||
|
@ -590,7 +615,10 @@ func (s *storeRecorder) Save() ([]byte, error) {
|
||||||
s.record("Save")
|
s.record("Save")
|
||||||
return nil, nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
func (s *storeRecorder) Recovery(b []byte) error { return nil }
|
func (s *storeRecorder) Recovery(b []byte) error {
|
||||||
|
s.record("Recovery")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
func (s *storeRecorder) TotalTransactions() uint64 { return 0 }
|
func (s *storeRecorder) TotalTransactions() uint64 { return 0 }
|
||||||
func (s *storeRecorder) JsonStats() []byte { return nil }
|
func (s *storeRecorder) JsonStats() []byte { return nil }
|
||||||
func (s *storeRecorder) DeleteExpiredKeys(cutoff time.Time) {
|
func (s *storeRecorder) DeleteExpiredKeys(cutoff time.Time) {
|
||||||
|
@ -636,6 +664,26 @@ func (p *storageRecorder) SaveSnap(st raftpb.Snapshot) {
|
||||||
p.record("SaveSnap")
|
p.record("SaveSnap")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type readyNode struct {
|
||||||
|
readyc chan raft.Ready
|
||||||
|
}
|
||||||
|
|
||||||
|
func newReadyNode(ready raft.Ready) *readyNode {
|
||||||
|
readyc := make(chan raft.Ready, 1)
|
||||||
|
readyc <- ready
|
||||||
|
return &readyNode{readyc: readyc}
|
||||||
|
}
|
||||||
|
func (n *readyNode) Tick() {}
|
||||||
|
func (n *readyNode) Campaign(ctx context.Context) error { return nil }
|
||||||
|
func (n *readyNode) Propose(ctx context.Context, data []byte) error { return nil }
|
||||||
|
func (n *readyNode) Configure(ctx context.Context, data []byte) error { return nil }
|
||||||
|
func (n *readyNode) Step(ctx context.Context, msg raftpb.Message) error { return nil }
|
||||||
|
func (n *readyNode) Ready() <-chan raft.Ready { return n.readyc }
|
||||||
|
func (n *readyNode) Stop() {}
|
||||||
|
func (n *readyNode) Compact(d []byte) {}
|
||||||
|
func (n *readyNode) AddNode(id int64) {}
|
||||||
|
func (n *readyNode) RemoveNode(id int64) {}
|
||||||
|
|
||||||
func TestGenID(t *testing.T) {
|
func TestGenID(t *testing.T) {
|
||||||
// Sanity check that the GenID function has been seeded appropriately
|
// Sanity check that the GenID function has been seeded appropriately
|
||||||
// (math/rand is seeded with 1 by default)
|
// (math/rand is seeded with 1 by default)
|
||||||
|
@ -648,3 +696,12 @@ func TestGenID(t *testing.T) {
|
||||||
t.Fatalf("GenID's rand seeded with 1!")
|
t.Fatalf("GenID's rand seeded with 1!")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// WARNING: This is a hack.
|
||||||
|
// Remove this when we are able to block/check the status of the go-routines.
|
||||||
|
func forceGosched() {
|
||||||
|
// possibility enough to sched upto 10 go routines.
|
||||||
|
for i := 0; i < 10000; i++ {
|
||||||
|
runtime.Gosched()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue