IM/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2026-05-13 21:45:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

msg database

Browse Source
This commit is contained in:
withchao
2023-02-23 18:48:14 +08:00
parent 2f1c064413
commit c8004279a4
5 changed files with 159 additions and 180 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/common/db/cache/redis_test.go
Vendored
+104 -104
View File
@@ -1,107 +1,107 @@
package cache
import (
"Open_IM/pkg/common/constant"
pbChat "Open_IM/pkg/proto/msg"
common "Open_IM/pkg/proto/sdkws"
"context"
"flag"
"fmt"
"github.com/stretchr/testify/assert"
"testing"
)
var DB RedisClient
func Test_SetTokenMapByUidPid(t *testing.T) {
m := make(map[string]int, 0)
m["test1"] = 1
m["test2"] = 2
m["2332"] = 4
err := DB.SetTokenMapByUidPid("1234", 2, m)
assert.Nil(t, err)
}
func Test_GetTokenMapByUidPid(t *testing.T) {
m, err := DB.GetTokenMapByUidPid("1234", "Android")
assert.Nil(t, err)
fmt.Println(m)
}
//func TestDataBases_GetMultiConversationMsgOpt(t *testing.T) {
// m, err := DB.GetMultiConversationMsgOpt("fg", []string{"user", "age", "color"})
// assert.Nil(t, err)
// fmt.Println(m)
//import (
// "Open_IM/pkg/common/constant"
// pbChat "Open_IM/pkg/proto/msg"
// common "Open_IM/pkg/proto/sdkws"
// "context"
// "flag"
// "fmt"
// "github.com/stretchr/testify/assert"
// "testing"
//)
//
//var DB RedisClient
//
//func Test_SetTokenMapByUidPid(t *testing.T) {
// //m := make(map[string]int, 0)
// //m["test1"] = 1
// //m["test2"] = 2
// //m["2332"] = 4
// //err := DB.SetTokenMapByUidPid("1234", 2, m)
// //assert.Nil(t, err)
//
//}
//func Test_GetTokenMapByUidPid(t *testing.T) {
// //m, err := DB.GetTokenMapByUidPid("1234", "Android")
// //assert.Nil(t, err)
// //fmt.Println(m)
//}
//
////func TestDataBases_GetMultiConversationMsgOpt(t *testing.T) {
//// m, err := DB.GetMultiConversationMsgOpt("fg", []string{"user", "age", "color"})
//// assert.Nil(t, err)
//// fmt.Println(m)
////}
//func Test_GetKeyTTL(t *testing.T) {
// ctx := context.Background()
// key := flag.String("key", "key", "key value")
// flag.Parse()
// ttl, err := DB.GetClient().TTL(ctx, *key).Result()
// assert.Nil(t, err)
// fmt.Println(ttl)
//}
//func Test_HGetAll(t *testing.T) {
// ctx := context.Background()
// key := flag.String("key", "key", "key value")
// flag.Parse()
// ttl, err := DB.GetClient().TTL(ctx, *key).Result()
// assert.Nil(t, err)
// fmt.Println(ttl)
//}
//
//func Test_NewSetMessageToCache(t *testing.T) {
// var msg pbChat.MsgDataToMQ
// m := make(map[string]bool)
// var offlinePush common.OfflinePushInfo
// offlinePush.Title = "3"
// offlinePush.Ex = "34"
// offlinePush.IOSPushSound = "+1"
// offlinePush.IOSBadgeCount = true
// m[constant.IsPersistent] = true
// m[constant.IsHistory] = true
// var data common.MsgData
// uid := "test_uid"
// data.Seq = 11
// data.ClientMsgID = "23jwhjsdf"
// data.SendID = "111"
// data.RecvID = "222"
// data.Content = []byte{1, 2, 3, 4, 5, 6, 7}
// data.Seq = 1212
// data.Options = m
// data.OfflinePushInfo = &offlinePush
// data.AtUserIDList = []string{"1212", "23232"}
// msg.MsgData = &data
// messageList := []*pbChat.MsgDataToMQ{&msg}
// err, _ := DB.SetMessageToCache(messageList, uid, "cacheTest")
// assert.Nil(t, err)
//
//}
//func Test_NewGetMessageListBySeq(t *testing.T) {
// var msg pbChat.MsgDataToMQ
// var data common.MsgData
// uid := "test_uid"
// data.Seq = 11
// data.ClientMsgID = "23jwhjsdf"
// msg.MsgData = &data
//
// seqMsg, failedSeqList, err := DB.GetMessageListBySeq(uid, []uint32{1212}, "cacheTest")
// assert.Nil(t, err)
// fmt.Println(seqMsg, failedSeqList)
//
//}
//
//func Test_SetFcmToken(t *testing.T) {
// uid := "test_uid"
// token := "dfnWBtOjSj-XIZnUvDlegv:APA91bG09XTtiXfpE6U7gUVMOhnKcUkNCv4WHn0UZr2clUi-tS1jEH-HiCEW8GIAhjLIGcfUJ6NIKteC023ZxDH7J0PJ5sTxoup3fHDUPLU7KgQoZS4tPyFqCbZ6bRB7esDPEnD1n_s0"
// platformID := 2
// err := DB.SetFcmToken(uid, platformID, token, 0)
// assert.Nil(t, err)
//}
//func Test_GetFcmToken(t *testing.T) {
// uid := "test_uid"
// platformID := 2
// token, err := DB.GetFcmToken(uid, platformID)
// assert.Nil(t, err)
// fmt.Println("token is :", token)
//}
func Test_GetKeyTTL(t *testing.T) {
ctx := context.Background()
key := flag.String("key", "key", "key value")
flag.Parse()
ttl, err := DB.GetClient().TTL(ctx, *key).Result()
assert.Nil(t, err)
fmt.Println(ttl)
}
func Test_HGetAll(t *testing.T) {
ctx := context.Background()
key := flag.String("key", "key", "key value")
flag.Parse()
ttl, err := DB.GetClient().TTL(ctx, *key).Result()
assert.Nil(t, err)
fmt.Println(ttl)
}
func Test_NewSetMessageToCache(t *testing.T) {
var msg pbChat.MsgDataToMQ
m := make(map[string]bool)
var offlinePush common.OfflinePushInfo
offlinePush.Title = "3"
offlinePush.Ex = "34"
offlinePush.IOSPushSound = "+1"
offlinePush.IOSBadgeCount = true
m[constant.IsPersistent] = true
m[constant.IsHistory] = true
var data common.MsgData
uid := "test_uid"
data.Seq = 11
data.ClientMsgID = "23jwhjsdf"
data.SendID = "111"
data.RecvID = "222"
data.Content = []byte{1, 2, 3, 4, 5, 6, 7}
data.Seq = 1212
data.Options = m
data.OfflinePushInfo = &offlinePush
data.AtUserIDList = []string{"1212", "23232"}
msg.MsgData = &data
messageList := []*pbChat.MsgDataToMQ{&msg}
err, _ := DB.SetMessageToCache(messageList, uid, "cacheTest")
assert.Nil(t, err)
}
func Test_NewGetMessageListBySeq(t *testing.T) {
var msg pbChat.MsgDataToMQ
var data common.MsgData
uid := "test_uid"
data.Seq = 11
data.ClientMsgID = "23jwhjsdf"
msg.MsgData = &data
seqMsg, failedSeqList, err := DB.GetMessageListBySeq(uid, []uint32{1212}, "cacheTest")
assert.Nil(t, err)
fmt.Println(seqMsg, failedSeqList)
}
func Test_SetFcmToken(t *testing.T) {
uid := "test_uid"
token := "dfnWBtOjSj-XIZnUvDlegv:APA91bG09XTtiXfpE6U7gUVMOhnKcUkNCv4WHn0UZr2clUi-tS1jEH-HiCEW8GIAhjLIGcfUJ6NIKteC023ZxDH7J0PJ5sTxoup3fHDUPLU7KgQoZS4tPyFqCbZ6bRB7esDPEnD1n_s0"
platformID := 2
err := DB.SetFcmToken(uid, platformID, token, 0)
assert.Nil(t, err)
}
func Test_GetFcmToken(t *testing.T) {
uid := "test_uid"
platformID := 2
token, err := DB.GetFcmToken(uid, platformID)
assert.Nil(t, err)
fmt.Println("token is :", token)
}