From 1394342dd259cbd2e40a4cdecfab3e20dfe71ae7 Mon Sep 17 00:00:00 2001
From: dickeyxxx <jeff@dickeyxxx.com>
Date: Thu, 18 Sep 2014 09:25:16 -0700
Subject: [PATCH] test version command

---
 .travis.yml     |  3 +++
 README.md       |  1 +
 util.go         | 13 +++++++++++++
 version.go      |  9 +--------
 version_test.go | 20 ++++++++++++++++++++
 5 files changed, 38 insertions(+), 8 deletions(-)
 create mode 100644 .travis.yml
 create mode 100644 util.go
 create mode 100644 version_test.go

diff --git a/.travis.yml b/.travis.yml
new file mode 100644
index 0000000..6a363c7
--- /dev/null
+++ b/.travis.yml
@@ -0,0 +1,3 @@
+language: go
+go:
+  - 1.3
diff --git a/README.md b/README.md
index cfa03e3..d227df2 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,5 @@
 ## forego
+[![Build Status](https://travis-ci.org/dickeyxxx/forego.svg)](https://travis-ci.org/dickeyxxx/forego)
 
 Foreman in Go.
 
diff --git a/util.go b/util.go
new file mode 100644
index 0000000..474b680
--- /dev/null
+++ b/util.go
@@ -0,0 +1,13 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"os"
+)
+
+var stdout io.Writer = os.Stdout
+
+func Println(a ...interface{}) (n int, err error) {
+	return fmt.Fprintln(stdout, a...)
+}
diff --git a/version.go b/version.go
index bb22465..b58627c 100644
--- a/version.go
+++ b/version.go
@@ -1,9 +1,5 @@
 package main
 
-import (
-	"fmt"
-)
-
 var Version = "dev"
 
 var cmdVersion = &Command{
@@ -19,9 +15,6 @@ Examples:
 `,
 }
 
-func init() {
-}
-
 func runVersion(cmd *Command, args []string) {
-	fmt.Println(Version)
+	Println(Version)
 }
diff --git a/version_test.go b/version_test.go
new file mode 100644
index 0000000..99bb027
--- /dev/null
+++ b/version_test.go
@@ -0,0 +1,20 @@
+package main
+
+import (
+	"bytes"
+	"testing"
+)
+
+func TestVersion(t *testing.T) {
+	var b bytes.Buffer
+	stdout = &b
+	cmdVersion.Run(cmdVersion, []string{})
+	output := b.String()
+	assertEqual(t, output, "dev\n")
+}
+
+func assertEqual(t *testing.T, a, b interface{}) {
+	if a != b {
+		t.Fatalf(`Expected %#v to equal %#v`, a, b)
+	}
+}