增加单元测试功能

Author: alex cai

README.md

@@ -27,6 +27,7 @@ go-db-models
   - 根据某个字段查询单行记录的函数，例如根据id字段进行查询
   - 多行结果查询函数
   - 返回多行结果时，可以选择生成list（默认）或map格式
+- 单元测试文件`*_tb_gen_test.go`
 
 `*_tb_gen.go`文件样例如下：
 
@@ -115,6 +116,42 @@ func AdProjectQuery(db *sql.DB, queryString string) (ad_project []*AdProjectTabl
 }
 ```
 
+## 生成单元测试文档
+
+在生成相应的结构和查询函数的同时，也会生成相应的单元测试文件，单元测试文件的文件名如`*_tb_gen_test.go`，连接数据库时，使用了一个`testGetDb`的函数，需要在外部自己实现，如`vim init_test.go`:
+
+```go
+package models
+
+import (
+	"database/sql"
+	"fmt"
+	_ "github.com/go-sql-driver/mysql"
+)
+
+const (
+	testHost     string = "dsp-dev-go"
+	testDB       string = "dsp_sys"
+	testUser     string = "dsp_sys"
+	testPassword string = "Dsp_sys-pwd"
+	testPort     int    = 3306
+)
+
+// 获取数据库连接
+// 单元测试文件会调用该方法获得数据库链接
+func testGetDb() *sql.DB {
+	// Open database connection
+	conn_string := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s", testUser, testPassword, testHost, testPort, testDB)
+
+	db, err := sql.Open("mysql", conn_string)
+	if err != nil {
+		panic(err)
+	}
+
+	return db
+}
+```
+
 ## Json数据表定义文件
 
 如果只是希望生成部分的数据表，则可以使用该文件。该文件的结构如：

help.go

@@ -1,7 +1,7 @@
 package main
 
 const (
-	versionText = "go-db-models v1.0"
+	versionText = "go-db-models v1.1"
 
 	usageText = `GOMODELSSCAN
     根据定义从数据库生成定义结构，每个数据表生成一个对应go文件，包括数据结构定义，及两种基本的查询功能，文件名和数据表名对应。如果json-filepath为空，则生成所有的数据表的结构体。

models/gen_file.go

@@ -60,6 +60,33 @@ func GenFile(table ParseTable) error {
 	return nil
 }
 
+// 生成各个数据表的测试文件：*_tb_gen_test.go
+func GenTestFile(table ParseTable) error {
+	outFile := table.Name + "_tb_gen_test.go"
+	fout, err := os.Create(outFile)
+	if err != nil {
+		return err
+	}
+	defer fout.Close()
+
+	fnMap := template.FuncMap{
+		"AddBackquote":      AddBackquote,
+		"Format2Title":      Format2Title,
+		"Format2StructTag":  Format2StructTag,
+		"Format2StructName": Format2StructName,
+	}
+	code, err := template.New("gomodels").Funcs(fnMap).Parse(testCodeTemplate)
+	if err != nil {
+		return err
+	}
+
+	if err := code.Execute(fout, table); err != nil {
+		return err
+	}
+
+	return nil
+}
+
 // 将下划线分割的字符串改为驼峰格式的字符串
 // 如：hello_world => HelloWorld
 func Format2StructName(str string) string {

models/parse_table.go

@@ -135,6 +135,12 @@ func ParseTablesStruct(tables []Table, packageName string, modelsConf *JsonConf)
 			return nil, err
 		}
 
+		// 生成代码文件
+		err = GenTestFile(ptable)
+		if err != nil {
+			return nil, err
+		}
+
 		parseTables = append(parseTables, ptable)
 	}
 

models/template.go

@@ -87,6 +87,7 @@ func do{{.Name|Format2StructName}}Query(db *sql.DB, queryString string) ({{.Name
 	}
 	defer rows.Close()
 
+	{{if .MapIndex.Name}}{{.Name}} = make(map[{{.MapIndex.Type}}]*{{.Name|Format2StructName}}Table){{end}}
 	for rows.Next() {
 		var oneRow = &{{.Name|Format2StructName}}Table{}
 		err = rows.Scan( {{range $k, $v := .Fields}}
@@ -97,10 +98,8 @@ func do{{.Name|Format2StructName}}Query(db *sql.DB, queryString string) ({{.Name
 		}
 
 		{{if .MapIndex.Name}}
-		{{.Name}}[oneRow.{{.MapIndex.Name|Format2StructName}}] = oneRow
-		{{else}}
-		{{.Name}} = append({{.Name}}, oneRow)
-		{{end}}
+		{{.Name}}[oneRow.{{.MapIndex.Name|Format2StructName}}] = oneRow{{else}}
+		{{.Name}} = append({{.Name}}, oneRow){{end}}
 	}
 
 	return {{.Name}}, nil