Updated version tests

geofffranks · Aug 6, 2015 · b49c478 · b49c478
1 parent 6d9ee6a
commit b49c478
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main_test.go b/main_test.go
@@ -139,7 +139,7 @@ func TestMain(t *testing.T) {
 				stderr = ""
 				main()
 				So(stdout, ShouldEqual, "")
-				So(stderr, ShouldEqual, "spruce - Version 0.3\n") // SED MARKER FOR AUTO VERSION BUMPING
+				So(stderr, ShouldEqual, fmt.Sprintf("spruce - Version %s\n", VERSION))
 				So(rc, ShouldEqual, 0)
 			})
 			Convey("When '--version' is specified", func() {
@@ -148,7 +148,7 @@ func TestMain(t *testing.T) {
 				stderr = ""
 				main()
 				So(stdout, ShouldEqual, "")
-				So(stderr, ShouldEqual, "spruce - Version 0.3\n") // SED MARKER FOR AUTO VERSION BUMPING
+				So(stderr, ShouldEqual, fmt.Sprintf("spruce - Version %s\n", VERSION))
 				So(rc, ShouldEqual, 0)
 			})
 		})