From 1d4d11619d27a421ff8121991e465b617fbc9f63 Mon Sep 17 00:00:00 2001
From: xiantang <zhujingdi1998@gmail.com>
Date: Sun, 5 Jun 2022 21:18:30 +0800
Subject: [PATCH] Add more coverage for air init  (#288)

* Fix dlv port address already in use

# Conflicts:
#	runner/engine.go
#	runner/util_linux.go

* move to engine test
---
 runner/config.go      |  3 +++
 runner/engine_test.go | 26 ++++++++++++++++++++++++++
 2 files changed, 29 insertions(+)

diff --git a/runner/config.go b/runner/config.go
index 85fcfd96..ceace24c 100644
--- a/runner/config.go
+++ b/runner/config.go
@@ -177,7 +177,10 @@ func defaultConfig() config {
 		Bin:          "./tmp/main",
 		Log:          "build-errors.log",
 		IncludeExt:   []string{"go", "tpl", "tmpl", "html"},
+		IncludeDir:   []string{},
+		ExcludeFile:  []string{},
 		ExcludeDir:   []string{"assets", "tmp", "vendor", "testdata"},
+		ArgsBin:      []string{},
 		ExcludeRegex: []string{"_test.go"},
 		Delay:        1000,
 		StopOnError:  true,
diff --git a/runner/engine_test.go b/runner/engine_test.go
index 5c461bca..5bbbefe2 100644
--- a/runner/engine_test.go
+++ b/runner/engine_test.go
@@ -461,3 +461,29 @@ func TestRebuildWhenRunCmdUsingDLV(t *testing.T) {
 	t.Logf("engine stopped")
 	assert.True(t, checkPortConnectionRefused(port))
 }
+
+func TestWriteDefaultConfig(t *testing.T) {
+	port, f := GetPort()
+	f()
+	t.Logf("port: %d", port)
+
+	tmpDir := initTestEnv(t, port)
+	// change dir to tmpDir
+	if err := os.Chdir(tmpDir); err != nil {
+		t.Fatal(err)
+	}
+	writeDefaultConfig()
+	// check the file is exist
+	if _, err := os.Stat(dftTOML); err != nil {
+		t.Fatal(err)
+	}
+
+	// check the file content is right
+	actual, err := readConfig(dftTOML)
+	if err != nil {
+		t.Fatal(err)
+	}
+	expect := defaultConfig()
+
+	assert.Equal(t, expect, *actual)
+}