create test function

2020-12-21 11:42:36 +08:00
parent 8e417699c0
commit cd302234af
2 changed files with 28 additions and 23 deletions
@@ -19,30 +19,36 @@ import (
 	"fmt"
 	"github.com/spf13/cobra"
 	"github.com/tarm/serial"
-	"os"
 	"it2000.com.cn/tele-recv/database"
+	"os"
 )

 // testCmd represents the test command
 var testCmd = &cobra.Command{
 	Use:   "test",
-	Short: "Test Config",
-	Long: `Test config file.
-Test serial port`,
-	Run: func(cmd *cobra.Command, args []string) {
-		fmt.Println("test called")
-		fmt.Println("device :",device)
-		c := &serial.Config{Name: device, Baud: baudrate, Size: bytesize}
-		s, err := serial.OpenPort(c)
-		if err != nil {
-			fmt.Println(err)
-			os.Exit(1)
-		}
-		fmt.Println(c , "opened")
-		_ = s.Close()
+	Short: "Test running environment",
+	Long: `loading config file
+Test serial port
+Test load sqlite database and init table`,
+	Run: test,
+}

-		database.Init(dbFile, dbInit)
-	},
+func test(cmd *cobra.Command, args []string) {
+	fmt.Println("test called")
+	fmt.Println("device :", device)
+	c := &serial.Config{Name: device, Baud: baudrate, Size: bytesize}
+	s, err := serial.OpenPort(c)
+	if err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+	fmt.Println(c, "opened")
+	_ = s.Close()
+
+	err = database.Init(dbFile, dbInit)
+	if err != nil {
+		println("create database error ", err)
+	}
 }

 func init() {