305 lines
10 KiB
Go
305 lines
10 KiB
Go
package tencentim
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"os"
|
||
"strconv"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func TestRealTencentIMCreateSendAndDestroyGroup(t *testing.T) {
|
||
if os.Getenv("TENCENT_IM_REAL_TEST") != "1" {
|
||
t.Skip("set TENCENT_IM_REAL_TEST=1 and Tencent IM env vars to run the real REST smoke test")
|
||
}
|
||
|
||
cfg := realRESTConfigFromEnv(t)
|
||
client, err := NewRESTClient(cfg)
|
||
if err != nil {
|
||
t.Fatalf("NewRESTClient failed: %v", err)
|
||
}
|
||
|
||
now := time.Now().UTC()
|
||
nowMS := now.UnixMilli()
|
||
groupID := fmt.Sprintf("hy_codex_bc_r_%d", nowMS)
|
||
eventID := fmt.Sprintf("codex_real_im_%d", nowMS)
|
||
ctx, cancel := context.WithTimeout(context.Background(), 45*time.Second)
|
||
defer cancel()
|
||
|
||
// 真实 IM 测试必须使用临时 GroupID，避免污染业务全局/区域播报群。
|
||
if err := client.EnsureGroup(ctx, GroupSpec{
|
||
GroupID: groupID,
|
||
Name: "codex real im smoke",
|
||
Type: DefaultGroupType,
|
||
ApplyJoinOption: "FreeAccess",
|
||
}); err != nil {
|
||
t.Fatalf("real Tencent IM create group failed: %v", err)
|
||
}
|
||
t.Cleanup(func() {
|
||
cleanupCtx, cleanupCancel := context.WithTimeout(context.Background(), 20*time.Second)
|
||
defer cleanupCancel()
|
||
if err := client.DestroyGroup(cleanupCtx, groupID); err != nil {
|
||
t.Errorf("real Tencent IM destroy group cleanup failed: %v", err)
|
||
}
|
||
})
|
||
memberUserID := nowMS % 9_000_000_000
|
||
if err := realImportAccount(ctx, client, memberUserID); err != nil {
|
||
t.Fatalf("real Tencent IM account import failed: %v", err)
|
||
}
|
||
if err := realAddGroupMember(ctx, client, groupID, memberUserID); err != nil {
|
||
t.Fatalf("real Tencent IM add group member failed: %v", err)
|
||
}
|
||
|
||
const actorUserID = "323411852102995968"
|
||
const targetUserID = "323411852102995969"
|
||
c2cEventID := fmt.Sprintf("codex_real_im_c2c_%d", nowMS)
|
||
c2cPayload, err := json.Marshal(map[string]any{
|
||
"event_id": c2cEventID,
|
||
"event_type": "room_user_kicked",
|
||
"actor_user_id": actorUserID,
|
||
"target_user_id": targetUserID,
|
||
"sender_user_id": actorUserID,
|
||
"receiver_user_id": targetUserID,
|
||
"sent_at_ms": nowMS,
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("marshal c2c smoke payload failed: %v", err)
|
||
}
|
||
if err := client.PublishUserCustomMessage(ctx, CustomUserMessage{
|
||
ToAccount: strconv.FormatInt(memberUserID, 10),
|
||
EventID: c2cEventID,
|
||
Desc: "codex_real_im_c2c_probe",
|
||
Ext: "im_c2c_smoke",
|
||
PayloadJSON: c2cPayload,
|
||
}); err != nil {
|
||
t.Fatalf("real Tencent IM send C2C custom message failed: %v", err)
|
||
}
|
||
// SyncOtherMachine=2 不同步管理员发送端，因此漫游消息必须从实际接收用户视角回读。
|
||
c2cData, c2cCloudData := realWaitForC2CPayload(t, ctx, client, strconv.FormatInt(memberUserID, 10), cfg.AdminIdentifier, c2cEventID, now.Unix())
|
||
realAssertStringUserIDs(t, c2cData, map[string]string{
|
||
"actor_user_id": actorUserID,
|
||
"target_user_id": targetUserID,
|
||
"sender_user_id": actorUserID,
|
||
"receiver_user_id": targetUserID,
|
||
})
|
||
realAssertStringUserIDs(t, c2cCloudData, map[string]string{
|
||
"actor_user_id": actorUserID,
|
||
"target_user_id": targetUserID,
|
||
"sender_user_id": actorUserID,
|
||
"receiver_user_id": targetUserID,
|
||
})
|
||
|
||
roomEvent := RoomEvent{
|
||
GroupID: groupID,
|
||
EventID: eventID,
|
||
RoomID: "codex-room-smoke",
|
||
EventType: "room_mic_change",
|
||
ActorUserID: actorUserID,
|
||
TargetUserID: targetUserID,
|
||
SeatNo: 3,
|
||
RoomVersion: 22,
|
||
Attributes: map[string]string{
|
||
"action": "change",
|
||
"from_seat": "1",
|
||
"to_seat": "3",
|
||
"mic_session_id": "codex-real-im-mic-session",
|
||
"target_user_ids": `["323411852102995968","323411852102995969"]`,
|
||
},
|
||
}
|
||
|
||
// 使用生产 RoomEvent 序列化和发送入口，避免真实测试绕开本次修复的代码路径。
|
||
if err := client.PublishRoomEvent(ctx, roomEvent); err != nil {
|
||
t.Fatalf("real Tencent IM send group custom message failed: %v", err)
|
||
}
|
||
groupData, groupCloudData := realWaitForGroupPayload(t, ctx, client, groupID, eventID)
|
||
for source, payload := range map[string]string{"Data": groupData, "CloudCustomData": groupCloudData} {
|
||
realAssertStringUserIDs(t, payload, map[string]string{
|
||
"actor_user_id": actorUserID,
|
||
"target_user_id": targetUserID,
|
||
})
|
||
var decoded map[string]any
|
||
if err := json.Unmarshal([]byte(payload), &decoded); err != nil {
|
||
t.Fatalf("decode group %s payload failed: %v", source, err)
|
||
}
|
||
attributes, ok := decoded["attributes"].(map[string]any)
|
||
if !ok {
|
||
t.Fatalf("group %s payload attributes type = %T, want object", source, decoded["attributes"])
|
||
}
|
||
var targetUserIDs []any
|
||
if err := json.Unmarshal([]byte(fmt.Sprint(attributes["target_user_ids"])), &targetUserIDs); err != nil {
|
||
t.Fatalf("decode group %s target_user_ids failed: %v", source, err)
|
||
}
|
||
if len(targetUserIDs) != 2 || targetUserIDs[0] != actorUserID || targetUserIDs[1] != targetUserID {
|
||
t.Fatalf("group %s target_user_ids = %#v, want exact string IDs", source, targetUserIDs)
|
||
}
|
||
}
|
||
if err := client.DeleteGroupMember(ctx, groupID, memberUserID); err != nil {
|
||
t.Fatalf("real Tencent IM delete group member failed: %v", err)
|
||
}
|
||
|
||
t.Logf("real Tencent IM transport and history readback passed: sdk_app_id=%d group_id=%s group_event_id=%s c2c_event_id=%s exact_actor_user_id=%s removed_user_id=%d", cfg.SDKAppID, groupID, eventID, c2cEventID, actorUserID, memberUserID)
|
||
}
|
||
|
||
type realHistoryMessage struct {
|
||
MsgBody []messageElement `json:"MsgBody"`
|
||
CloudCustomData string `json:"CloudCustomData"`
|
||
}
|
||
|
||
type realGroupHistoryResponse struct {
|
||
restResponse
|
||
RspMsgList []realHistoryMessage `json:"RspMsgList"`
|
||
}
|
||
|
||
type realC2CHistoryResponse struct {
|
||
restResponse
|
||
MsgList []realHistoryMessage `json:"MsgList"`
|
||
}
|
||
|
||
func realWaitForGroupPayload(t *testing.T, ctx context.Context, client *RESTClient, groupID, eventID string) (string, string) {
|
||
t.Helper()
|
||
for attempt := 0; attempt < 10; attempt++ {
|
||
var response realGroupHistoryResponse
|
||
if err := client.post(ctx, "v4/group_open_http_svc/group_msg_get_simple", map[string]any{
|
||
"GroupId": groupID,
|
||
"ReqMsgNumber": 20,
|
||
}, &response); err != nil {
|
||
t.Fatalf("real Tencent IM group history request failed: %v", err)
|
||
}
|
||
if err := response.err(); err != nil {
|
||
t.Fatalf("real Tencent IM group history failed: %v", err)
|
||
}
|
||
if data, cloudData, ok := realFindHistoryPayload(response.RspMsgList, eventID); ok {
|
||
return data, cloudData
|
||
}
|
||
select {
|
||
case <-ctx.Done():
|
||
t.Fatalf("real Tencent IM group history readback timed out: %v", ctx.Err())
|
||
case <-time.After(500 * time.Millisecond):
|
||
}
|
||
}
|
||
t.Fatalf("real Tencent IM group history did not contain event_id=%s", eventID)
|
||
return "", ""
|
||
}
|
||
|
||
func realWaitForC2CPayload(t *testing.T, ctx context.Context, client *RESTClient, operatorAccount, peerAccount, eventID string, sentAtSeconds int64) (string, string) {
|
||
t.Helper()
|
||
for attempt := 0; attempt < 10; attempt++ {
|
||
var response realC2CHistoryResponse
|
||
if err := client.post(ctx, "v4/openim/admin_getroammsg", map[string]any{
|
||
"Operator_Account": operatorAccount,
|
||
"Peer_Account": peerAccount,
|
||
"MaxCnt": 100,
|
||
"MinTime": sentAtSeconds - 60,
|
||
"MaxTime": time.Now().UTC().Unix() + 60,
|
||
}, &response); err != nil {
|
||
t.Fatalf("real Tencent IM C2C history request failed: %v", err)
|
||
}
|
||
if err := response.err(); err != nil {
|
||
t.Fatalf("real Tencent IM C2C history failed: %v", err)
|
||
}
|
||
if data, cloudData, ok := realFindHistoryPayload(response.MsgList, eventID); ok {
|
||
return data, cloudData
|
||
}
|
||
select {
|
||
case <-ctx.Done():
|
||
t.Fatalf("real Tencent IM C2C history readback timed out: %v", ctx.Err())
|
||
case <-time.After(500 * time.Millisecond):
|
||
}
|
||
}
|
||
t.Fatalf("real Tencent IM C2C history did not contain event_id=%s", eventID)
|
||
return "", ""
|
||
}
|
||
|
||
func realFindHistoryPayload(messages []realHistoryMessage, eventID string) (string, string, bool) {
|
||
for _, message := range messages {
|
||
for _, element := range message.MsgBody {
|
||
if element.MsgType != "TIMCustomElem" || !strings.Contains(element.MsgContent.Data, eventID) {
|
||
continue
|
||
}
|
||
return element.MsgContent.Data, message.CloudCustomData, true
|
||
}
|
||
}
|
||
return "", "", false
|
||
}
|
||
|
||
func realAssertStringUserIDs(t *testing.T, payload string, expected map[string]string) {
|
||
t.Helper()
|
||
var decoded map[string]any
|
||
if err := json.Unmarshal([]byte(payload), &decoded); err != nil {
|
||
t.Fatalf("decode Tencent IM payload failed: %v", err)
|
||
}
|
||
for key, want := range expected {
|
||
got, ok := decoded[key].(string)
|
||
if !ok {
|
||
t.Fatalf("Tencent IM payload %s type = %T, want string", key, decoded[key])
|
||
}
|
||
if got != want {
|
||
t.Fatalf("Tencent IM payload %s = %q, want %q", key, got, want)
|
||
}
|
||
}
|
||
}
|
||
|
||
func realImportAccount(ctx context.Context, client *RESTClient, userID int64) error {
|
||
return client.ImportAccount(ctx, AccountProfile{
|
||
UserID: userID,
|
||
Nick: "codex-im-smoke",
|
||
})
|
||
}
|
||
|
||
func realAddGroupMember(ctx context.Context, client *RESTClient, groupID string, userID int64) error {
|
||
var response restResponse
|
||
if err := client.post(ctx, "v4/group_open_http_svc/add_group_member", map[string]any{
|
||
"GroupId": groupID,
|
||
"Silence": 1,
|
||
"MemberList": []map[string]any{
|
||
{"Member_Account": strconv.FormatInt(userID, 10)},
|
||
},
|
||
}, &response); err != nil {
|
||
return err
|
||
}
|
||
return response.err()
|
||
}
|
||
|
||
func realRESTConfigFromEnv(t *testing.T) RESTConfig {
|
||
t.Helper()
|
||
|
||
sdkAppID, err := strconv.ParseInt(requiredEnv(t, "TENCENT_IM_SDK_APP_ID"), 10, 64)
|
||
if err != nil || sdkAppID <= 0 {
|
||
t.Fatalf("TENCENT_IM_SDK_APP_ID is invalid")
|
||
}
|
||
endpoint := strings.TrimSpace(os.Getenv("TENCENT_IM_ENDPOINT"))
|
||
if endpoint == "" {
|
||
endpoint = DefaultEndpoint
|
||
}
|
||
adminUserSigTTL := time.Hour
|
||
if rawTTL := strings.TrimSpace(os.Getenv("TENCENT_IM_ADMIN_USERSIG_TTL")); rawTTL != "" {
|
||
parsed, err := time.ParseDuration(rawTTL)
|
||
if err != nil || parsed <= 0 {
|
||
t.Fatalf("TENCENT_IM_ADMIN_USERSIG_TTL is invalid")
|
||
}
|
||
adminUserSigTTL = parsed
|
||
}
|
||
|
||
return RESTConfig{
|
||
SDKAppID: sdkAppID,
|
||
SecretKey: requiredEnv(t, "TENCENT_IM_SECRET_KEY"),
|
||
AdminIdentifier: requiredEnv(t, "TENCENT_IM_ADMIN_IDENTIFIER"),
|
||
AdminUserSigTTL: adminUserSigTTL,
|
||
Endpoint: endpoint,
|
||
GroupType: DefaultGroupType,
|
||
}
|
||
}
|
||
|
||
func requiredEnv(t *testing.T, key string) string {
|
||
t.Helper()
|
||
|
||
value := strings.TrimSpace(os.Getenv(key))
|
||
if value == "" {
|
||
t.Fatalf("%s is required for real Tencent IM smoke test", key)
|
||
}
|
||
return value
|
||
}
|