bee tools add router & test

2025-07-17 18:22:17 +00:00 · 2013-12-23 23:28:31 +08:00
parent d4cd5a0671
commit df4f44fde1
2 changed files with 110 additions and 21 deletions
--- a/apiapp.go
+++ b/apiapp.go
@@ -36,9 +36,13 @@ In the appname folder has the follow struct:
 	│   └── app.conf
 	├── controllers
 	│   └── default.go
+	├── routers
+	│   └── router.go
+	├── tests
+	│   └── default_test.go		
 	├── main.go
 	└── models
-	    └── object.go             
+	    └── object.go            

 `,
 }
@@ -53,8 +57,8 @@ copyrequestbody = true
 var apiMaingo = `package main

 import (
-	"github.com/astaxie/beego"
-	"{{.Appname}}/controllers"
+	_ "{{.Appname}}/routes"
+	"github.com/astaxie/beego"	
 )

 //		Objects
@@ -67,11 +71,21 @@ import (
 //	/object/<objectId>	DELETE					Deleting Objects

 func main() {
-	beego.RESTRouter("/object", &controllers.ObjectController{})
-	beego.Router("/ping", &controllers.ObjectController{},"get:Ping")
 	beego.Run()
 }
 `
+var apirouter = `package routes
+
+import (
+	"{{.Appname}}/controllers"
+	"github.com/astaxie/beego"
+)
+
+func init() {
+    beego.RESTRouter("/object", &controllers.ObjectController{})
+}
+`
+
 var apiModels = `package models

 import (
@@ -192,22 +206,34 @@ func (this *ObjectController) Ping() {

 `

-var apiTests = `package tests
+var apiTests = `package test

 import (
-    "testing"
-	beetest "github.com/astaxie/beego/testing"
-	"io/ioutil"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	_ "{{.Appname}}/routes"
+		
+	"github.com/astaxie/beego"
+	. "github.com/smartystreets/goconvey/convey"
 )

-func TestHelloWorld(t *testing.T) {
-	request:=beetest.Get("/ping")
-	response,_:=request.Response()
-	defer response.Body.Close()
-	contents, _ := ioutil.ReadAll(response.Body)
-	if string(contents)!="pong"{
-        t.Errorf("response sould be pong")
-    }
+// TestGet is a sample to run an endpoint test
+func TestGet(t *testing.T) {
+	r, _ := http.NewRequest("GET", "/", nil)
+	w := httptest.NewRecorder()
+	beego.BeeApp.Handlers.ServeHTTP(w, r)
+	
+	beego.Trace("testing", "TestGet", "Code[%d]\n%s", w.Code, w.Body.String())
+	
+	Convey("Subject: Test Station Endpoint\n", t, func() {
+	        Convey("Status Code Should Be 200", func() {
+	                So(w.Code, ShouldEqual, 200)
+	        })
+	        Convey("The Result Should Not Be Empty", func() {
+	                So(w.Body.Len(), ShouldBeGreaterThan, 0)
+	        })
+	})
 }

 `
@@ -233,6 +259,8 @@ func createapi(cmd *Command, args []string) {
 	os.Mkdir(path.Join(apppath, "controllers"), 0755)
 	fmt.Println("create controllers:", path.Join(apppath, "controllers"))
 	os.Mkdir(path.Join(apppath, "models"), 0755)
+	fmt.Println(path.Join(apppath, "routes") + string(path.Separator))
+	os.Mkdir(path.Join(apppath, "routes"), 0755)
 	fmt.Println("create models:", path.Join(apppath, "models"))
 	os.Mkdir(path.Join(apppath, "tests"), 0755)
 	fmt.Println("create tests:", path.Join(apppath, "tests"))
@@ -247,7 +275,11 @@ func createapi(cmd *Command, args []string) {

 	fmt.Println("create tests default.go:", path.Join(apppath, "tests", "default_test.go"))
 	writetofile(path.Join(apppath, "tests", "default_test.go"),
-		apiTests)
+		strings.Replace(apiTests, "{{.Appname}}", packpath, -1))
+
+	fmt.Println("create routers router.go:", path.Join(apppath, "routers", "router.go"))
+	writetofile(path.Join(apppath, "routers", "router.go"),
+		strings.Replace(apirouter, "{{.Appname}}", packpath, -1))

 	fmt.Println("create models object.go:", path.Join(apppath, "models", "object.go"))
 	writetofile(path.Join(apppath, "models", "object.go"), apiModels)
--- a/new.go
+++ b/new.go
@@ -37,7 +37,11 @@ The [appname] folder has following structure:
    |- controllers
         |- default.go
    |- models
-    |- static
+    |- routes
+         |- router.go	
+    |- tests
+         |- default_test.go
+	|- static
         |- js
         |- css
         |- img             
@@ -102,6 +106,10 @@ func createApp(cmd *Command, args []string) {
 	fmt.Println(path.Join(apppath, "controllers") + string(path.Separator))
 	os.Mkdir(path.Join(apppath, "models"), 0755)
 	fmt.Println(path.Join(apppath, "models") + string(path.Separator))
+	os.Mkdir(path.Join(apppath, "routes"), 0755)
+	fmt.Println(path.Join(apppath, "routes") + string(path.Separator))
+	os.Mkdir(path.Join(apppath, "tests"), 0755)
+	fmt.Println(path.Join(apppath, "tests") + string(path.Separator))
 	os.Mkdir(path.Join(apppath, "static"), 0755)
 	fmt.Println(path.Join(apppath, "static") + string(path.Separator))
 	os.Mkdir(path.Join(apppath, "static", "js"), 0755)
@@ -121,6 +129,12 @@ func createApp(cmd *Command, args []string) {
 	fmt.Println(path.Join(apppath, "views", "index.tpl"))
 	writetofile(path.Join(apppath, "views", "index.tpl"), indextpl)

+	fmt.Println(path.Join(apppath, "routers", "router.go"))
+	writetofile(path.Join(apppath, "routers", "router.go"), strings.Replace(router, "{{.Appname}}", strings.Join(strings.Split(apppath[len(appsrcpath)+1:], string(path.Separator)), "/"), -1))
+
+	fmt.Println(path.Join(apppath, "tests", "default_test.go"))
+	writetofile(path.Join(apppath, "tests", "default_test.go"), strings.Replace(test, "{{.Appname}}", strings.Join(strings.Split(apppath[len(appsrcpath)+1:], string(path.Separator)), "/"), -1))
+
 	fmt.Println(path.Join(apppath, "main.go"))
 	writetofile(path.Join(apppath, "main.go"), strings.Replace(maingo, "{{.Appname}}", strings.Join(strings.Split(apppath[len(appsrcpath)+1:], string(path.Separator)), "/"), -1))

@@ -135,16 +149,59 @@ runmode = dev
 var maingo = `package main

 import (
-	"{{.Appname}}/controllers"
+	_ "{{.Appname}}/routes"
 	"github.com/astaxie/beego"
 )

 func main() {
-	beego.Router("/", &controllers.MainController{})
 	beego.Run()
 }

 `
+var router = `package routes
+
+import (
+	"{{.Appname}}/controllers"
+	"github.com/astaxie/beego"
+)
+
+func init() {
+    beego.Router("/", &controllers.MainController{})
+}
+`
+
+var test = `package test
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	_ "{{.Appname}}/routes"
+		
+	"github.com/astaxie/beego"
+	. "github.com/smartystreets/goconvey/convey"
+)
+
+// TestMain is a sample to run an endpoint test
+func TestMain(t *testing.T) {
+	r, _ := http.NewRequest("GET", "/object", nil)
+	w := httptest.NewRecorder()
+	beego.BeeApp.Handlers.ServeHTTP(w, r)
+	
+	beego.Trace("testing", "TestMain", "Code[%d]\n%s", w.Code, w.Body.String())
+	
+	Convey("Subject: Test Station Endpoint\n", t, func() {
+	        Convey("Status Code Should Be 200", func() {
+	                So(w.Code, ShouldEqual, 200)
+	        })
+	        Convey("The Result Should Not Be Empty", func() {
+	                So(w.Body.Len(), ShouldBeGreaterThan, 0)
+	        })
+	})
+}
+
+`
+
 var controllers = `package controllers

 import (