@@ -283,6 +283,13 @@ func TestServer(t *testing.T) {
283
283
0 ,
284
284
[]string {"-test-verify-only" },
285
285
},
286
+ {
287
+ "recovery_mode" ,
288
+ testBaseHCL (t , "" ) + inmemHCL ,
289
+ "" ,
290
+ 0 ,
291
+ []string {"-test-verify-only" , "-recovery" },
292
+ },
286
293
}
287
294
288
295
for _ , tc := range cases {
@@ -292,26 +299,21 @@ func TestServer(t *testing.T) {
292
299
t .Parallel ()
293
300
294
301
ui , cmd := testServerCommand (t )
295
- f , err := ioutil .TempFile ("" , "" )
296
- if err != nil {
297
- t .Fatalf ("error creating temp dir: %v" , err )
298
- }
299
- f .WriteString (tc .contents )
300
- f .Close ()
301
- defer os .Remove (f .Name ())
302
302
303
- args := append (tc .args , "-config" , f .Name ())
303
+ f , err := os .CreateTemp (t .TempDir (), "" )
304
+ require .NoErrorf (t , err , "error creating temp dir: %v" , err )
304
305
305
- code := cmd . Run ( args )
306
- output := ui . ErrorWriter . String () + ui . OutputWriter . String ( )
306
+ _ , err = f . WriteString ( tc . contents )
307
+ require . NoErrorf ( t , err , "cannot write temp file contents" )
307
308
308
- if code != tc .code {
309
- t .Errorf ("expected %d to be %d: %s" , code , tc .code , output )
310
- }
309
+ err = f .Close ()
310
+ require .NoErrorf (t , err , "unable to close temp file" )
311
311
312
- if ! strings .Contains (output , tc .exp ) {
313
- t .Fatalf ("expected %q to contain %q" , output , tc .exp )
314
- }
312
+ args := append (tc .args , "-config" , f .Name ())
313
+ code := cmd .Run (args )
314
+ output := ui .ErrorWriter .String () + ui .OutputWriter .String ()
315
+ require .Equal (t , tc .code , code , "expected %d to be %d: %s" , code , tc .code , output )
316
+ require .Contains (t , output , tc .exp , "expected %q to contain %q" , output , tc .exp )
315
317
})
316
318
}
317
319
}
0 commit comments