mirror of
https://github.com/larksuite/cli.git
synced 2026-07-05 15:47:54 +08:00
96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
// Copyright (c) 2026 Lark Technologies Pte. Ltd.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package doc
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/larksuite/cli/shortcuts/common"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func TestBuildFetchBodyIncludesSceneFromContext(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctx := context.WithValue(context.Background(), docsSceneContextKey, " DoubaoCLI ")
|
|
runtime := newFetchBodyTestRuntime(ctx)
|
|
|
|
body := buildFetchBody(runtime)
|
|
if got := body["scene"]; got != "DoubaoCLI" {
|
|
t.Fatalf("scene = %#v, want %q", got, "DoubaoCLI")
|
|
}
|
|
}
|
|
|
|
func TestBuildCreateBodyIncludesSceneFromContext(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctx := context.WithValue(context.Background(), docsSceneContextKey, "DoubaoCLI")
|
|
runtime := newCreateBodyTestRuntime(ctx)
|
|
|
|
body := buildCreateBody(runtime)
|
|
if got := body["scene"]; got != "DoubaoCLI" {
|
|
t.Fatalf("scene = %#v, want %q", got, "DoubaoCLI")
|
|
}
|
|
}
|
|
|
|
func TestBuildUpdateBodyIncludesSceneFromContext(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctx := context.WithValue(context.Background(), docsSceneContextKey, "DoubaoCLI")
|
|
runtime := newUpdateBodyTestRuntime(ctx)
|
|
|
|
body := buildUpdateBody(runtime)
|
|
if got := body["scene"]; got != "DoubaoCLI" {
|
|
t.Fatalf("scene = %#v, want %q", got, "DoubaoCLI")
|
|
}
|
|
}
|
|
|
|
func TestBuildFetchBodyOmitsEmptyScene(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
runtime := newFetchBodyTestRuntime(context.Background())
|
|
|
|
body := buildFetchBody(runtime)
|
|
if _, ok := body["scene"]; ok {
|
|
t.Fatalf("did not expect empty scene in fetch body: %#v", body)
|
|
}
|
|
}
|
|
|
|
func newFetchBodyTestRuntime(ctx context.Context) *common.RuntimeContext {
|
|
cmd := &cobra.Command{Use: "+fetch"}
|
|
cmd.Flags().String("doc-format", "xml", "")
|
|
cmd.Flags().String("detail", "simple", "")
|
|
cmd.Flags().Int("revision-id", -1, "")
|
|
cmd.Flags().String("scope", "full", "")
|
|
cmd.Flags().String("start-block-id", "", "")
|
|
cmd.Flags().String("end-block-id", "", "")
|
|
cmd.Flags().String("keyword", "", "")
|
|
cmd.Flags().Int("context-before", 0, "")
|
|
cmd.Flags().Int("context-after", 0, "")
|
|
cmd.Flags().Int("max-depth", -1, "")
|
|
return common.TestNewRuntimeContextWithCtx(ctx, cmd, nil)
|
|
}
|
|
|
|
func newCreateBodyTestRuntime(ctx context.Context) *common.RuntimeContext {
|
|
cmd := &cobra.Command{Use: "+create"}
|
|
cmd.Flags().String("doc-format", "xml", "")
|
|
cmd.Flags().String("content", "<title>hello</title>", "")
|
|
cmd.Flags().String("parent-token", "", "")
|
|
cmd.Flags().String("parent-position", "", "")
|
|
return common.TestNewRuntimeContextWithCtx(ctx, cmd, nil)
|
|
}
|
|
|
|
func newUpdateBodyTestRuntime(ctx context.Context) *common.RuntimeContext {
|
|
cmd := &cobra.Command{Use: "+update"}
|
|
cmd.Flags().String("doc-format", "xml", "")
|
|
cmd.Flags().String("command", "append", "")
|
|
cmd.Flags().Int("revision-id", 0, "")
|
|
cmd.Flags().String("content", "<p>hello</p>", "")
|
|
cmd.Flags().String("pattern", "", "")
|
|
cmd.Flags().String("block-id", "", "")
|
|
cmd.Flags().String("src-block-ids", "", "")
|
|
return common.TestNewRuntimeContextWithCtx(ctx, cmd, nil)
|
|
}
|