mmm-go / ability · 提交

转到一个项目

作者 yangfu
提交 df0a235c4e940ce7b7ea209b2015c45d7886e348 1 个父辈 9477cfbf

mock 测试

内嵌 并排对比
正在显示 11 个修改的文件 包含 139 行增加 和 16 行删除
@@ -19,4 +19,6 @@ @@ -19,4 +19,6 @@
19 19
20 *.tmp 20 *.tmp
21 21
22 -*.sum  
  22 +*.sum
  23 +
  24 +ability
1 [dev] 1 [dev]
2 #数据库相关 2 #数据库相关
3 -data_source = "root:123456@tcp(192.168.100.102:3306)/ability_display" 3 +data_source = "root:123456@tcp(192.168.100.102:3306)/ability_display?loc=Local"
4 #data_source = "root:sutianxia2015@tcp(115.29.205.99:3306)/ability_display" 4 #data_source = "root:sutianxia2015@tcp(115.29.205.99:3306)/ability_display"
5 5
6 6
1 [prod] 1 [prod]
2 #数据库相关 2 #数据库相关
3 -data_source = "root:123456@tcp(127.0.0.1:3306)/ability_display" 3 +data_source = "root:123456@tcp(127.0.0.1:3306)/ability_display?loc=Local"
4 4
5 5
6 #redis相关配置 6 #redis相关配置
@@ -5,6 +5,7 @@ import ( @@ -5,6 +5,7 @@ import (
5 "log" 5 "log"
6 "reflect" 6 "reflect"
7 "testing" 7 "testing"
  8 + "time"
8 9
9 "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/mybeego" 10 "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/mybeego"
10 "gitlab.fjmaimaimai.com/mmm-go/gocomm/common" 11 "gitlab.fjmaimaimai.com/mmm-go/gocomm/common"
@@ -42,3 +43,7 @@ func Benchmark_GenMessage(b *testing.B) { @@ -42,3 +43,7 @@ func Benchmark_GenMessage(b *testing.B) {
42 } 43 }
43 } 44 }
44 } 45 }
  46 +
  47 +func TestXXX(t *testing.T){
  48 + t.Log(time.Now())
  49 +}
  1 +package repository
  2 +
  3 +import "ability/models"
  4 +
  5 +type IConfigRepository interface {
  6 + GetCfgClient(clintId, clientSecret string) (v *models.CfgClient, err error)
  7 +}
  8 +
  9 +type ConfigRepository struct{}
  10 +
  11 +func (r *ConfigRepository) GetCfgClient(clintId, clientSecret string) (v *models.CfgClient, err error) {
  12 + return models.GetCfgClient(clintId, clientSecret)
  13 +}
  1 +package repository
  2 +
  3 +import "ability/models"
  4 +
  5 +type ConfigMockeRepository struct{}
  6 +
  7 +func (r *ConfigMockeRepository) GetCfgClient(clintId, clientSecret string) (v *models.CfgClient, err error) {
  8 + return
  9 +}
