From 3451623c7113ade84fba66c3b1880251845f0e5f Mon Sep 17 00:00:00 2001 From: Xiang Li Date: Wed, 6 Jul 2016 09:56:20 -0700 Subject: [PATCH] etcdserver: fix TestSnap --- etcdserver/server_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/etcdserver/server_test.go b/etcdserver/server_test.go index cae030d54..0531f7203 100644 --- a/etcdserver/server_test.go +++ b/etcdserver/server_test.go @@ -845,7 +845,7 @@ func TestSnapshot(t *testing.T) { s := raft.NewMemoryStorage() s.Append([]raftpb.Entry{{Index: 1}}) st := mockstore.NewRecorder() - p := mockstorage.NewStorageRecorder("") + p := mockstorage.NewStorageRecorderStream("") srv := &EtcdServer{ Cfg: &ServerConfig{}, r: raftNode{ @@ -869,7 +869,7 @@ func TestSnapshot(t *testing.T) { if !reflect.DeepEqual(gaction[1], testutil.Action{Name: "SaveNoCopy"}) { t.Errorf("action = %s, want SaveNoCopy", gaction[1]) } - gaction = p.Action() + gaction, _ = p.Wait(1) if len(gaction) != 1 { t.Fatalf("len(action) = %d, want 1", len(gaction)) }