@@ -116,18 +116,19 @@ func TestBackupRestore1(t *testing.T) {
116
116
117
117
func TestBackupRestore2 (t * testing.T ) {
118
118
tmpdir , err := ioutil .TempDir ("" , "badger-test" )
119
- require .NoError (t , err )
120
-
119
+ if err != nil {
120
+ t .Fatal (err )
121
+ }
121
122
defer removeDir (tmpdir )
122
123
123
124
s1Path := filepath .Join (tmpdir , "test1" )
124
125
s2Path := filepath .Join (tmpdir , "test2" )
125
126
s3Path := filepath .Join (tmpdir , "test3" )
126
127
127
128
db1 , err := Open (getTestOptions (s1Path ))
128
- require . NoError ( t , err )
129
-
130
- defer db1 . Close ()
129
+ if err != nil {
130
+ t . Fatal ( err )
131
+ }
131
132
key1 := []byte ("key1" )
132
133
key2 := []byte ("key2" )
133
134
rawValue := []byte ("NotLongValue" )
@@ -138,30 +139,35 @@ func TestBackupRestore2(t *testing.T) {
138
139
}
139
140
return tx .SetEntry (NewEntry (key2 , rawValue ))
140
141
})
141
- require .NoError (t , err )
142
-
142
+ if err != nil {
143
+ t .Fatal (err )
144
+ }
143
145
for i := byte (1 ); i < N ; i ++ {
144
146
err = db1 .Update (func (tx * Txn ) error {
145
147
if err := tx .SetEntry (NewEntry (append (key1 , i ), rawValue )); err != nil {
146
148
return err
147
149
}
148
150
return tx .SetEntry (NewEntry (append (key2 , i ), rawValue ))
149
151
})
150
- require .NoError (t , err )
151
-
152
+ if err != nil {
153
+ t .Fatal (err )
154
+ }
152
155
}
153
156
var backup bytes.Buffer
154
157
_ , err = db1 .Backup (& backup , 0 )
155
- require .NoError (t , err )
156
-
158
+ if err != nil {
159
+ t .Fatal (err )
160
+ }
157
161
fmt .Println ("backup1 length:" , backup .Len ())
158
162
159
163
db2 , err := Open (getTestOptions (s2Path ))
160
- require . NoError ( t , err )
161
-
162
- defer db2 . Close ()
164
+ if err != nil {
165
+ t . Fatal ( err )
166
+ }
163
167
err = db2 .Load (& backup , 16 )
164
- require .NoError (t , err )
168
+ if err != nil {
169
+ t .Fatal (err )
170
+ }
165
171
166
172
for i := byte (1 ); i < N ; i ++ {
167
173
err = db2 .View (func (tx * Txn ) error {
@@ -182,8 +188,9 @@ func TestBackupRestore2(t *testing.T) {
182
188
}
183
189
return nil
184
190
})
185
- require .NoError (t , err )
186
-
191
+ if err != nil {
192
+ t .Fatal (err )
193
+ }
187
194
}
188
195
189
196
for i := byte (1 ); i < N ; i ++ {
@@ -193,22 +200,26 @@ func TestBackupRestore2(t *testing.T) {
193
200
}
194
201
return tx .SetEntry (NewEntry (append (key2 , i ), rawValue ))
195
202
})
196
- require .NoError (t , err )
197
-
203
+ if err != nil {
204
+ t .Fatal (err )
205
+ }
198
206
}
199
207
200
208
backup .Reset ()
201
209
_ , err = db2 .Backup (& backup , 0 )
202
- require .NoError (t , err )
203
-
210
+ if err != nil {
211
+ t .Fatal (err )
212
+ }
204
213
fmt .Println ("backup2 length:" , backup .Len ())
205
214
db3 , err := Open (getTestOptions (s3Path ))
206
- require . NoError ( t , err )
207
-
208
- defer db3 . Close ()
215
+ if err != nil {
216
+ t . Fatal ( err )
217
+ }
209
218
210
219
err = db3 .Load (& backup , 16 )
211
- require .NoError (t , err )
220
+ if err != nil {
221
+ t .Fatal (err )
222
+ }
212
223
213
224
for i := byte (1 ); i < N ; i ++ {
214
225
err = db3 .View (func (tx * Txn ) error {
@@ -229,8 +240,9 @@ func TestBackupRestore2(t *testing.T) {
229
240
}
230
241
return nil
231
242
})
232
- require .NoError (t , err )
233
-
243
+ if err != nil {
244
+ t .Fatal (err )
245
+ }
234
246
}
235
247
236
248
}
@@ -298,8 +310,9 @@ func TestBackup(t *testing.T) {
298
310
}
299
311
t .Run ("disk mode" , func (t * testing.T ) {
300
312
tmpdir , err := ioutil .TempDir ("" , "badger-test" )
301
- require .NoError (t , err )
302
-
313
+ if err != nil {
314
+ t .Fatal (err )
315
+ }
303
316
defer removeDir (tmpdir )
304
317
opt := DefaultOptions (filepath .Join (tmpdir , "backup0" ))
305
318
runBadgerTest (t , & opt , func (t * testing.T , db * DB ) {
@@ -317,9 +330,11 @@ func TestBackup(t *testing.T) {
317
330
318
331
func TestBackupRestore3 (t * testing.T ) {
319
332
var bb bytes.Buffer
320
- tmpdir , err := ioutil .TempDir ("" , "badger-test" )
321
- require .NoError (t , err )
322
333
334
+ tmpdir , err := ioutil .TempDir ("" , "badger-test" )
335
+ if err != nil {
336
+ t .Fatal (err )
337
+ }
323
338
defer removeDir (tmpdir )
324
339
325
340
N := 1000
@@ -328,9 +343,10 @@ func TestBackupRestore3(t *testing.T) {
328
343
// backup
329
344
{
330
345
db1 , err := Open (DefaultOptions (filepath .Join (tmpdir , "backup1" )))
331
- require .NoError (t , err )
346
+ if err != nil {
347
+ t .Fatal (err )
348
+ }
332
349
333
- defer db1 .Close ()
334
350
require .NoError (t , populateEntries (db1 , entries ))
335
351
336
352
_ , err = db1 .Backup (& bb , 0 )
@@ -342,9 +358,9 @@ func TestBackupRestore3(t *testing.T) {
342
358
343
359
// restore
344
360
db2 , err := Open (DefaultOptions (filepath .Join (tmpdir , "restore1" )))
345
- require . NoError ( t , err )
346
-
347
- defer db2 . Close ()
361
+ if err != nil {
362
+ t . Fatal ( err )
363
+ }
348
364
require .NoError (t , db2 .Load (& bb , 16 ))
349
365
350
366
// verify
@@ -374,8 +390,9 @@ func TestBackupRestore3(t *testing.T) {
374
390
375
391
func TestBackupLoadIncremental (t * testing.T ) {
376
392
tmpdir , err := ioutil .TempDir ("" , "badger-test" )
377
- require .NoError (t , err )
378
-
393
+ if err != nil {
394
+ t .Fatal (err )
395
+ }
379
396
defer removeDir (tmpdir )
380
397
381
398
N := 100
@@ -386,9 +403,9 @@ func TestBackupLoadIncremental(t *testing.T) {
386
403
// backup
387
404
{
388
405
db1 , err := Open (DefaultOptions (filepath .Join (tmpdir , "backup2" )))
389
- require . NoError ( t , err )
390
-
391
- defer db1 . Close ()
406
+ if err != nil {
407
+ t . Fatal ( err )
408
+ }
392
409
393
410
require .NoError (t , populateEntries (db1 , entries ))
394
411
since , err := db1 .Backup (& bb , 0 )
@@ -446,10 +463,9 @@ func TestBackupLoadIncremental(t *testing.T) {
446
463
447
464
// restore
448
465
db2 , err := Open (getTestOptions (filepath .Join (tmpdir , "restore2" )))
449
- require .NoError (t , err )
450
-
451
- defer db2 .Close ()
452
-
466
+ if err != nil {
467
+ t .Fatal (err )
468
+ }
453
469
require .NoError (t , db2 .Load (& bb , 16 ))
454
470
455
471
// verify
0 commit comments