Skip to content

Commit 1539da7

Browse files
authored
test: simplify cmd/micro tests (zyedidia#3470)
1 parent a3211dc commit 1539da7

File tree

1 file changed

+26
-47
lines changed

1 file changed

+26
-47
lines changed

cmd/micro/micro_test.go

Lines changed: 26 additions & 47 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ package main
22

33
import (
44
"fmt"
5-
"io/ioutil"
65
"log"
76
"os"
87
"testing"
@@ -26,7 +25,7 @@ func init() {
2625
func startup(args []string) (tcell.SimulationScreen, error) {
2726
var err error
2827

29-
tempDir, err = ioutil.TempDir("", "micro_test")
28+
tempDir, err = os.MkdirTemp("", "micro_test")
3029
if err != nil {
3130
return nil, err
3231
}
@@ -164,20 +163,22 @@ func findBuffer(file string) *buffer.Buffer {
164163
return buf
165164
}
166165

167-
func createTestFile(name string, content string) (string, error) {
168-
testf, err := ioutil.TempFile("", name)
166+
func createTestFile(t *testing.T, content string) string {
167+
f, err := os.CreateTemp(t.TempDir(), "")
169168
if err != nil {
170-
return "", err
169+
t.Fatal(err)
171170
}
171+
defer func() {
172+
if err := f.Close(); err != nil {
173+
t.Fatal(err)
174+
}
175+
}()
172176

173-
if _, err := testf.Write([]byte(content)); err != nil {
174-
return "", err
175-
}
176-
if err := testf.Close(); err != nil {
177-
return "", err
177+
if _, err := f.WriteString(content); err != nil {
178+
t.Fatal(err)
178179
}
179180

180-
return testf.Name(), nil
181+
return f.Name()
181182
}
182183

183184
func TestMain(m *testing.M) {
@@ -194,18 +195,12 @@ func TestMain(m *testing.M) {
194195
}
195196

196197
func TestSimpleEdit(t *testing.T) {
197-
file, err := createTestFile("micro_simple_edit_test", "base content")
198-
if err != nil {
199-
t.Error(err)
200-
return
201-
}
202-
defer os.Remove(file)
198+
file := createTestFile(t, "base content")
203199

204200
openFile(file)
205201

206202
if findBuffer(file) == nil {
207-
t.Errorf("Could not find buffer %s", file)
208-
return
203+
t.Fatalf("Could not find buffer %s", file)
209204
}
210205

211206
injectKey(tcell.KeyEnter, rune(tcell.KeyEnter), tcell.ModNone)
@@ -223,28 +218,21 @@ func TestSimpleEdit(t *testing.T) {
223218

224219
injectKey(tcell.KeyCtrlS, rune(tcell.KeyCtrlS), tcell.ModCtrl)
225220

226-
data, err := ioutil.ReadFile(file)
221+
data, err := os.ReadFile(file)
227222
if err != nil {
228-
t.Error(err)
229-
return
223+
t.Fatal(err)
230224
}
231225

232226
assert.Equal(t, "firstfoobar\nbase content\n", string(data))
233227
}
234228

235229
func TestMouse(t *testing.T) {
236-
file, err := createTestFile("micro_mouse_test", "base content")
237-
if err != nil {
238-
t.Error(err)
239-
return
240-
}
241-
defer os.Remove(file)
230+
file := createTestFile(t, "base content")
242231

243232
openFile(file)
244233

245234
if findBuffer(file) == nil {
246-
t.Errorf("Could not find buffer %s", file)
247-
return
235+
t.Fatalf("Could not find buffer %s", file)
248236
}
249237

250238
// buffer:
@@ -275,10 +263,9 @@ func TestMouse(t *testing.T) {
275263
// base content
276264
injectKey(tcell.KeyCtrlS, rune(tcell.KeyCtrlS), tcell.ModCtrl)
277265

278-
data, err := ioutil.ReadFile(file)
266+
data, err := os.ReadFile(file)
279267
if err != nil {
280-
t.Error(err)
281-
return
268+
t.Fatal(err)
282269
}
283270

284271
assert.Equal(t, "firstline\nsecondline\nbase content\n", string(data))
@@ -301,18 +288,12 @@ Ernleȝe test_string æðelen
301288
`
302289

303290
func TestSearchAndReplace(t *testing.T) {
304-
file, err := createTestFile("micro_search_replace_test", srTestStart)
305-
if err != nil {
306-
t.Error(err)
307-
return
308-
}
309-
defer os.Remove(file)
291+
file := createTestFile(t, srTestStart)
310292

311293
openFile(file)
312294

313295
if findBuffer(file) == nil {
314-
t.Errorf("Could not find buffer %s", file)
315-
return
296+
t.Fatalf("Could not find buffer %s", file)
316297
}
317298

318299
injectKey(tcell.KeyCtrlE, rune(tcell.KeyCtrlE), tcell.ModCtrl)
@@ -321,10 +302,9 @@ func TestSearchAndReplace(t *testing.T) {
321302

322303
injectKey(tcell.KeyCtrlS, rune(tcell.KeyCtrlS), tcell.ModCtrl)
323304

324-
data, err := ioutil.ReadFile(file)
305+
data, err := os.ReadFile(file)
325306
if err != nil {
326-
t.Error(err)
327-
return
307+
t.Fatal(err)
328308
}
329309

330310
assert.Equal(t, srTest2, string(data))
@@ -337,10 +317,9 @@ func TestSearchAndReplace(t *testing.T) {
337317

338318
injectKey(tcell.KeyCtrlS, rune(tcell.KeyCtrlS), tcell.ModCtrl)
339319

340-
data, err = ioutil.ReadFile(file)
320+
data, err = os.ReadFile(file)
341321
if err != nil {
342-
t.Error(err)
343-
return
322+
t.Fatal(err)
344323
}
345324

346325
assert.Equal(t, srTest3, string(data))

0 commit comments

Comments
 (0)