xinmi/internal/datasource/connector/feishu/wiki/lark_e2e_test.go

package wiki

import (
	"context"
	"strings"
	"testing"

	"github.com/Tencent/WeKnora/internal/datasource/connector/feishu/core"
	"github.com/Tencent/WeKnora/internal/types"
)

// A Lark connector must surface wiki links on the Lark host. Before Lark became
// its own connector type, every resource URL was hardcoded to feishu.cn, so Lark
// users got links into a cloud their content does not live on.
//
// The fake server stands in for the Open Platform (base_url points at it), but
// the resource URL is built from the region's web host, so this exercises the
// real ListResources → wikiNodeToResource → region.WikiURL path.
func TestListResources_ResourceURLFollowsRegion(t *testing.T) {
	cases := []struct {
		region      core.Region
		wantURL     string
		forbiddenIn string
	}{
		{core.RegionFeishu, "https://feishu.cn/wiki/space1", "larksuite"},
		{core.RegionLark, "https://larksuite.com/wiki/space1", "feishu"},
	}

	for _, c := range cases {
		t.Run(c.region.Label, func(t *testing.T) {
			ts, cfg := fakeFeishu(nil)
			defer ts.Close()

			conn := NewConnector(c.region)
			resources, err := conn.ListResources(context.Background(), makeConfig(cfg, nil), "")
			if err != nil {
				t.Fatalf("ListResources: %v", err)
			}
			if len(resources) != 1 {
				t.Fatalf("got %d resources, want 1", len(resources))
			}
			if got := resources[0].URL; got != c.wantURL {
				t.Errorf("resource URL = %q, want %q", got, c.wantURL)
			}
			if strings.Contains(resources[0].URL, c.forbiddenIn) {
				t.Errorf("%s resource URL leaks the other cloud's host: %q", c.region.Label, resources[0].URL)
			}
		})
	}
}

// A Lark connector with no base_url override must reach open.larksuite.com,
// never open.feishu.cn — the SDK-less client has no default of its own, so the
// region is the only thing standing between a Lark app and the wrong cloud.
func TestClient_ResolvesRegionHostWithoutOverride(t *testing.T) {
	cases := []struct {
		region core.Region
		want   string
	}{
		{core.RegionFeishu, "https://open.feishu.cn"},
		{core.RegionLark, "https://open.larksuite.com"},
	}

	for _, c := range cases {
		t.Run(c.region.Label, func(t *testing.T) {
			cfg, err := core.ParseFeishuConfig(makeConfigNoBaseURL(), c.region)
			if err != nil {
				t.Fatalf("ParseFeishuConfig: %v", err)
			}
			if cfg.GetBaseURL() != c.want {
				t.Errorf("baseURL = %q, want %q", cfg.GetBaseURL(), c.want)
			}
		})
	}
}

// makeConfigNoBaseURL builds credentials without base_url, so the region default
// is what resolves.
func makeConfigNoBaseURL() *types.DataSourceConfig {
	return &types.DataSourceConfig{
		Credentials: map[string]interface{}{
			"app_id":     "cli_x",
			"app_secret": "s",
		},
	}
}