package document

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestGenerateURLFromDocumentID(t *testing.T) {
	t.Parallel()

	cli := &docCli{larkBaseURL: "pingcap.feishu.cn"}

	require.Equal(t, "https://pingcap.feishu.cn/docx/docx_1", cli.GenerateURLFromDocumentID("docx_1", false))
	require.Equal(t, "https://pingcap.feishu.cn/wiki/wiki_1", cli.GenerateURLFromDocumentID("wiki_1", true))
}