mukan-sdk/x/gov/client/cli/prompt_test.go

//go:build !race

// Disabled -race because the package github.com/manifoldco/promptui@v0.9.0
// has a data race and this code exposes it, but fixing it would require
// holding up the associated change to this.

package cli_test

import (
	"fmt"
	"math"
	"os"
	"testing"

	"github.com/chzyer/readline"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/cosmos/cosmos-sdk/x/gov/client/cli"
)

type st struct {
	I int
}

// Tests that we successfully report overflows in parsing ints
// See https://github.com/cosmos/cosmos-sdk/issues/13346
func TestPromptIntegerOverflow(t *testing.T) {
	// Intentionally sending values out of the range of int.
	intOverflowers := []string{
		"-9223372036854775809",
		"9223372036854775808",
		"9923372036854775809",
		"-9923372036854775809",
		"18446744073709551616",
		"-18446744073709551616",
	}

	for _, intOverflower := range intOverflowers {
		overflowStr := intOverflower
		t.Run(overflowStr, func(t *testing.T) {
			origStdin := readline.Stdin
			defer func() {
				readline.Stdin = origStdin
			}()

			fin, fw := readline.NewFillableStdin(os.Stdin)
			readline.Stdin = fin
			_, err := fw.Write([]byte(overflowStr + "\n"))
			require.NoError(t, err)

			v, err := cli.Prompt(st{}, "")
			assert.Equal(t, st{}, v, "expected a value of zero")
			require.NotNil(t, err, "expected a report of an overflow")
			require.Contains(t, err.Error(), "range")
		})
	}
}

func TestPromptParseInteger(t *testing.T) {
	// Intentionally sending a value out of the range of
	values := []struct {
		in   string
		want int
	}{
		{fmt.Sprintf("%d", math.MinInt), math.MinInt},
		{"19991", 19991},
		{"991000000199", 991000000199},
	}

	for _, tc := range values {
		t.Run(tc.in, func(t *testing.T) {
			origStdin := readline.Stdin
			defer func() {
				readline.Stdin = origStdin
			}()

			fin, fw := readline.NewFillableStdin(os.Stdin)
			readline.Stdin = fin
			_, err := fw.Write([]byte(tc.in + "\n"))
			assert.NoError(t, err)

			v, err := cli.Prompt(st{}, "")
			assert.Nil(t, err, "expected a nil error")
			assert.Equal(t, tc.want, v.I, "expected %d = %d", tc.want, v.I)
		})
	}
}