Function TestUserSupportsMultipleProtocols

internal/serverapi/users_test.go:173–235

(t *testing.T)

}

func TestUserSupportsMultipleProtocols(t *testing.T) {
	mux, ibStore := setupTestMux(t, map[string]func(body any) (json.RawMessage, error){
		"Restart": func(any) (json.RawMessage, error) {
			return json.RawMessage(`{"running":true}`), nil
		},
	})

	// 在 node-1 上注册 trojan 和 shadowsocks 入站
	_, _ = ibStore.UpsertInbound(inbounds.Inbound{
		ID:       "ib-trojan",
		NodeID:   "node-1",
		Protocol: "trojan",
		Tag:      "pulse-trojan-node-1",
		Port:     443,
	})
	_, _ = ibStore.UpsertHost(inbounds.Host{
		ID:        "host-trojan",
		InboundID: "ib-trojan",
		Address:   "example.com",
		Port:      443,
	})
	_, _ = ibStore.UpsertInbound(inbounds.Inbound{
		ID:       "ib-ss",
		NodeID:   "node-1",
		Protocol: "shadowsocks",
		Tag:      "pulse-ss-node-1",
		Port:     8443,
		Method:   "aes-256-gcm",
	})
	_, _ = ibStore.UpsertHost(inbounds.Host{
		ID:        "host-ss",
		InboundID: "ib-ss",
		Address:   "example.com",
		Port:      8443,
	})

	// 创建用户，分别绑定 trojan 和 ss 两个 inbound
	userID := createUser(t, mux, `{"id":"user-multi","username":"alice"}`)
	trojanAccessID := createUserInbound(t, mux, userID, "ib-trojan")
	ssAccessID := createUserInbound(t, mux, userID, "ib-ss")

	// trojan 绑定的订阅只包含 trojan 链接
	rec := httptest.NewRecorder()
	req := httptest.NewRequest(http.MethodGet, "/v1/users/"+userID+"/inbounds/"+trojanAccessID+"/subscription", nil)
	mux.ServeHTTP(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("trojan subscription status = %d body=%s", rec.Code, rec.Body.String())
	}
	if !strings.Contains(rec.Body.String(), "trojan://") {
		t.Fatalf("expected trojan link, got %s", rec.Body.String())
	}

	// ss 绑定的订阅只包含 ss 链接
	rec = httptest.NewRecorder()
	req = httptest.NewRequest(http.MethodGet, "/v1/users/"+userID+"/inbounds/"+ssAccessID+"/subscription", nil)
	mux.ServeHTTP(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("ss subscription status = %d body=%s", rec.Code, rec.Body.String())