fix(helm): add support for local version only

This modifies 'helm version' to allow for local-only or server-only
versions to avoid cases where calling 'helm version' was resulting in
errors.

Closes #1440

cmd/helm/version.go

@@ -30,8 +30,10 @@ import (
 )
 
 type versionCmd struct {
-	out    io.Writer
-	client helm.Interface
+	out        io.Writer
+	client     helm.Interface
+	clientOnly bool
+	serverOnly bool
 }
 
 func newVersionCmd(c helm.Interface, out io.Writer) *cobra.Command {
@@ -41,23 +43,39 @@ func newVersionCmd(c helm.Interface, out io.Writer) *cobra.Command {
 	}
 
 	cmd := &cobra.Command{
-		Use:               "version",
-		Short:             "print the client/server version information",
-		PersistentPreRunE: setupConnection,
+		Use:   "version",
+		Short: "print the client/server version information",
 		RunE: func(cmd *cobra.Command, args []string) error {
+			if !version.clientOnly {
+				// We do this manually instead of in PreRun because we only
+				// need a tunnel if server version is requested.
+				setupConnection(cmd, args)
+			}
 			version.client = ensureHelmClient(version.client)
 			return version.run()
 		},
 	}
+	f := cmd.Flags()
+	f.BoolVarP(&version.clientOnly, "client-only", "c", false, "if set does not query Tiller version")
+	f.BoolVarP(&version.serverOnly, "server-only", "s", false, "if set does not query Helm client version")
 
 	return cmd
 }
 
 func (v *versionCmd) run() error {
-	// Regardless of whether we can talk to server or not, just print the client
-	// version.
-	cv := version.GetVersionProto()
-	fmt.Fprintf(v.out, "Client: %#v\n", cv)
+
+	if v.clientOnly && v.serverOnly {
+		return errors.New("cannot set both client-only and server-only")
+	}
+
+	if !v.serverOnly {
+		cv := version.GetVersionProto()
+		fmt.Fprintf(v.out, "Client: %#v\n", cv)
+	}
+
+	if v.clientOnly {
+		return nil
+	}
 
 	resp, err := v.client.GetVersion()
 	if err != nil {

cmd/helm/version_test.go

+/*
+Copyright 2016 The Kubernetes Authors All rights reserved.
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package main
+
+import (
+	"bytes"
+	"strings"
+	"testing"
+
+	"k8s.io/helm/pkg/version"
+)
+
+func TestVersion(t *testing.T) {
+
+	lver := version.GetVersionProto().SemVer
+	sver := "1.2.3-fakeclient+testonly"
+
+	tests := []struct {
+		name           string
+		client, server bool
+		args           []string
+		fail           bool
+	}{
+		{"default", true, true, []string{}, false},
+		{"client", true, false, []string{"-c"}, false},
+		{"server", false, true, []string{"-s"}, false},
+		{"neither", false, false, []string{"-c", "-s"}, true},
+	}
+
+	for _, tt := range tests {
+		b := new(bytes.Buffer)
+		c := &fakeReleaseClient{}
+
+		cmd := newVersionCmd(c, b)
+		cmd.ParseFlags(tt.args)
+		if err := cmd.RunE(cmd, tt.args); err != nil {
+			if tt.fail {
+				continue
+			}
+			t.Fatal(err)
+		}
+
+		if tt.client && !strings.Contains(b.String(), lver) {
+			t.Errorf("Expected %q to contain %q", b.String(), lver)
+		}
+		if tt.server && !strings.Contains(b.String(), sver) {
+			t.Errorf("Expected %q to contain %q", b.String(), sver)
+		}
+	}
+}