mirror of
https://gitea.com/gitea/tea.git
synced 2026-10-26 11:01:25 +01:00
64 lines
3.0 KiB
Go
64 lines
3.0 KiB
Go
// Copyright 2026 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package terminal
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestJSONFieldSelectionAndJQ(t *testing.T) {
|
|
var out bytes.Buffer
|
|
value := []map[string]any{{"number": 1, "title": "first"}, {"number": 2, "title": "second"}}
|
|
require.NoError(t, WriteJSON(&out, value, "title", ".[0].title"))
|
|
require.Equal(t, "first\n", out.String())
|
|
out.Reset()
|
|
require.NoError(t, WriteJSON(&out, []map[string]any{}, "*", ""))
|
|
require.Equal(t, "[]\n", out.String())
|
|
require.Error(t, WriteJSON(&out, value, "unknown", ""))
|
|
require.NoError(t, ValidateOutput("number, title", "", []string{"number", "title"}))
|
|
require.EqualError(t, ValidateOutput("number, foo", "", []string{"number", "title"}), "unknown JSON field \"foo\"\navailable fields: number, title")
|
|
require.EqualError(t, ValidateOutput("number,,title", "", []string{"number", "title"}), "--json field names cannot be empty; use --json alone for every field")
|
|
}
|
|
|
|
func TestJSONFieldSelectionKeepsDocumentedFieldsOmittedByOmitempty(t *testing.T) {
|
|
type item struct {
|
|
Name string `json:"name"`
|
|
Parent *string `json:"parent,omitempty"`
|
|
}
|
|
var out bytes.Buffer
|
|
require.NoError(t, JSONOutput{Fields: "name,parent", Allowed: []string{"name", "parent"}}.Write(&out, []item{{Name: "x"}}))
|
|
require.Equal(t, "[{\"name\":\"x\",\"parent\":null}]\n", out.String())
|
|
require.EqualError(t, JSONOutput{Fields: "parent", Allowed: []string{"name"}}.Write(&out, []item{{Name: "x"}}), `unknown JSON field "parent"`)
|
|
}
|
|
|
|
func TestJSONOutputSafeStripsControlsAndStaysOnOneLine(t *testing.T) {
|
|
value := map[string]any{"title": "a\x1b]0;x\x07b\u202e"}
|
|
var out bytes.Buffer
|
|
require.NoError(t, JSONOutput{Query: ".title", Safe: true}.Write(&out, value))
|
|
require.Equal(t, "a]0;xb\n", out.String())
|
|
out.Reset()
|
|
require.NoError(t, JSONOutput{Safe: true}.Write(&out, value))
|
|
require.Equal(t, "{\"title\":\"a\\u001b]0;x\\u0007b\"}\n", out.String())
|
|
}
|
|
|
|
func TestJSONOutputIsIndentedAndColoredForTerminals(t *testing.T) {
|
|
var out bytes.Buffer
|
|
value := []map[string]any{{"title": "a \u202e\x1b[31mb", "done": true, "due": nil, "number": 1}}
|
|
require.NoError(t, JSONOutput{Fields: "number,title", Pretty: true}.Write(&out, value))
|
|
require.Equal(t, "[\n {\n \"number\": 1,\n \"title\": \"a \\u001b[31mb\"\n }\n]\n", out.String())
|
|
out.Reset()
|
|
require.NoError(t, JSONOutput{Pretty: true, Color: true}.Write(&out, map[string]any{"done": true, "due": nil, "title": "x: \"y\""}))
|
|
require.Equal(t, "{\n \x1b[1;34m\"done\"\x1b[0m: \x1b[33mtrue\x1b[0m,\n \x1b[1;34m\"due\"\x1b[0m: \x1b[36mnull\x1b[0m,\n \x1b[1;34m\"title\"\x1b[0m: \x1b[32m\"x: \\\"y\\\"\"\x1b[0m\n}\n", out.String())
|
|
out.Reset()
|
|
// Raw jq strings on a terminal cannot carry escape sequences; piped they stay byte-exact.
|
|
require.NoError(t, JSONOutput{Query: ".title", Pretty: true}.Write(&out, value[0]))
|
|
require.Equal(t, "a [31mb\n", out.String())
|
|
out.Reset()
|
|
require.NoError(t, WriteJSON(&out, value[0], "", ".title"))
|
|
require.Equal(t, "a \u202e\x1b[31mb\n", out.String())
|
|
}
|