constraint_enum_test.go

package testing

import "testing"

func Test_Constraint_Enum_Valid(t *testing.T) {
	rsl := `
args:
	name string
	name enum ["alice", "bob", "charlie"]
print("Hi", name)
`
	setupAndRunCode(t, rsl, "alice")
	assertOnlyOutput(t, stdOutBuffer, "Hi alice\n")
	assertNoErrors(t)
}

func Test_Constraint_Enum_ErrorsOnInvalid(t *testing.T) {
	rsl := `
args:
	name string
	name enum ["alice", "bob", "charlie"]
print("Hi", name)
`
	setupAndRunCode(t, rsl, "david", "--color=never")
	expected := `Invalid 'name' value: david (valid values: alice, bob, charlie)

Usage:
  <name>

Script args:
      --name string   Valid values: [alice, bob, charlie].

` + scriptGlobalFlagHelp
	assertError(t, 1, expected)
}

func Test_Constraint_Enum_ErrorsIfNonStringEnum(t *testing.T) {
	rsl := `
args:
    name string
    name enum ["alice", 2]
print("Hi", name)
`
	setupAndRunCode(t, rsl, "david", "--color=never")
	expected := `Error at L4:25

      name enum ["alice", 2]
                          ^ Invalid syntax
`
	assertError(t, 1, expected)
}

func Test_Constraint_Enum_CanHaveArgNamedEnum(t *testing.T) {
	rsl := `
args:
	enum string
	enum enum ["alice", "bob", "charlie"]
print("Hi", enum)
`
	setupAndRunCode(t, rsl, "alice")
	assertOnlyOutput(t, stdOutBuffer, "Hi alice\n")
	assertNoErrors(t)
}