@@ -5,6 +5,9 @@ import "ability/models" @@ -5,6 +5,9 @@ import "ability/models"
5 type IUserRepository interface { 5 type IUserRepository interface {
6 GetUsersByMobile(mobile string) (v *models.Users, err error) 6 GetUsersByMobile(mobile string) (v *models.Users, err error)
7 GetUserInfoByMobile(mobile string) (v *models.UserInfo, err error) 7 GetUserInfoByMobile(mobile string) (v *models.UserInfo, err error)
  8 + UpdateUserInfoById(m *models.UserInfo) (err error)
  9 + GetUserInfoByAuthCode(authCode string) (v *models.UserInfo, err error)
  10 + GetUserInfoByRefreshToken(refreshToken string) (v *models.UserInfo, err error)
8 } 11 }
9 12
10 func assertImplement() { 13 func assertImplement() {
@@ -20,3 +23,15 @@ func (r *UserRepository) GetUsersByMobile(mobile string) (v *models.Users, err e @@ -20,3 +23,15 @@ func (r *UserRepository) GetUsersByMobile(mobile string) (v *models.Users, err e
20 func (r *UserRepository) GetUserInfoByMobile(mobile string) (v *models.UserInfo, err error) { 23 func (r *UserRepository) GetUserInfoByMobile(mobile string) (v *models.UserInfo, err error) {
21 return models.GetUserInfoByMobile(mobile) 24 return models.GetUserInfoByMobile(mobile)
22 } 25 }
  26 +
  27 +func (r *UserRepository) UpdateUserInfoById(m *models.UserInfo) (err error) {
  28 + return models.UpdateUserInfoById(m)
  29 +}
  30 +
  31 +func (r *UserRepository) GetUserInfoByAuthCode(authCode string) (v *models.UserInfo, err error) {
  32 + return models.GetUserInfoByAuthCode(authCode)
  33 +}
  34 +
  35 +func (r *UserRepository)GetUserInfoByRefreshToken(refreshToken string) (v *models.UserInfo, err error) {
  36 + return models.GetUserInfoByRefreshToken(refreshToken)
  37 +}
  1 +package repository
  2 +
  3 +import "ability/models"
  4 +
  5 +type UserMockRepository struct{}
  6 +
  7 +var userInfo =&models.UserInfo{
  8 + Auth:"897ca746f46b11e98771000c29ad8d6d",
  9 + AccessToken:"897ca746f46b11e98771000c29ad8d7d",
  10 + RefreshToken:"897ca746f46b11e98771000c29ad8d8d",
  11 +}
  12 +
  13 +func (r *UserMockRepository) GetUsersByMobile(mobile string) (v *models.Users, err error) {
  14 + v =&models.Users{
  15 + Username:"18065048389",
  16 + Password:"$2y$10$YWg7jPRVLBzc3kevokMkW.boswtCvhToqC.TappIwfqwJ.cI0efvy",
  17 + }
  18 + return
  19 +}
  20 +
  21 +func (r *UserMockRepository) GetUserInfoByMobile(mobile string) (v *models.UserInfo, err error) {
  22 + v =userInfo
  23 + return
  24 +}
  25 +
  26 +func (r *UserMockRepository) UpdateUserInfoById(m *models.UserInfo) (err error) {
  27 + return
  28 +}
  29 +
  30 +func (r *UserMockRepository) GetUserInfoByAuthCode(authCode string) (v *models.UserInfo, err error) {
  31 + v =userInfo
  32 + return
  33 +}
  34 +
  35 +func (r *UserMockRepository)GetUserInfoByRefreshToken(refreshToken string) (v *models.UserInfo, err error) {
  36 + v =userInfo
  37 + return
  38 +}
@@ -10,9 +10,11 @@ import ( @@ -10,9 +10,11 @@ import (
10 "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/log" 10 "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/log"
11 "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/orm" 11 "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/orm"
12 "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/redis" 12 "gitlab.fjmaimaimai.com/mmm-go/gocomm/pkg/redis"
  13 + "time"
13 ) 14 )
14 15
15 func init() { 16 func init() {
  17 + time.Local = time.FixedZone("CST", 3600*8)
16 log.InitLog(config.Logger{ 18 log.InitLog(config.Logger{
17 Filename: "app.log", 19 Filename: "app.log",
18 Level: "7", 20 Level: "7",
@@ -40,7 +40,7 @@ func assertImplement() { @@ -40,7 +40,7 @@ func assertImplement() {
40 var ( 40 var (
41 //服务 41 //服务
42 sms s_sms.ISmsService = &s_sms.YunPianSmsService{} 42 sms s_sms.ISmsService = &s_sms.YunPianSmsService{}
43 - 43 + ConfigRepository repository.IConfigRepository = &repository.ConfigRepository{}
44 //仓储 44 //仓储
45 UserRepository repository.IUserRepository = &repository.UserRepository{} 45 UserRepository repository.IUserRepository = &repository.UserRepository{}
46 ) 46 )
@@ -87,7 +87,9 @@ Success: @@ -87,7 +87,9 @@ Success:
87 if userInfo.Auth == "" { 87 if userInfo.Auth == "" {
88 userInfo.Auth = uid.NewV1().StringNoDash() 88 userInfo.Auth = uid.NewV1().StringNoDash()
89 } 89 }
90 - if err = models.UpdateUserInfoById(userInfo); err != nil { 90 + //TODO:更新
  91 + userInfo.AuthExp = time.Now().Add(time.Second*protocol.TokenExpire)
  92 + if err = UserRepository.UpdateUserInfoById(userInfo); err != nil {
91 return 93 return
92 } 94 }
93 rsp = &protocol.LoginResponse{AuthCode: userInfo.Auth} 95 rsp = &protocol.LoginResponse{AuthCode: userInfo.Auth}
@@ -106,19 +108,20 @@ func (s *AuthService) AccessToken(request *protocol.AccessTokenRequest) (rsp *pr @@ -106,19 +108,20 @@ func (s *AuthService) AccessToken(request *protocol.AccessTokenRequest) (rsp *pr
106 var ( 108 var (
107 userInfo *models.UserInfo 109 userInfo *models.UserInfo
108 ) 110 )
109 - _, err = models.GetCfgClient(request.ClientId, request.ClientSecret) 111 + _, err = ConfigRepository.GetCfgClient(request.ClientId, request.ClientSecret)
110 if err != nil { 112 if err != nil {
111 - common.NewError(101, err) 113 + err = common.NewError(101, err)
112 return 114 return
113 } 115 }
114 - userInfo, err = models.GetUserInfoByAuthCode(request.AuthCode) 116 + userInfo, err = UserRepository.GetUserInfoByAuthCode(request.AuthCode)
115 if err != nil { 117 if err != nil {
116 - common.NewError(4139, err) 118 + err = common.NewError(4139, err)
117 return 119 return
118 } 120 }
119 userInfo.AccessToken = uid.NewV1().StringNoDash() 121 userInfo.AccessToken = uid.NewV1().StringNoDash()
120 userInfo.RefreshToken = uid.NewV1().StringNoDash() 122 userInfo.RefreshToken = uid.NewV1().StringNoDash()
121 - if err = models.UpdateUserInfoById(userInfo); err != nil { 123 + if err = UserRepository.UpdateUserInfoById(userInfo); err != nil {
  124 + log.Error(err)
122 return 125 return
123 } 126 }
124 //valid token 127 //valid token
@@ -134,25 +137,24 @@ func (s *AuthService) AccessToken(request *protocol.AccessTokenRequest) (rsp *pr @@ -134,25 +137,24 @@ func (s *AuthService) AccessToken(request *protocol.AccessTokenRequest) (rsp *pr
134 func (s *AuthService) RefreshToken(request *protocol.RefreshTokenRequest) (rsp *protocol.RefreshTokenResponse, err error) { 137 func (s *AuthService) RefreshToken(request *protocol.RefreshTokenRequest) (rsp *protocol.RefreshTokenResponse, err error) {
135 var ( 138 var (
136 userInfo *models.UserInfo 139 userInfo *models.UserInfo
137 - newAccess *protocol.Access  
138 ) 140 )
139 - _, err = models.GetCfgClient(request.ClientId, request.ClientSecret) 141 + _, err = ConfigRepository.GetCfgClient(request.ClientId, request.ClientSecret)
140 if err != nil { 142 if err != nil {
141 common.NewError(101, err) 143 common.NewError(101, err)
142 return 144 return
143 } 145 }
144 - userInfo, err = models.GetUserInfoByAuthCode(request.RefreshToken) 146 + userInfo, err = UserRepository.GetUserInfoByRefreshToken(request.RefreshToken)
145 if err != nil { 147 if err != nil {
146 common.NewError(4139, err) 148 common.NewError(4139, err)
147 return 149 return
148 } 150 }
149 userInfo.AccessToken = uid.NewV1().StringNoDash() 151 userInfo.AccessToken = uid.NewV1().StringNoDash()
150 - if err = models.UpdateUserInfoById(userInfo); err != nil { 152 + if err = UserRepository.UpdateUserInfoById(userInfo); err != nil {
151 return 153 return
152 } 154 }
153 rsp = &protocol.RefreshTokenResponse{ 155 rsp = &protocol.RefreshTokenResponse{
154 - AccessToken: newAccess.AccessToken,  
155 - RefreshToken: newAccess.RefreshToken, 156 + AccessToken: userInfo.AccessToken,
  157 + RefreshToken: userInfo.RefreshToken,
156 ExpiresIn: protocol.TokenExpire, 158 ExpiresIn: protocol.TokenExpire,
157 } 159 }
158 return 160 return
1 package auth 1 package auth
2 2
3 import ( 3 import (
  4 + "ability/internal/repository"
4 "testing" 5 "testing"
5 6
6 "ability/protocol" 7 "ability/protocol"
@@ -9,8 +10,12 @@ import ( @@ -9,8 +10,12 @@ import (
9 10
10 func init() { 11 func init() {
11 tests.Init() 12 tests.Init()
  13 + UserRepository = &repository.UserMockRepository{}
  14 + ConfigRepository = &repository.ConfigMockeRepository{}
12 } 15 }
13 16
  17 +var auth =AuthService{}
  18 +
14 func Test_SmsCode(t *testing.T) { 19 func Test_SmsCode(t *testing.T) {
15 var ( 20 var (
16 resp *protocol.SmsCodeResponse 21 resp *protocol.SmsCodeResponse
@@ -35,3 +40,35 @@ func Test_SmsCode(t *testing.T) { @@ -35,3 +40,35 @@ func Test_SmsCode(t *testing.T) {
35 } 40 }
36 } 41 }
37 } 42 }
  43 +
  44 +func Test_Login(t *testing.T){
  45 + login:=&protocol.LoginRequest{
  46 + Phone:"18065048389",
  47 + Code:"562246",
  48 + GrantType:"signInPassword",
  49 + PassWord:"$2y$10$YWg7jPRVLBzc3kevokMkW.boswtCvhToqC.TappIwfqwJ.cI0efvy",
  50 + //ClientId:"123456",
  51 + }
  52 + loginRsp,err :=auth.Login(login)
  53 + if err!=nil{
  54 + t.Fatal(err,loginRsp)
  55 + }
  56 +}
  57 +
  58 +func Test_AccessToken(t *testing.T){
  59 + req :=&protocol.AccessTokenRequest{
  60 + AuthCode:"897ca746f46b11e98771000c29ad8d6d",
  61 + }
  62 + if rsp,err :=auth.AccessToken(req);err!=nil{
  63 + t.Fatal(err,rsp)
  64 + }
  65 +}
  66 +
  67 +func Test_RefreshToken(t *testing.T) {
  68 + req :=&protocol.RefreshTokenRequest{
  69 +
  70 + }
  71 + if rsp,err :=auth.RefreshToken(req);err!=nil{
  72 + t.Fatal(err,rsp)
  73 + }
  74 +}