commands/commandstest: Add

Author: rajivr

commands/commandstest/fake_command_line.go

@@ -0,0 +1,100 @@
+package commandstest
+
+import (
+	"github.com/codegangsta/cli"
+)
+
+type FakeFlagger struct {
+	Data map[string]interface{}
+}
+
+type FakeCommandLine struct {
+	LocalFlags, GlobalFlags *FakeFlagger
+	HelpShown, VersionShown bool
+	CliArgs                 []string
+}
+
+func (ff FakeFlagger) String(key string) string {
+	if value, ok := ff.Data[key]; ok {
+		return value.(string)
+	}
+	return ""
+}
+
+func (ff FakeFlagger) StringSlice(key string) []string {
+	if value, ok := ff.Data[key]; ok {
+		return value.([]string)
+	}
+	return []string{}
+}
+
+func (ff FakeFlagger) Int(key string) int {
+	if value, ok := ff.Data[key]; ok {
+		return value.(int)
+	}
+	return 0
+}
+
+func (ff FakeFlagger) Bool(key string) bool {
+	if value, ok := ff.Data[key]; ok {
+		return value.(bool)
+	}
+	return false
+}
+
+func (fcli *FakeCommandLine) IsSet(key string) bool {
+	_, ok := fcli.LocalFlags.Data[key]
+	return ok
+}
+
+func (fcli *FakeCommandLine) String(key string) string {
+	return fcli.LocalFlags.String(key)
+}
+
+func (fcli *FakeCommandLine) StringSlice(key string) []string {
+	return fcli.LocalFlags.StringSlice(key)
+}
+
+func (fcli *FakeCommandLine) Int(key string) int {
+	return fcli.LocalFlags.Int(key)
+}
+
+func (fcli *FakeCommandLine) Bool(key string) bool {
+	if fcli.LocalFlags == nil {
+		return false
+	}
+	return fcli.LocalFlags.Bool(key)
+}
+
+func (fcli *FakeCommandLine) GlobalString(key string) string {
+	return fcli.GlobalFlags.String(key)
+}
+
+func (fcli *FakeCommandLine) Generic(name string) interface{} {
+	return fcli.LocalFlags.Data[name]
+}
+
+func (fcli *FakeCommandLine) FlagNames() []string {
+	flagNames := []string{}
+	for key := range fcli.LocalFlags.Data {
+		flagNames = append(flagNames, key)
+	}
+
+	return flagNames
+}
+
+func (fcli *FakeCommandLine) ShowHelp() {
+	fcli.HelpShown = true
+}
+
+func (fcli *FakeCommandLine) Application() *cli.App {
+	return cli.NewApp()
+}
+
+func (fcli *FakeCommandLine) Args() cli.Args {
+	return fcli.CliArgs
+}
+
+func (fcli *FakeCommandLine) ShowVersion() {
+	fcli.VersionShown = true
+}

commands/commandstest/stdout_capture.go

@@ -0,0 +1,54 @@
+package commandstest
+
+import (
+	"bytes"
+	"io"
+
+	"os"
+)
+
+var (
+	stdout *os.File
+)
+
+func init() {
+	stdout = os.Stdout
+}
+
+type StdoutGetter interface {
+	Output() string
+	Stop()
+}
+
+type stdoutCapturer struct {
+	stdout *os.File
+	output chan string
+}
+
+func NewStdoutGetter() StdoutGetter {
+	r, w, _ := os.Pipe()
+	os.Stdout = w
+
+	output := make(chan string)
+	go func() {
+		var testOutput bytes.Buffer
+		io.Copy(&testOutput, r)
+		output <- testOutput.String()
+	}()
+
+	return &stdoutCapturer{
+		stdout: w,
+		output: output,
+	}
+}
+
+func (c *stdoutCapturer) Output() string {
+	c.stdout.Close()
+	text := <-c.output
+	close(c.output)
+	return text
+}
+
+func (c *stdoutCapturer) Stop() {
+	os.Stdout = stdout
+}