ModelRouter/internal/core/channels_lobster_claim_test.go

package core

import (
	"encoding/json"
	"fmt"
	"net/http"
	"net/http/httptest"
	"testing"

	"llmsproxy/internal/lua"
)

// 有道每日活动领取是**三步流程**，参考注释点名：无可用活动是**正常业务状态**，
// 不能显示成"领取失败"（否则用户会去查网络/账号，方向全错）。
// 本判据钉住决策表的四类结果：inactive / already-claimed / success / failed。
func TestLobsterActivityClaimDecisionTable(t *testing.T) {
	type c struct {
		name       string
		slot       string
		ctx        string
		wantKind   string
		wantCalled bool // 是否应发出 claim 请求
	}
	cases := []c{
		{"槽位不可用→inactive", `{"code":0,"data":{"slotState":"unavailable","activity":{"activityCode":"A1"}}}`, "", "inactive", false},
		{"无 activityCode→inactive", `{"code":0,"data":{"slotState":"available","activity":{}}}`, "", "inactive", false},
		{"今天已领→already-claimed", `{"code":0,"data":{"slotState":"available","activity":{"activityCode":"A1","configRevision":7}}}`,
			`{"code":0,"data":{"state":{"claimedToday":true},"actions":["check_in"]}}`, "already-claimed", false},
		{"无 check_in 动作→inactive", `{"code":0,"data":{"slotState":"available","activity":{"activityCode":"A1","configRevision":7}}}`,
			`{"code":0,"data":{"state":{"claimedToday":false},"actions":["share"]}}`, "inactive", false},
		{"可领→success", `{"code":0,"data":{"slotState":"available","activity":{"activityCode":"A1","configRevision":7}}}`,
			`{"code":0,"data":{"state":{"claimedToday":false},"actions":["check_in"]}}`, "success", true},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			claimHit := 0
			var claimBody string
			ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				w.Header().Set("Content-Type", "application/json")
				switch {
				case r.URL.Path == "/api/client-activities/slot":
					fmt.Fprint(w, tc.slot)
				case r.URL.Path == "/api/client-activities/A1/context":
					if tc.ctx == "" {
						w.WriteHeader(500)
						return
					}
					fmt.Fprint(w, tc.ctx)
				case r.URL.Path == "/api/client-activities/A1/actions/check_in":
					claimHit++
					b := make([]byte, r.ContentLength)
					_, _ = r.Body.Read(b)
					claimBody = string(b)
					fmt.Fprint(w, `{"code":0,"data":{"result":{"creditsGranted":30}}}`)
				default:
					w.WriteHeader(404)
				}
			}))
			defer ts.Close()

			c := kernelTestCore(t)
			ps := c.Plugins()
			src, err := readBundledChannels(t)
			if err != nil {
				t.Fatal(err)
			}
			if err := ps.LoadSource("channels", src); err != nil {
				t.Fatal(err)
			}
			// 把账号基址指到假上游
			if _, err := ps.Call("channels", "set_endpoints", map[string]interface{}{
				"platform":              "lobsterai",
				"lobster_activity_base": map[string]interface{}{"url": ts.URL},
			}); err != nil {
				t.Fatal(err)
			}
			// 存入凭据（claim 需要结构化 credential 对象）
			cred, _ := json.Marshal(map[string]interface{}{"access_token": "AT-1", "uid": "u1"})
			body, _ := json.Marshal(map[string]string{
				"platform": "lobsterai", "credential": string(cred), "account": "t",
			})
			if _, err := ps.InvokeHandler("channels", "complete_login", &lua.HandlerRequest{
				Method: "POST", Role: "admin", Body: string(body),
			}); err != nil {
				t.Fatal(err)
			}

			out, err := ps.InvokeHandler("channels", "claim_daily", &lua.HandlerRequest{
				Method: "POST", Role: "admin", Body: `{"platform":"lobsterai"}`,
			})
			if err != nil {
				t.Fatalf("claim_daily: %v", err)
			}
			j, _ := out.JSON.(map[string]interface{})
			oc, _ := j["outcome"].(map[string]interface{})
			if oc == nil {
				t.Fatalf("应返回 outcome 判别联合，实得 %v", j)
			}
			if k, _ := oc["kind"].(string); k != tc.wantKind {
				t.Fatalf("kind 应为 %s，实得 %v（%v）", tc.wantKind, oc["kind"], oc["message"])
			}
			if (claimHit > 0) != tc.wantCalled {
				t.Errorf("claim 请求次数 = %d，期望发出=%v", claimHit, tc.wantCalled)
			}
			if tc.wantCalled && claimBody == "" {
				t.Error("claim 请求体不应为空（要带 configRevision 与幂等键）")
			}
		})
	}
}