mirror of
https://github.com/larksuite/cli.git
synced 2026-07-03 14:02:43 +08:00
900c12ce8d
* feat: add FileIO extension for file transfer abstraction Introduce extension/fileio package with Provider/FileIO/File interfaces and a global registry, following the same pattern as extension/credential. - Add LocalFileIO default implementation with path validation and atomic writes - Wire FileIOProvider into Factory and resolve at runtime via RuntimeContext.FileIO() - Factory holds Provider (not resolved instance), deferring resolution to execution time
112 lines
3.6 KiB
Go
112 lines
3.6 KiB
Go
// Copyright (c) 2026 Lark Technologies Pte. Ltd.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package cmdutil
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"net/http"
|
|
"os"
|
|
"testing"
|
|
|
|
lark "github.com/larksuite/oapi-sdk-go/v3"
|
|
larkcore "github.com/larksuite/oapi-sdk-go/v3/core"
|
|
|
|
"github.com/larksuite/cli/extension/fileio"
|
|
"github.com/larksuite/cli/internal/core"
|
|
"github.com/larksuite/cli/internal/credential"
|
|
"github.com/larksuite/cli/internal/httpmock"
|
|
"github.com/larksuite/cli/internal/vfs"
|
|
)
|
|
|
|
// noopKeychain is a no-op KeychainAccess for tests that don't need keychain.
|
|
type noopKeychain struct{}
|
|
|
|
func (n *noopKeychain) Get(service, account string) (string, error) { return "", nil }
|
|
func (n *noopKeychain) Set(service, account, value string) error { return nil }
|
|
func (n *noopKeychain) Remove(service, account string) error { return nil }
|
|
|
|
// TestFactory creates a Factory for testing.
|
|
// Returns (factory, stdout buffer, stderr buffer, http mock registry).
|
|
func TestFactory(t *testing.T, config *core.CliConfig) (*Factory, *bytes.Buffer, *bytes.Buffer, *httpmock.Registry) {
|
|
t.Helper()
|
|
|
|
reg := &httpmock.Registry{}
|
|
t.Cleanup(func() { reg.Verify(t) })
|
|
|
|
stdoutBuf := &bytes.Buffer{}
|
|
stderrBuf := &bytes.Buffer{}
|
|
|
|
mockClient := httpmock.NewClient(reg)
|
|
sdkMockClient := &http.Client{
|
|
Transport: &UserAgentTransport{Base: reg},
|
|
}
|
|
|
|
var testLarkClient *lark.Client
|
|
if config != nil && config.AppID != "" {
|
|
opts := []lark.ClientOptionFunc{
|
|
lark.WithEnableTokenCache(false),
|
|
lark.WithLogLevel(larkcore.LogLevelError),
|
|
lark.WithHttpClient(sdkMockClient),
|
|
lark.WithHeaders(BaseSecurityHeaders()),
|
|
}
|
|
if config.Brand != "" {
|
|
opts = append(opts, lark.WithOpenBaseUrl(core.ResolveOpenBaseURL(config.Brand)))
|
|
}
|
|
testLarkClient = lark.NewClient(config.AppID, credential.RuntimeAppSecret(config.AppSecret), opts...)
|
|
}
|
|
|
|
testCred := credential.NewCredentialProvider(
|
|
nil,
|
|
&testDefaultAcct{config: config},
|
|
&testDefaultToken{},
|
|
func() (*http.Client, error) { return mockClient, nil },
|
|
)
|
|
|
|
f := &Factory{
|
|
Config: func() (*core.CliConfig, error) { return config, nil },
|
|
HttpClient: func() (*http.Client, error) { return mockClient, nil },
|
|
LarkClient: func() (*lark.Client, error) { return testLarkClient, nil },
|
|
IOStreams: &IOStreams{In: nil, Out: stdoutBuf, ErrOut: stderrBuf},
|
|
Keychain: &noopKeychain{},
|
|
Credential: testCred,
|
|
FileIOProvider: fileio.GetProvider(),
|
|
}
|
|
return f, stdoutBuf, stderrBuf, reg
|
|
}
|
|
|
|
type testDefaultAcct struct {
|
|
config *core.CliConfig
|
|
}
|
|
|
|
func (a *testDefaultAcct) ResolveAccount(ctx context.Context) (*credential.Account, error) {
|
|
if a.config == nil {
|
|
return &credential.Account{}, nil
|
|
}
|
|
return credential.AccountFromCliConfig(a.config), nil
|
|
}
|
|
|
|
// TestChdir changes the working directory to dir for the duration of the test.
|
|
// The original directory is restored via t.Cleanup.
|
|
// This enables tests to use LocalFileIO (which resolves relative paths under cwd)
|
|
// with temporary directories, keeping test artifacts out of the source tree.
|
|
// Not compatible with t.Parallel() — os.Chdir is process-wide.
|
|
func TestChdir(t *testing.T, dir string) {
|
|
t.Helper()
|
|
orig, err := vfs.Getwd()
|
|
if err != nil {
|
|
t.Fatalf("Getwd: %v", err)
|
|
}
|
|
if err := os.Chdir(dir); err != nil { //nolint:forbidigo // no vfs.Chdir yet; test-only, process-wide chdir
|
|
t.Fatalf("Chdir(%s): %v", dir, err)
|
|
}
|
|
t.Cleanup(func() { os.Chdir(orig) }) //nolint:forbidigo // matching restore
|
|
}
|
|
|
|
type testDefaultToken struct{}
|
|
|
|
func (t *testDefaultToken) ResolveToken(ctx context.Context, req credential.TokenSpec) (*credential.TokenResult, error) {
|
|
return &credential.TokenResult{Token: "test-token"}, nil
|
|
}
|