blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-06 04:15:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

add test file

Browse Source
This commit is contained in:
Gordon 2022-07-29 15:47:09 +08:00
parent 64bfff5690
commit c43290338d
4 changed files with 31 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/api/third/fcm_update_token.go
View File

@ -39,7 +39,7 @@ func FcmUpdateToken(c *gin.Context) {
} }
log.NewInfo(req.OperationID, utils.GetSelfFuncName(), req, UserId) log.NewInfo(req.OperationID, utils.GetSelfFuncName(), req, UserId)
//逻辑处理开始 //逻辑处理开始
err := db.DB.SetFcmToken(UserId, int(req.Platform), req.FcmToken, 0) err := db.DB.SetFcmToken(UserId, req.Platform, req.FcmToken, 0)
if err != nil { if err != nil {
errMsg := req.OperationID + " " + "SetFcmToken failed " + err.Error() + " token:" + c.Request.Header.Get("token") errMsg := req.OperationID + " " + "SetFcmToken failed " + err.Error() + " token:" + c.Request.Header.Get("token")
log.NewError(req.OperationID, errMsg) log.NewError(req.OperationID, errMsg)

1
internal/push/fcm/push.go
View File

@ -50,6 +50,7 @@ func (f *Fcm) Push(accounts []string, alert, detailContent, operationID string,
for _, account := range accounts { for _, account := range accounts {
IosfcmToken, IosErr := db.DB.GetFcmToken(account, 1) IosfcmToken, IosErr := db.DB.GetFcmToken(account, 1)
AndroidfcmToken, AndroidErr := db.DB.GetFcmToken(account, 2) AndroidfcmToken, AndroidErr := db.DB.GetFcmToken(account, 2)
if IosErr == nil { if IosErr == nil {
Tokens = append(Tokens, IosfcmToken) Tokens = append(Tokens, IosfcmToken)
} }

15
internal/push/fcm/push_test.go Normal file
View File

@ -0,0 +1,15 @@
package fcm
import (
"Open_IM/internal/push"
"fmt"
"github.com/stretchr/testify/assert"
"testing"
)
func Test_Push(t *testing.T) {
offlinePusher := NewFcm()
resp, err := offlinePusher.Push([]string{"test_uid"}, "哈哈", "嘿嘿", "12321", push.PushOpts{})
assert.Nil(t, err)
fmt.Println(resp)
}

20
pkg/common/db/redisModel_test.go
View File

@ -35,7 +35,7 @@ func Test_GetKeyTTL(t *testing.T) {
ctx := context.Background() ctx := context.Background()
key := flag.String("key", "key", "key value") key := flag.String("key", "key", "key value")
flag.Parse() flag.Parse()
ttl, err := DB.rdb.TTL(ctx, *key).Result() ttl, err := DB.RDB.TTL(ctx, *key).Result()
assert.Nil(t, err) assert.Nil(t, err)
fmt.Println(ttl) fmt.Println(ttl)
} }
@ -43,7 +43,7 @@ func Test_HGetAll(t *testing.T) {
ctx := context.Background() ctx := context.Background()
key := flag.String("key", "key", "key value") key := flag.String("key", "key", "key value")
flag.Parse() flag.Parse()
ttl, err := DB.rdb.TTL(ctx, *key).Result() ttl, err := DB.RDB.TTL(ctx, *key).Result()
assert.Nil(t, err) assert.Nil(t, err)
fmt.Println(ttl) fmt.Println(ttl)
} }
@ -119,9 +119,17 @@ func Test_GetAccountCode(t *testing.T) {
assert.Nil(t, err) assert.Nil(t, err)
fmt.Println(code) fmt.Println(code)
} }
func Test_GetGroupMemberList(t *testing.T) { func Test_SetFcmToken(t *testing.T) {
groupID := "3791742301" uid := "test_uid"
list, err := DB.GetGroupMemberIDListFromCache(groupID) token := "dfnWBtOjSj-XIZnUvDlegv:APA91bG09XTtiXfpE6U7gUVMOhnKcUkNCv4WHn0UZr2clUi-tS1jEH-HiCEW8GIAhjLIGcfUJ6NIKteC023ZxDH7J0PJ5sTxoup3fHDUPLU7KgQoZS4tPyFqCbZ6bRB7esDPEnD1n_s0"
platformID := 2
err := DB.SetFcmToken(uid, platformID, token, 0)
assert.Nil(t, err) assert.Nil(t, err)
fmt.Println(list)
} }
//func Test_GetGroupMemberList(t *testing.T) {
// groupID := "3791742301"
// list, err := DB.GetGroupMemberIDListFromCache(groupID)
// assert.Nil(t, err)
// fmt.Println(list)
//}