packer-cn/main_test.go

70 lines
1.4 KiB
Go

package main
import (
"math/rand"
"reflect"
"strings"
"testing"
"github.com/hashicorp/packer/command"
"github.com/mitchellh/cli"
)
func TestExcludeHelpFunc(t *testing.T) {
commands := map[string]cli.CommandFactory{
"build": func() (cli.Command, error) {
return &command.BuildCommand{
Meta: command.Meta{},
}, nil
},
"fix": func() (cli.Command, error) {
return &command.FixCommand{
Meta: command.Meta{},
}, nil
},
}
helpFunc := excludeHelpFunc(commands, []string{"fix"})
helpText := helpFunc(commands)
if strings.Contains(helpText, "fix") {
t.Fatalf("Found fix in help text even though we excluded it: \n\n%s\n\n", helpText)
}
}
func TestExtractMachineReadable(t *testing.T) {
var args, expected, result []string
var mr bool
// Not
args = []string{"foo", "bar", "baz"}
result, mr = extractMachineReadable(args)
expected = []string{"foo", "bar", "baz"}
if !reflect.DeepEqual(result, expected) {
t.Fatalf("bad: %#v", result)
}
if mr {
t.Fatal("should not be mr")
}
// Yes
args = []string{"foo", "-machine-readable", "baz"}
result, mr = extractMachineReadable(args)
expected = []string{"foo", "baz"}
if !reflect.DeepEqual(result, expected) {
t.Fatalf("bad: %#v", result)
}
if !mr {
t.Fatal("should be mr")
}
}
func TestRandom(t *testing.T) {
if rand.Intn(9999999) == 8498210 {
t.Fatal("math.rand is not seeded properly")
}
}