init

tests/cfg_test.go

@@ -0,0 +1,58 @@
+package main
+
+import (
+	"log"
+	"os"
+	"testing"
+	"time"
+
+	"gopkg.in/ini.v1"
+)
+
+func TestIntCreate(t *testing.T) {
+
+	os.Remove("./my_test.cfg")
+	time.Sleep(time.Second * 1)
+
+	for i := 0; i < 3; i++ {
+
+		var portid string
+		var baud int
+		var rtimeout time.Duration
+
+		var cfg *ini.File
+		cfg, err := ini.Load("my_test.cfg")
+		if err != nil {
+			log.Println(err)
+			log.Println("加载配置my.cfg失败, 将使用默认值")
+			f, err := os.OpenFile("./my_test.cfg", os.O_CREATE|os.O_WRONLY, 0666)
+			if err != nil {
+				panic(err)
+			}
+			f.WriteString("[config]\nportid = COM1\nbaud = 9600\nreadtimeout = 5")
+			cfg, err = ini.Load("my_test.cfg")
+			if err != nil {
+				panic(err)
+			}
+		}
+
+		section := cfg.Section("config")
+		portid = section.Key("portid").MustString("COM1")
+		baud = section.Key("baud").MustInt(9600)
+		rtimeout = time.Second * time.Duration(section.Key("readtimeout").MustInt64(5))
+
+		if portid != "COM1" {
+			t.Error(`portid != "COM1"`)
+		}
+
+		if baud != 9600 {
+			t.Error(`baud != 9600`)
+		}
+
+		if rtimeout != time.Second*5 {
+			t.Error(`rtimeout != time.Second * 5`)
+		}
+	}
+
+	os.Remove("./my_test.cfg")
+}