@@ -39,13 +39,13 @@ import (
39
39
"go.etcd.io/etcd/tests/v3/framework/e2e"
40
40
)
41
41
42
- func writeCustomV2Data (t testing.TB , epc * e2e.EtcdProcessCluster , count int ) {
42
+ func writeCustomV2Data (tb testing.TB , epc * e2e.EtcdProcessCluster , count int ) {
43
43
for i := 0 ; i < count ; i ++ {
44
44
if err := e2e .CURLPut (epc , e2e.CURLReq {
45
45
Endpoint : "/v2/keys/foo" , Value : "bar" + fmt .Sprint (i ),
46
46
Expected : expect.ExpectedResponse {Value : `{"action":"set","node":{"key":"/foo","value":"bar` + fmt .Sprint (i )},
47
47
}); err != nil {
48
- t .Fatalf ("failed put with curl (%v)" , err )
48
+ tb .Fatalf ("failed put with curl (%v)" , err )
49
49
}
50
50
}
51
51
}
@@ -191,47 +191,47 @@ func TestV2DeprecationSnapshotRecover(t *testing.T) {
191
191
assert .NoError (t , epc .Close ())
192
192
}
193
193
194
- func runEtcdAndCreateSnapshot (t testing.TB , serverVersion e2e.ClusterVersion , dataDir string , snapshotCount uint64 ) * e2e.EtcdProcessCluster {
194
+ func runEtcdAndCreateSnapshot (tb testing.TB , serverVersion e2e.ClusterVersion , dataDir string , snapshotCount uint64 ) * e2e.EtcdProcessCluster {
195
195
cfg := e2e .ConfigStandalone (* e2e .NewConfig (
196
196
e2e .WithVersion (serverVersion ),
197
197
e2e .WithDataDirPath (dataDir ),
198
198
e2e .WithSnapshotCount (snapshotCount ),
199
199
e2e .WithKeepDataDir (true ),
200
200
))
201
- epc , err := e2e .NewEtcdProcessCluster (t .Context (), t , e2e .WithConfig (cfg ))
202
- assert .NoError (t , err )
201
+ epc , err := e2e .NewEtcdProcessCluster (tb .Context (), tb , e2e .WithConfig (cfg ))
202
+ assert .NoError (tb , err )
203
203
return epc
204
204
}
205
205
206
- func addAndRemoveKeysAndMembers (ctx context.Context , t testing.TB , cc * e2e.EtcdctlV3 , snapshotCount uint64 ) (members []uint64 ) {
206
+ func addAndRemoveKeysAndMembers (ctx context.Context , tb testing.TB , cc * e2e.EtcdctlV3 , snapshotCount uint64 ) (members []uint64 ) {
207
207
// Execute some non-trivial key&member operation
208
208
var i uint64
209
209
for i = 0 ; i < snapshotCount * 3 ; i ++ {
210
210
err := cc .Put (ctx , fmt .Sprintf ("%d" , i ), "1" , config.PutOptions {})
211
- require .NoError (t , err )
211
+ require .NoError (tb , err )
212
212
}
213
213
member1 , err := cc .MemberAddAsLearner (ctx , "member1" , []string {"http://127.0.0.1:2000" })
214
- require .NoError (t , err )
214
+ require .NoError (tb , err )
215
215
members = append (members , member1 .Member .ID )
216
216
217
217
for i = 0 ; i < snapshotCount * 2 ; i ++ {
218
218
_ , err = cc .Delete (ctx , fmt .Sprintf ("%d" , i ), config.DeleteOptions {})
219
- require .NoError (t , err )
219
+ require .NoError (tb , err )
220
220
}
221
221
_ , err = cc .MemberRemove (ctx , member1 .Member .ID )
222
- require .NoError (t , err )
222
+ require .NoError (tb , err )
223
223
224
224
for i = 0 ; i < snapshotCount ; i ++ {
225
225
err = cc .Put (ctx , fmt .Sprintf ("%d" , i ), "2" , config.PutOptions {})
226
- require .NoError (t , err )
226
+ require .NoError (tb , err )
227
227
}
228
228
member2 , err := cc .MemberAddAsLearner (ctx , "member2" , []string {"http://127.0.0.1:2001" })
229
- require .NoError (t , err )
229
+ require .NoError (tb , err )
230
230
members = append (members , member2 .Member .ID )
231
231
232
232
for i = 0 ; i < snapshotCount / 2 ; i ++ {
233
233
err = cc .Put (ctx , fmt .Sprintf ("%d" , i ), "3" , config.PutOptions {})
234
- assert .NoError (t , err )
234
+ assert .NoError (tb , err )
235
235
}
236
236
return members
237
237
}
@@ -240,39 +240,39 @@ func filterSnapshotFiles(path string) bool {
240
240
return strings .HasSuffix (path , ".snap" )
241
241
}
242
242
243
- func assertSnapshotsMatch (t testing.TB , firstDataDir , secondDataDir string , patch func ([]byte ) []byte ) {
244
- lg := zaptest .NewLogger (t )
243
+ func assertSnapshotsMatch (tb testing.TB , firstDataDir , secondDataDir string , patch func ([]byte ) []byte ) {
244
+ lg := zaptest .NewLogger (tb )
245
245
firstFiles , err := fileutil .ListFiles (firstDataDir , filterSnapshotFiles )
246
- require .NoError (t , err )
246
+ require .NoError (tb , err )
247
247
secondFiles , err := fileutil .ListFiles (secondDataDir , filterSnapshotFiles )
248
- require .NoError (t , err )
249
- assert .NotEmpty (t , firstFiles )
250
- assert .NotEmpty (t , secondFiles )
251
- assert .Len (t , secondFiles , len (firstFiles ))
248
+ require .NoError (tb , err )
249
+ assert .NotEmpty (tb , firstFiles )
250
+ assert .NotEmpty (tb , secondFiles )
251
+ assert .Len (tb , secondFiles , len (firstFiles ))
252
252
sort .Strings (firstFiles )
253
253
sort .Strings (secondFiles )
254
254
for i := 0 ; i < len (firstFiles ); i ++ {
255
255
firstSnapshot , err := snap .Read (lg , firstFiles [i ])
256
- require .NoError (t , err )
256
+ require .NoError (tb , err )
257
257
secondSnapshot , err := snap .Read (lg , secondFiles [i ])
258
- require .NoError (t , err )
259
- assertMembershipEqual (t , openSnap (patch (firstSnapshot .Data )), openSnap (patch (secondSnapshot .Data )))
258
+ require .NoError (tb , err )
259
+ assertMembershipEqual (tb , openSnap (patch (firstSnapshot .Data )), openSnap (patch (secondSnapshot .Data )))
260
260
}
261
261
}
262
262
263
- func assertMembershipEqual (t testing.TB , firstStore v2store.Store , secondStore v2store.Store ) {
264
- rc1 := membership .NewCluster (zaptest .NewLogger (t ))
263
+ func assertMembershipEqual (tb testing.TB , firstStore v2store.Store , secondStore v2store.Store ) {
264
+ rc1 := membership .NewCluster (zaptest .NewLogger (tb ))
265
265
rc1 .SetStore (firstStore )
266
266
rc1 .Recover (func (lg * zap.Logger , v * semver.Version ) {})
267
267
268
- rc2 := membership .NewCluster (zaptest .NewLogger (t ))
268
+ rc2 := membership .NewCluster (zaptest .NewLogger (tb ))
269
269
rc2 .SetStore (secondStore )
270
270
rc2 .Recover (func (lg * zap.Logger , v * semver.Version ) {})
271
271
272
272
// membership should match
273
273
if ! reflect .DeepEqual (rc1 .Members (), rc2 .Members ()) {
274
- t .Logf ("memberids_from_last_version = %+v, member_ids_from_current_version = %+v" , rc1 .MemberIDs (), rc2 .MemberIDs ())
275
- t .Errorf ("members_from_last_version_snapshot = %+v, members_from_current_version_snapshot %+v" , rc1 .Members (), rc2 .Members ())
274
+ tb .Logf ("memberids_from_last_version = %+v, member_ids_from_current_version = %+v" , rc1 .MemberIDs (), rc2 .MemberIDs ())
275
+ tb .Errorf ("members_from_last_version_snapshot = %+v, members_from_current_version_snapshot %+v" , rc1 .Members (), rc2 .Members ())
276
276
}
277
277
}
278
278
0 commit comments