update go mock to allow for both version and env sub-commands

1 files changed, 8 insertions, 1 deletions

diff --git a/test/segments/go_version.spec b/test/segments/go_version.spec
index 173086e0..a085f3b3 100755
--- a/test/segments/go_version.spec
+++ b/test/segments/go_version.spec
@@ -12,7 +12,14 @@ function setUp() {
 }
 
 function mockGo() {
-  echo 'go version go1.5.3 darwin/amd64'
+  case "$1" in
+  'version')
+    echo 'go version go1.5.3 darwin/amd64'
+    ;;
+  'env')
+    echo "$HOME/go"
+    ;;
+  esac
 }
 
 function testGo() {