Ctrl K
ComunidadGuías…

pkg/jsplugin/cli_test.go

Rama main · réplica de lectura del repositorio original.

Archivo original ↗Preguntar a KiBot
package jsplugin

import (
	"bytes"
	"os"
	"path/filepath"
	"testing"

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

func TestPluginCLI(t *testing.T) {
	tempDir := t.TempDir()
	pluginPath := filepath.Join(tempDir, "fixture.js")
	fixturePath := filepath.Join(tempDir, "fixture.json")
	require.NoError(t, os.WriteFile(pluginPath, []byte(cliFixturePluginSource), 0o600))
	require.NoError(t, os.WriteFile(fixturePath, []byte(`{"unixNow":42,"cases":[{"hook":"value","args":[],"expected":42}]}`), 0o600))

	var stdout bytes.Buffer
	var stderr bytes.Buffer
	assert.Equal(t, 0, RunCLI([]string{"lint", pluginPath}, &stdout, &stderr))
	assert.Contains(t, stdout.String(), "plugin cli-fixture@1.0.0 is valid")
	assert.Empty(t, stderr.String())

	stdout.Reset()
	assert.Equal(t, 0, RunCLI([]string{"test", pluginPath, "--fixture", fixturePath}, &stdout, &stderr))
	assert.Contains(t, stdout.String(), "1/1 cases")
}

func TestPluginCLIWarnsOnParseTaskResultInProgressFallback(t *testing.T) {
	tempDir := t.TempDir()
	pluginPath := filepath.Join(tempDir, "fallback.js")
	require.NoError(t, os.WriteFile(pluginPath, []byte(`
export const meta = { apiVersion: 1, key: "fallback", name: "Fallback", version: "1.0.0", author: {name: "Test"}, models: ["m"], fetchMode: "per_task" };
export function buildSubmitRequest(ctx) { return {url: ctx.baseUrl}; }
export function parseSubmitResponse() { return {taskId: "task"}; }
export function buildQueryRequest(ctx) { return {url: ctx.baseUrl}; }
export function parseTaskResult(ctx, body) { return {status: statuses[body.status] || "IN_PROGRESS"}; }
const statuses = { done: "SUCCESS" };
`), 0o600))

	var stdout bytes.Buffer
	var stderr bytes.Buffer
	assert.Equal(t, 0, RunCLI([]string{"lint", pluginPath}, &stdout, &stderr))
	assert.Contains(t, stdout.String(), "plugin fallback@1.0.0 is valid")
	assert.Contains(t, stderr.String(), `|| "IN_PROGRESS"`)
}

const cliFixturePluginSource = `
export const meta = { apiVersion: 1, key: "cli-fixture", name: "CLI Fixture", version: "1.0.0", author: {name: "Test"}, channelTypes: [1003], models: ["fixture-model"], fetchMode: "per_task" };
export function buildSubmitRequest(ctx) { return {url: ctx.baseUrl}; }
export function parseSubmitResponse(ctx, resp) { return {taskId: "task"}; }
export function buildQueryRequest(ctx) { return {url: ctx.baseUrl}; }
export function parseTaskResult(ctx, body) { return body; }
export function value() { return utils.unixNow(); }
`