添加router entry 测试

Author: yumaojun03

http/router/entry.go

@@ -1,14 +1,33 @@
 package router
 
+import (
+	"fmt"
+	"strings"
+)
+
+// NewEntry 行健条目
+func NewEntry(path, method, resource string) *Entry {
+	return &Entry{
+		Path:     path,
+		Method:   method,
+		Resource: resource,
+		Labels:   map[string]string{},
+	}
+}
+
 // Entry 路由条目
 type Entry struct {
-	Path             string            `json:"path,omitempty"`
-	Method           string            `json:"method,omitempty"`
-	FunctionName     string            `json:"function_name,omitempty"`
-	Resource         string            `json:"resource,omitempty"`
-	AuthEnable       bool              `json:"auth_enable"`
-	PermissionEnable bool              `json:"permission_enable"`
-	Labels           map[string]string `json:"labels,omitempty"`
+	Path             string            `bson:"path" json:"path,omitempty"`
+	Method           string            `bson:"method" json:"method,omitempty"`
+	FunctionName     string            `bson:"function_name" json:"function_name,omitempty"`
+	Resource         string            `bson:"resource" json:"resource,omitempty"`
+	AuthEnable       bool              `bson:"auth_enable" json:"auth_enable"`
+	PermissionEnable bool              `bson:"permission_enable" json:"permission_enable"`
+	Labels           map[string]string `bson:"labels" json:"labels,omitempty"`
+}
+
+func (e *Entry) String() string {
+	return fmt.Sprintf("%s [%s] %s %v", e.Path, e.Method, e.Resource, e.Labels)
 }
 
 // AddLabel 添加Label
@@ -54,9 +73,20 @@ type EntrySet struct {
 	Items []*Entry `json:"items"`
 }
 
+func (s *EntrySet) String() string {
+	strs := []string{}
+	for i := range s.Items {
+		strs = append(strs, s.Items[i].String())
+	}
+
+	return strings.Join(strs, "\n")
+}
+
 // AddEntry 添加Entry
-func (s *EntrySet) AddEntry(e Entry) {
-	s.Items = append(s.Items, &e)
+func (s *EntrySet) AddEntry(es ...Entry) {
+	for i := range es {
+		s.Items = append(s.Items, &es[i])
+	}
 }
 
 // GetEntry 获取条目

http/router/entry_test.go

@@ -0,0 +1,25 @@
+package router_test
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"github.com/infraboard/mcube/http/label"
+	"github.com/infraboard/mcube/http/router"
+)
+
+func TestEntry(t *testing.T) {
+	should := assert.New(t)
+
+	e := router.NewEntry("/mcube/v1/", "GET", "Monkey")
+	e.EnableAuth()
+	e.EnablePermission()
+	e.AddLabel(label.Get)
+
+	should.Equal("/mcube/v1/ [GET] Monkey map[action:get]", e.String())
+
+	set := router.NewEntrySet()
+	set.AddEntry(*e, *e)
+	should.Equal("/mcube/v1/ [GET] Monkey map[action:get]\n/mcube/v1/ [GET] Monkey map[action:get]", set.String())
+}