Add manager test

Author: LyricTian

manage/manage_test.go

@@ -0,0 +1,27 @@
+package manage
+
+import (
+	"testing"
+
+	. "github.com/smartystreets/goconvey/convey"
+	"gopkg.in/oauth2.v2/generates"
+	"gopkg.in/oauth2.v2/models"
+	"gopkg.in/oauth2.v2/store/client"
+	"gopkg.in/oauth2.v2/store/token"
+)
+
+func TestManager(t *testing.T) {
+	Convey("Manager Test", t, func() {
+		manager := NewManager()
+
+		manager.MapClientModel(models.NewClient())
+		manager.MapTokenModel(models.NewToken())
+		manager.MapAuthorizeGenerate(generates.NewAuthorizeGenerate())
+		manager.MapAccessGenerate(generates.NewAccessGenerate())
+		manager.MapClientStorage(client.NewTempStore())
+		manager.MustTokenStorage(token.NewRedisStore(
+			&token.RedisConfig{Addr: "192.168.33.70:6379"},
+		))
+
+	})
+}

manage/manager.go

@@ -78,8 +78,8 @@ func (m *Manager) MapAuthorizeGenerate(gen oauth2.AuthorizeGenerate) {
 	m.injector.Map(gen)
 }
 
-// MapTokenGenerate 注入访问令牌生成接口
-func (m *Manager) MapTokenGenerate(gen oauth2.AccessGenerate) {
+// MapAccessGenerate 注入访问令牌生成接口
+func (m *Manager) MapAccessGenerate(gen oauth2.AccessGenerate) {
 	if gen == nil {
 		panic(ErrNilValue)
 	}

manage/util_test.go

@@ -7,8 +7,10 @@ import (
 )
 
 func TestUtil(t *testing.T) {
-	Convey("ValidateURI Test", t, func() {
-		err := ValidateURI("http://www.example.com", "http://www.example.com/cb?code=xxx")
-		So(err, ShouldBeNil)
+	Convey("Util Test", t, func() {
+		Convey("ValidateURI Test", func() {
+			err := ValidateURI("http://www.example.com", "http://www.example.com/cb?code=xxx")
+			So(err, ShouldBeNil)
+		})
 	})
 }

models/client.go

@@ -1,5 +1,10 @@
 package models
 
+// NewClient 创建客户端模型实例
+func NewClient() *Client {
+	return &Client{}
+}
+
 // Client 客户端信息
 type Client struct {
 	ID     string // 客户端ID

models/token.go

@@ -2,6 +2,11 @@ package models
 
 import "time"
 
+// NewToken 创建令牌模型实例
+func NewToken() *Token {
+	return &Token{}
+}
+
 // Token 令牌信息
 type Token struct {
 	ClientID         string        `bson:"ClientID"`         // 客户端标识