Fixed token mongo store

Author: LyricTian

store/token/mongo.go

@@ -35,7 +35,7 @@ func NewMongoStore(cfg *MongoConfig) (store oauth2.TokenStore, err error) {
 	// 创建自动过期索引
 	err = handler.C(cfg.C).EnsureIndex(mgo.Index{
 		Key:         []string{"ExpiredAt"},
-		ExpireAfter: time.Second,
+		ExpireAfter: time.Second * 1,
 	})
 	if err != nil {
 		return
@@ -122,7 +122,7 @@ func (ms *MongoStore) RemoveByRefresh(refresh string) (err error) {
 func (ms *MongoStore) get(find interface{}) (info oauth2.TokenInfo, err error) {
 	ms.handler.CHandle(ms.cfg.C, func(c *mgo.Collection) {
 		var tm models.Token
-		aerr := c.Find(find).Select(bson.M{"_id": 0}).One(&tm)
+		aerr := c.Find(find).Select(bson.M{"_id": 0}).Sort("-_id").One(&tm)
 		if aerr != nil {
 			if aerr == mgo.ErrNotFound {
 				return

store/token/mongo_test.go

@@ -18,34 +18,12 @@ func TestMongoStore(t *testing.T) {
 		store, err := NewMongoStore(cfg)
 		So(err, ShouldBeNil)
 
-		Convey("Test mongo store access", func() {
+		Convey("Test access token store", func() {
 			testAccessStore(store)
 		})
 
-		Convey("Test mongo store refresh", func() {
+		Convey("Test refresh token store", func() {
 			testRefreshStore(store)
 		})
 	})
 }
-
-func TestMongoStoreAccessExpired(t *testing.T) {
-	Convey("Test mongo store access token expired", t, func() {
-		cfg := &MongoConfig{
-			URL: mongoURL,
-		}
-		store, err := NewMongoStore(cfg)
-		So(err, ShouldBeNil)
-		testAccessExpired(store)
-	})
-}
-
-func TestMongoStoreRefreshExpired(t *testing.T) {
-	Convey("Test mongo store refresh token expired", t, func() {
-		cfg := &MongoConfig{
-			URL: mongoURL,
-		}
-		store, err := NewMongoStore(cfg)
-		So(err, ShouldBeNil)
-		testRefreshExpired(store)
-	})
-}

store/token/redis_test.go

@@ -14,34 +14,12 @@ func TestRedisStore(t *testing.T) {
 		store, err := NewRedisStore(cfg)
 		So(err, ShouldBeNil)
 
-		Convey("Test redis store access", func() {
+		Convey("Test access token store", func() {
 			testAccessStore(store)
 		})
 
-		Convey("Test redis store refresh", func() {
+		Convey("Test refresh token store", func() {
 			testRefreshStore(store)
 		})
 	})
 }
-
-func TestRedisStoreAccessExpired(t *testing.T) {
-	Convey("Test redis store access token expired", t, func() {
-		cfg := &RedisConfig{
-			Addr: "192.168.33.70:6379",
-		}
-		store, err := NewRedisStore(cfg)
-		So(err, ShouldBeNil)
-		testAccessExpired(store)
-	})
-}
-
-func TestRedisStoreRefreshExpired(t *testing.T) {
-	Convey("Test redis store refresh token expired", t, func() {
-		cfg := &RedisConfig{
-			Addr: "192.168.33.70:6379",
-		}
-		store, err := NewRedisStore(cfg)
-		So(err, ShouldBeNil)
-		testRefreshExpired(store)
-	})
-}

store/token/token_test.go

@@ -37,14 +37,14 @@ func testAccessStore(store oauth2.TokenStore) {
 func testRefreshStore(store oauth2.TokenStore) {
 	info := &models.Token{
 		ClientID:         "1",
-		UserID:           "1_1",
+		UserID:           "1_2",
 		RedirectURI:      "http://localhost/",
 		Scope:            "all",
 		AuthType:         oauth2.Code.String(),
-		Access:           "1_1_2",
+		Access:           "1_2_1",
 		AccessCreateAt:   time.Now(),
 		AccessExpiresIn:  time.Second * 5,
-		Refresh:          "1_1_2_1",
+		Refresh:          "1_2_2",
 		RefreshCreateAt:  time.Now(),
 		RefreshExpiresIn: time.Minute * 1,
 	}
@@ -62,52 +62,3 @@ func testRefreshStore(store oauth2.TokenStore) {
 	So(err, ShouldBeNil)
 	So(rinfo, ShouldBeNil)
 }
-
-func testAccessExpired(store oauth2.TokenStore) {
-	info := &models.Token{
-		ClientID:        "1",
-		UserID:          "1_2",
-		RedirectURI:     "http://localhost/",
-		Scope:           "all",
-		AuthType:        oauth2.Code.String(),
-		Access:          "1_2_1",
-		AccessCreateAt:  time.Now(),
-		AccessExpiresIn: time.Second * 1,
-	}
-	err := store.Create(info)
-	So(err, ShouldBeNil)
-
-	time.Sleep(time.Millisecond * 3000)
-
-	ainfo, err := store.GetByAccess(info.GetAccess())
-	So(err, ShouldBeNil)
-	So(ainfo, ShouldBeNil)
-}
-
-func testRefreshExpired(store oauth2.TokenStore) {
-	info := &models.Token{
-		ClientID:         "1",
-		UserID:           "1_3",
-		RedirectURI:      "http://localhost/",
-		Scope:            "all",
-		AuthType:         oauth2.Code.String(),
-		Access:           "1_3_1",
-		AccessCreateAt:   time.Now(),
-		AccessExpiresIn:  time.Second * 2,
-		Refresh:          "1_3_2",
-		RefreshCreateAt:  time.Now(),
-		RefreshExpiresIn: time.Second * 1,
-	}
-	err := store.Create(info)
-	So(err, ShouldBeNil)
-
-	time.Sleep(time.Millisecond * 3000)
-
-	ainfo, err := store.GetByAccess(info.GetAccess())
-	So(err, ShouldBeNil)
-	So(ainfo, ShouldBeNil)
-
-	rinfo, err := store.GetByRefresh(info.GetRefresh())
-	So(err, ShouldBeNil)
-	So(rinfo, ShouldBeNil)
-}