hyapp-server/pkg/tencentrtc/rest_client_test.go

package tencentrtc

import (
	"bytes"
	"context"
	"encoding/json"
	"io"
	"net/http"
	"strings"
	"testing"
	"time"
)

// TestRESTClientRemoveUserByStrRoomIDBuildsTencentCloudRequest 锁定 TRTC 字符串房间踢人的 API 3.0 请求形态。
func TestRESTClientRemoveUserByStrRoomIDBuildsTencentCloudRequest(t *testing.T) {
	var capturedBody map[string]any
	var capturedHeader http.Header
	client, err := NewRESTClient(RESTConfig{
		SDKAppID:  1400000001,
		SecretID:  "AKIDEXAMPLE",
		SecretKey: "secret",
		Region:    "ap-guangzhou",
		HTTPClient: &http.Client{Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) {
			capturedHeader = request.Header.Clone()
			if err := json.NewDecoder(request.Body).Decode(&capturedBody); err != nil {
				t.Fatalf("decode request body failed: %v", err)
			}
			return &http.Response{
				StatusCode: http.StatusOK,
				Body:       io.NopCloser(bytes.NewReader([]byte(`{"Response":{"RequestId":"req-1"}}`))),
				Header:     make(http.Header),
			}, nil
		})},
		Now: func() time.Time { return time.Unix(1_700_000_000, 0).UTC() },
	})
	if err != nil {
		t.Fatalf("NewRESTClient failed: %v", err)
	}

	if err := client.RemoveUserByStrRoomID(context.Background(), "room_1001", 10002); err != nil {
		t.Fatalf("RemoveUserByStrRoomID failed: %v", err)
	}

	if capturedBody["SdkAppId"].(float64) != 1400000001 || capturedBody["RoomId"] != "room_1001" {
		t.Fatalf("unexpected remove user body: %+v", capturedBody)
	}
	users, ok := capturedBody["UserIds"].([]any)
	if !ok || len(users) != 1 || users[0] != "10002" {
		t.Fatalf("unexpected user ids: %+v", capturedBody)
	}
	if capturedHeader.Get("X-TC-Action") != removeUserByStrRoomAction || capturedHeader.Get("X-TC-Version") != trtcAPIVersion {
		t.Fatalf("unexpected tencent cloud headers: %+v", capturedHeader)
	}
	if !strings.HasPrefix(capturedHeader.Get("Authorization"), tencentCloudTC3Algorithm+" Credential=AKIDEXAMPLE/") {
		t.Fatalf("authorization header missing TC3 credential: %s", capturedHeader.Get("Authorization"))
	}
}

// TestRESTClientSetUserAudioBlockedByStrRoomIDBuildsAudioOnlyRequest 锁定服务端禁言只关闭上行音频，
// 解除时恢复音视频能力，避免误用 IsMute=1 把未来的视频房能力一并关闭。
func TestRESTClientSetUserAudioBlockedByStrRoomIDBuildsAudioOnlyRequest(t *testing.T) {
	var bodies []map[string]any
	var actions []string
	client, err := NewRESTClient(RESTConfig{
		SDKAppID:  1400000001,
		SecretID:  "AKIDEXAMPLE",
		SecretKey: "secret",
		Region:    "ap-singapore",
		HTTPClient: &http.Client{Transport: roundTripFunc(func(request *http.Request) (*http.Response, error) {
			var body map[string]any
			if err := json.NewDecoder(request.Body).Decode(&body); err != nil {
				t.Fatalf("decode request body failed: %v", err)
			}
			bodies = append(bodies, body)
			actions = append(actions, request.Header.Get("X-TC-Action"))
			return &http.Response{
				StatusCode: http.StatusOK,
				Body:       io.NopCloser(bytes.NewReader([]byte(`{"Response":{"RequestId":"req-block"}}`))),
				Header:     make(http.Header),
			}, nil
		})},
		Now: func() time.Time { return time.Unix(1_700_000_000, 0).UTC() },
	})
	if err != nil {
		t.Fatalf("NewRESTClient failed: %v", err)
	}

	if err := client.SetUserAudioBlockedByStrRoomID(context.Background(), "room_1001", 10002, true); err != nil {
		t.Fatalf("block user audio failed: %v", err)
	}
	if err := client.SetUserAudioBlockedByStrRoomID(context.Background(), "room_1001", 10002, false); err != nil {
		t.Fatalf("unblock user audio failed: %v", err)
	}

	if len(bodies) != 2 || len(actions) != 2 {
		t.Fatalf("unexpected request count: bodies=%d actions=%d", len(bodies), len(actions))
	}
	for index, expectedMute := range []float64{trtcUserAudioDisabled, trtcUserMediaEnabled} {
		if actions[index] != setUserBlockedByStrAction || bodies[index]["SdkAppId"] != float64(1400000001) || bodies[index]["StrRoomId"] != "room_1001" || bodies[index]["UserId"] != "10002" || bodies[index]["IsMute"] != expectedMute {
			t.Fatalf("unexpected set user blocked request %d: action=%s body=%+v", index, actions[index], bodies[index])
		}
	}
}

// TestRESTClientSetUserAudioBlockedTreatsAbsentRTCUserAsConverged 防止用户先退 RTC 后的历史
// RoomUserMuted 事件把 durable outbox 永久卡在无意义重试上。
func TestRESTClientSetUserAudioBlockedTreatsAbsentRTCUserAsConverged(t *testing.T) {
	client, err := NewRESTClient(RESTConfig{
		SDKAppID:  1400000001,
		SecretID:  "AKIDEXAMPLE",
		SecretKey: "secret",
		Region:    "ap-singapore",
		HTTPClient: &http.Client{Transport: roundTripFunc(func(*http.Request) (*http.Response, error) {
			return &http.Response{
				StatusCode: http.StatusOK,
				Body:       io.NopCloser(bytes.NewReader([]byte(`{"Response":{"Error":{"Code":"FailedOperation.UserNotExist","Message":"user absent"},"RequestId":"req-absent"}}`))),
				Header:     make(http.Header),
			}, nil
		})},
	})
	if err != nil {
		t.Fatalf("NewRESTClient failed: %v", err)
	}
	if err := client.SetUserAudioBlockedByStrRoomID(context.Background(), "room_1001", 10002, true); err != nil {
		t.Fatalf("absent RTC user must be treated as converged: %v", err)
	}
}

type roundTripFunc func(*http.Request) (*http.Response, error)

func (f roundTripFunc) RoundTrip(request *http.Request) (*http.Response, error) {
	return f(request)
}