Refine go test

Author: Wuvist

README.md

@@ -67,9 +67,16 @@
 
 ## 如何参与项目
 
-* 写测试 Writing test
-* 代码审查 Code review
-* 添加新的template
+### 写测试 Writing test
+
+#### go test
+
+* 先完成编译`go build`
+* 将编译出来的可执行文件设置到`PROTOAPI_EXE`环境变量
+* 再执行`go test`
+
+### 代码审查 Code review
+
 
 ### 添加新的template
 

main.go

@@ -34,6 +34,7 @@ func main() {
 
 	// when no any parameter and not reading from char device, treat it as being called by protoc
 	if len(args) == 1 && err == nil && (stat.Mode()&os.ModeCharDevice) == 0 {
+		// check for PROTOAPI_PORT, if set means it's in go test
 		protoapiPort := os.Getenv("PROTOAPI_PORT")
 		if protoapiPort != "" {
 			resp, err := http.Post("http://127.0.0.1:"+protoapiPort+"/", "application/protobuf", os.Stdin)
@@ -63,7 +64,6 @@ func main() {
 			util.Die(err)
 		}
 	} else {
-		go serv()
 		cmd.Execute()
 	}
 }

main_test.go

@@ -40,6 +40,12 @@ func test(t *testing.T, args string) {
 }
 
 func TestCmd(t *testing.T) {
+	executable := os.Getenv("PROTOAPI_EXE")
+	if executable == "" {
+		t.Error("PROTOAPI_EXE is not set")
+		return
+	}
+
 	test(t, "gen --lang=go test/result/go test/proto/test.proto")
 	// test(t, "gen --lang=go test/result/go test/proto/calc.proto")
 	// test(t, "gen --lang=go test/result/go test/proto/todolist.proto")