mirror of https://github.com/pulumi/pulumi.git
131 lines
4.1 KiB
Go
131 lines
4.1 KiB
Go
// Copyright 2016-2023, Pulumi Corporation.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package display
|
|
|
|
import (
|
|
"bytes"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/pulumi/pulumi/pkg/v3/backend/display/internal/terminal"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/diag/colors"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// Table Test using different terminal widths and heights to ensure that the display does not panic.
|
|
func TestTreeFrameSize(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
// Table Test using different terminal widths and heights
|
|
tests := []struct {
|
|
name string
|
|
width int
|
|
height int
|
|
}{
|
|
{"narrow", 1, 100},
|
|
{"short", 100, 1},
|
|
{"small", 1, 1},
|
|
{"normal", 100, 100},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
tt := tt
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
var buf bytes.Buffer
|
|
term := terminal.NewMockTerminal(&buf, tt.width, tt.height, true)
|
|
treeRenderer := newInteractiveRenderer(term, "this-is-a-fake-permalink", Options{
|
|
Color: colors.Always,
|
|
}).(*treeRenderer)
|
|
|
|
// Fill the renderer with too many rows of strings to fit in the terminal.
|
|
for i := 0; i < 1000; i++ {
|
|
// Fill the renderer with strings that are too long to fit in the terminal.
|
|
treeRenderer.systemMessages = append(treeRenderer.systemMessages, strings.Repeat("a", 1000))
|
|
}
|
|
treeRenderer.treeTableRows = treeRenderer.systemMessages
|
|
|
|
// Required to get the frame to render.
|
|
treeRenderer.markDirty()
|
|
|
|
// This should not panic.
|
|
treeRenderer.frame(false /* locked */, false /* done */)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestTreeKeyboardHandling(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var buf bytes.Buffer
|
|
term := terminal.NewMockTerminal(&buf, 80, 24, true)
|
|
treeRenderer := newInteractiveRenderer(term, "this-is-a-fake-permalink", Options{
|
|
Color: colors.Always,
|
|
}).(*treeRenderer)
|
|
|
|
// Fill the renderer with too many rows of strings to fit in the terminal.
|
|
for i := 0; i < 1000; i++ {
|
|
// Fill the renderer with strings that are too long to fit in the terminal.
|
|
treeRenderer.systemMessages = append(treeRenderer.systemMessages, strings.Repeat("a", 1000))
|
|
}
|
|
treeRenderer.treeTableRows = treeRenderer.systemMessages
|
|
|
|
// Required to get the frame to render.
|
|
treeRenderer.markDirty()
|
|
|
|
// This should not panic.
|
|
treeRenderer.frame(false /* locked */, false /* done */)
|
|
|
|
// manually move the current position to the middle
|
|
treeRenderer.treeTableOffset = treeRenderer.maxTreeTableOffset / 2
|
|
treeRenderer.markDirty()
|
|
treeRenderer.frame(false /* locked */, false /* done */)
|
|
|
|
tests := []struct {
|
|
name string
|
|
key string
|
|
expectedChange int
|
|
expectedAbsolute int
|
|
}{
|
|
{"up arrow", terminal.KeyUp, -1, 0},
|
|
{"VIM up", "k", -1, 0},
|
|
{"Down arrow", terminal.KeyDown, 1, 0},
|
|
{"VIM down", "j", 1, 0},
|
|
{"Page Up", terminal.KeyPageUp, -24, 0},
|
|
{"Page Down", terminal.KeyPageDown, 24, 0},
|
|
{"Home", terminal.KeyHome, 0, 0},
|
|
{"End", terminal.KeyEnd, 0, treeRenderer.maxTreeTableOffset},
|
|
{"VIM home", "g", 0, 0},
|
|
{"VIM end", "G", 0, treeRenderer.maxTreeTableOffset},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
initialValue := treeRenderer.treeTableOffset
|
|
treeRenderer.keys <- tt.key
|
|
// allow the key handler the opportunity to run
|
|
runtime.Gosched()
|
|
treeRenderer.frame(false /* locked */, false /* done */)
|
|
|
|
if tt.expectedChange != 0 {
|
|
assert.Equal(t, tt.expectedChange+initialValue, treeRenderer.treeTableOffset,
|
|
"Current line was not moved to the expected value of %d for %v", tt.expectedChange+initialValue, tt.name)
|
|
} else {
|
|
assert.Equal(t, tt.expectedAbsolute, treeRenderer.treeTableOffset,
|
|
"Current line was not moved to the expected value of %d for %v", tt.expectedAbsolute, tt.name)
|
|
}
|
|
}
|
|
}
|