Skip to content

Commit 8e683af

Browse files
committed
Apply code review comments
1 parent 068f042 commit 8e683af

File tree

1 file changed

+18
-33
lines changed

1 file changed

+18
-33
lines changed

server/etcdserver/server_test.go

Lines changed: 18 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -97,8 +97,7 @@ func TestApplyRepeat(t *testing.T) {
9797
storage: mockstorage.NewStorageRecorder(""),
9898
transport: newNopTransporter(),
9999
})
100-
s := newServer(t, nil, testEtcdServerConfig{
101-
lg: zaptest.NewLogger(t),
100+
s := newServer(t, nil, &testEtcdServerConfig{
102101
r: r,
103102
v2store: st,
104103
cluster: cl,
@@ -163,8 +162,7 @@ func TestV2SetMemberAttributes(t *testing.T) {
163162
ServerFeatureGate: features.NewDefaultServerFeatureGate("test", nil),
164163
}
165164

166-
srv := newServer(t, nil, testEtcdServerConfig{
167-
lg: zaptest.NewLogger(t),
165+
srv := newServer(t, nil, &testEtcdServerConfig{
168166
v2store: mockstore.NewRecorder(),
169167
cluster: cl,
170168
be: be,
@@ -244,7 +242,7 @@ func TestV2SetClusterVersion(t *testing.T) {
244242

245243
func TestApplyConfStateWithRestart(t *testing.T) {
246244
n := newNodeRecorder()
247-
srv := newServer(t, n)
245+
srv := newServer(t, n, nil)
248246
defer srv.Cleanup()
249247

250248
assert.Equal(t, uint64(0), srv.consistIndex.ConsistentIndex())
@@ -341,10 +339,9 @@ type testEtcdServerConfig struct {
341339
cfg config.ServerConfig
342340
}
343341

344-
func newServer(t *testing.T, recorder *nodeRecorder, cfgs ...testEtcdServerConfig) *EtcdServer {
345-
var cfg testEtcdServerConfig
346-
if len(cfgs) > 0 {
347-
cfg = cfgs[0]
342+
func newServer(t *testing.T, recorder *nodeRecorder, cfg *testEtcdServerConfig) *EtcdServer {
343+
if cfg == nil {
344+
cfg = &testEtcdServerConfig{}
348345
}
349346
lg := cfg.lg
350347
if lg == nil {
@@ -464,8 +461,7 @@ func TestApplyConfChangeError(t *testing.T) {
464461
}
465462
for i, tt := range tests {
466463
n := newNodeRecorder()
467-
srv := newServer(t, n, testEtcdServerConfig{
468-
lg: zaptest.NewLogger(t),
464+
srv := newServer(t, n, &testEtcdServerConfig{
469465
cluster: cl,
470466
})
471467
_, err := srv.applyConfChange(tt.cc, nil, true)
@@ -502,8 +498,7 @@ func TestApplyConfChangeShouldStop(t *testing.T) {
502498
Node: newNodeNop(),
503499
transport: newNopTransporter(),
504500
})
505-
srv := newServer(t, nil, testEtcdServerConfig{
506-
lg: lg,
501+
srv := newServer(t, nil, &testEtcdServerConfig{
507502
r: r,
508503
cluster: cl,
509504
})
@@ -548,8 +543,7 @@ func TestApplyConfigChangeUpdatesConsistIndex(t *testing.T) {
548543
schema.CreateMetaBucket(be.BatchTx())
549544

550545
ci := cindex.NewConsistentIndex(be)
551-
srv := newServer(t, nil, testEtcdServerConfig{
552-
lg: lg,
546+
srv := newServer(t, nil, &testEtcdServerConfig{
553547
r: realisticRaftNode(lg, 1, nil),
554548
cluster: cl,
555549
w: wait.New(),
@@ -641,8 +635,7 @@ func TestApplyMultiConfChangeShouldStop(t *testing.T) {
641635
transport: newNopTransporter(),
642636
})
643637
ci := cindex.NewFakeConsistentIndex(0)
644-
srv := newServer(t, nil, testEtcdServerConfig{
645-
lg: lg,
638+
srv := newServer(t, nil, &testEtcdServerConfig{
646639
r: r,
647640
cluster: cl,
648641
w: wait.New(),
@@ -691,8 +684,7 @@ func TestSnapshotDisk(t *testing.T) {
691684
raftStorage: s,
692685
storage: p,
693686
})
694-
srv := newServer(t, nil, testEtcdServerConfig{
695-
lg: zaptest.NewLogger(t),
687+
srv := newServer(t, nil, &testEtcdServerConfig{
696688
r: r,
697689
v2store: st,
698690
be: be,
@@ -814,8 +806,7 @@ func TestSnapshotOrdering(t *testing.T) {
814806
ServerFeatureGate: features.NewDefaultServerFeatureGate("test", lg),
815807
}
816808

817-
s := newServer(t, nil, testEtcdServerConfig{
818-
lg: lg,
809+
s := newServer(t, nil, &testEtcdServerConfig{
819810
cfg: cfg,
820811
r: r,
821812
v2store: st,
@@ -903,8 +894,7 @@ func TestConcurrentApplyAndSnapshotV3(t *testing.T) {
903894
raftStorage: rs,
904895
})
905896
ci := cindex.NewConsistentIndex(be)
906-
s := newServer(t, nil, testEtcdServerConfig{
907-
lg: lg,
897+
s := newServer(t, nil, &testEtcdServerConfig{
908898
cfg: config.ServerConfig{
909899
Logger: lg,
910900
DataDir: testdir,
@@ -1156,8 +1146,7 @@ func TestUpdateMember(t *testing.T) {
11561146
storage: mockstorage.NewStorageRecorder(""),
11571147
transport: newNopTransporter(),
11581148
})
1159-
s := newServer(t, nil, testEtcdServerConfig{
1160-
lg: lg,
1149+
s := newServer(t, nil, &testEtcdServerConfig{
11611150
r: r,
11621151
v2store: st,
11631152
cluster: cl,
@@ -1195,8 +1184,7 @@ func TestPublishV3(t *testing.T) {
11951184
lg := zaptest.NewLogger(t)
11961185
be, _ := betesting.NewDefaultTmpBackend(t)
11971186
defer betesting.Close(t, be)
1198-
srv := newServer(t, nil, testEtcdServerConfig{
1199-
lg: lg,
1187+
srv := newServer(t, nil, &testEtcdServerConfig{
12001188
cfg: config.ServerConfig{Logger: lg, TickMs: 1, SnapshotCatchUpEntries: DefaultSnapshotCatchUpEntries, MaxRequestBytes: 1000},
12011189
r: newRaftNode(raftNodeConfig{lg: lg, Node: n}),
12021190
cluster: &membership.RaftCluster{},
@@ -1237,8 +1225,7 @@ func TestPublishV3Stopped(t *testing.T) {
12371225
Node: newNodeNop(),
12381226
transport: newNopTransporter(),
12391227
})
1240-
srv := newServer(t, nil, testEtcdServerConfig{
1241-
lg: zaptest.NewLogger(t),
1228+
srv := newServer(t, nil, &testEtcdServerConfig{
12421229
cfg: config.ServerConfig{Logger: zaptest.NewLogger(t), TickMs: 1, SnapshotCatchUpEntries: DefaultSnapshotCatchUpEntries},
12431230
r: r,
12441231
cluster: &membership.RaftCluster{},
@@ -1262,8 +1249,7 @@ func TestPublishV3Retry(t *testing.T) {
12621249
lg := zaptest.NewLogger(t)
12631250
be, _ := betesting.NewDefaultTmpBackend(t)
12641251
defer betesting.Close(t, be)
1265-
srv := newServer(t, nil, testEtcdServerConfig{
1266-
lg: lg,
1252+
srv := newServer(t, nil, &testEtcdServerConfig{
12671253
cfg: config.ServerConfig{Logger: lg, TickMs: 1, SnapshotCatchUpEntries: DefaultSnapshotCatchUpEntries, MaxRequestBytes: 1000},
12681254
r: newRaftNode(raftNodeConfig{lg: lg, Node: n}),
12691255
cluster: &membership.RaftCluster{},
@@ -1312,8 +1298,7 @@ func TestUpdateVersionV3(t *testing.T) {
13121298
lg := zaptest.NewLogger(t)
13131299
be, _ := betesting.NewDefaultTmpBackend(t)
13141300
defer betesting.Close(t, be)
1315-
srv := newServer(t, nil, testEtcdServerConfig{
1316-
lg: lg,
1301+
srv := newServer(t, nil, &testEtcdServerConfig{
13171302
cfg: config.ServerConfig{Logger: lg, TickMs: 1, SnapshotCatchUpEntries: DefaultSnapshotCatchUpEntries, MaxRequestBytes: 1000},
13181303
r: newRaftNode(raftNodeConfig{lg: zaptest.NewLogger(t), Node: n}),
13191304
cluster: &membership.RaftCluster{},

0 commit comments

Comments
 (0)