Merge pull request #83 from alexflint/tweak-examples

Add expected outputs to all runnable examples

1 files changed, 12 insertions, 8 deletions

diff --git a/parse.go b/parse.go
index 32fc619..e23731c 100644
--- a/parse.go
+++ b/parse.go
@@ -13,6 +13,9 @@ import (
 	scalar "github.com/alexflint/go-scalar"
 )
 
+// to enable monkey-patching during tests
+var osExit = os.Exit
+
 // spec represents a command line option
 type spec struct {
 	dest       reflect.Value
@@ -38,20 +41,21 @@ func MustParse(dest ...interface{}) *Parser {
 	p, err := NewParser(Config{}, dest...)
 	if err != nil {
 		fmt.Println(err)
-		os.Exit(-1)
+		osExit(-1)
 	}
+
 	err = p.Parse(flags())
-	if err == ErrHelp {
+	switch {
+	case err == ErrHelp:
 		p.WriteHelp(os.Stdout)
-		os.Exit(0)
-	}
-	if err == ErrVersion {
+		osExit(0)
+	case err == ErrVersion:
 		fmt.Println(p.version)
-		os.Exit(0)
-	}
-	if err != nil {
+		osExit(0)
+	case err != nil:
 		p.Fail(err.Error())
 	}
+
 	return p
 }