summaryrefslogtreecommitdiff
path: root/compflag/compflag_test.go
diff options
context:
space:
mode:
authorEyal Posener <[email protected]>2019-11-22 11:40:09 +0200
committerEyal Posener <[email protected]>2019-11-22 11:43:37 +0200
commite8bbd08d57dcfee050a1b098e793cea633b5d35c (patch)
tree76704b78ed8046c51d1775ee87c6e04497489f7a /compflag/compflag_test.go
parentcf002b9a50d777b10beaf228f316c5da34f8178a (diff)
Extract flag options to predict package
Diffstat (limited to 'compflag/compflag_test.go')
-rw-r--r--compflag/compflag_test.go19
1 files changed, 10 insertions, 9 deletions
diff --git a/compflag/compflag_test.go b/compflag/compflag_test.go
index 037ef07..73dc2c1 100644
--- a/compflag/compflag_test.go
+++ b/compflag/compflag_test.go
@@ -5,6 +5,7 @@ import (
"testing"
"github.com/posener/complete/v2"
+ "github.com/posener/complete/v2/predict"
"github.com/stretchr/testify/assert"
)
@@ -30,7 +31,7 @@ func TestString(t *testing.T) {
t.Run("options invalid not checked", func(t *testing.T) {
var cmd FlagSet
- value := cmd.String("a", "", "", OptValues("1", "2"))
+ value := cmd.String("a", "", "", predict.OptValues("1", "2"))
err := cmd.Parse([]string{"-a", "3"})
assert.NoError(t, err)
assert.Equal(t, "3", *value)
@@ -38,7 +39,7 @@ func TestString(t *testing.T) {
t.Run("options valid checked", func(t *testing.T) {
var cmd FlagSet
- value := cmd.String("a", "", "", OptValues("1", "2"), OptCheck())
+ value := cmd.String("a", "", "", predict.OptValues("1", "2"), predict.OptCheck())
err := cmd.Parse([]string{"-a", "2"})
assert.NoError(t, err)
assert.Equal(t, "2", *value)
@@ -46,14 +47,14 @@ func TestString(t *testing.T) {
t.Run("options invalid checked", func(t *testing.T) {
var cmd FlagSet
- _ = cmd.String("a", "", "", OptValues("1", "2"), OptCheck())
+ _ = cmd.String("a", "", "", predict.OptValues("1", "2"), predict.OptCheck())
err := cmd.Parse([]string{"-a", "3"})
assert.Error(t, err)
})
t.Run("complete", func(t *testing.T) {
var cmd FlagSet
- _ = cmd.String("a", "", "", OptValues("1", "2"))
+ _ = cmd.String("a", "", "", predict.OptValues("1", "2"))
complete.Test(t, complete.FlagSet((*flag.FlagSet)(&cmd)), "-a ", []string{"1", "2"})
complete.Test(t, complete.FlagSet((*flag.FlagSet)(&cmd)), "-a=", []string{"1", "2"})
complete.Test(t, complete.FlagSet((*flag.FlagSet)(&cmd)), "-a 1", []string{"1"})
@@ -66,7 +67,7 @@ func TestInt(t *testing.T) {
t.Run("options invalid not checked", func(t *testing.T) {
var cmd FlagSet
- value := cmd.Int("a", 0, "", OptValues("1", "2"))
+ value := cmd.Int("a", 0, "", predict.OptValues("1", "2"))
err := cmd.Parse([]string{"-a", "3"})
assert.NoError(t, err)
assert.Equal(t, 3, *value)
@@ -74,7 +75,7 @@ func TestInt(t *testing.T) {
t.Run("options valid checked", func(t *testing.T) {
var cmd FlagSet
- value := cmd.Int("a", 0, "", OptValues("1", "2"), OptCheck())
+ value := cmd.Int("a", 0, "", predict.OptValues("1", "2"), predict.OptCheck())
err := cmd.Parse([]string{"-a", "2"})
assert.NoError(t, err)
assert.Equal(t, 2, *value)
@@ -82,21 +83,21 @@ func TestInt(t *testing.T) {
t.Run("options invalid checked", func(t *testing.T) {
var cmd FlagSet
- _ = cmd.Int("a", 0, "", OptValues("1", "2"), OptCheck())
+ _ = cmd.Int("a", 0, "", predict.OptValues("1", "2"), predict.OptCheck())
err := cmd.Parse([]string{"-a", "3"})
assert.Error(t, err)
})
t.Run("options invalid int value", func(t *testing.T) {
var cmd FlagSet
- _ = cmd.Int("a", 0, "", OptValues("1", "2", "x"), OptCheck())
+ _ = cmd.Int("a", 0, "", predict.OptValues("1", "2", "x"), predict.OptCheck())
err := cmd.Parse([]string{"-a", "x"})
assert.Error(t, err)
})
t.Run("complete", func(t *testing.T) {
var cmd FlagSet
- _ = cmd.Int("a", 0, "", OptValues("1", "2"))
+ _ = cmd.Int("a", 0, "", predict.OptValues("1", "2"))
complete.Test(t, complete.FlagSet((*flag.FlagSet)(&cmd)), "-a ", []string{"1", "2"})
complete.Test(t, complete.FlagSet((*flag.FlagSet)(&cmd)), "-a=", []string{"1", "2"})
complete.Test(t, complete.FlagSet((*flag.FlagSet)(&cmd)), "-a 1", []string{"1"})