github/kubernetes
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix kube2sky flakes. Fix tools.GetEtcdVersion to work with etcd > 2.0.7

Browse Source
This commit is contained in:
CJ Cullen
2015-05-02 00:13:52 -07:00
parent 83b70c4411
commit a21713a771
4 changed files with 47 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
pkg/tools/etcd_helper.go
View File

@@ -17,7 +17,6 @@ limitations under the License.
package tools
import (
"encoding/json"
"errors"
"fmt"
"io/ioutil"
@@ -475,35 +474,22 @@ func (h *EtcdHelper) PrefixEtcdKey(key string) string {
return path.Join("/", h.PathPrefix, key)
}
// GetEtcdVersion performs a version check against the provided Etcd server, returning a triplet
// of the release version, internal version, and error (if any).
func GetEtcdVersion(host string) (releaseVersion, internalVersion string, err error) {
// GetEtcdVersion performs a version check against the provided Etcd server,
// returning the string response, and error (if any).
func GetEtcdVersion(host string) (string, error) {
response, err := http.Get(host + "/version")
if err != nil {
return "", "", err
return "", err
}
defer response.Body.Close()
body, err := ioutil.ReadAll(response.Body)
if response.StatusCode != http.StatusOK {
return "", fmt.Errorf("Unsuccessful response from server: %v", err)
}
versionBytes, err := ioutil.ReadAll(response.Body)
if err != nil {
return "", "", err
return "", err
}
var dat map[string]interface{}
if err := json.Unmarshal(body, &dat); err != nil {
return "", "", fmt.Errorf("unknown server: %s", string(body))
}
if obj := dat["releaseVersion"]; obj != nil {
if s, ok := obj.(string); ok {
releaseVersion = s
}
}
if obj := dat["internalVersion"]; obj != nil {
if s, ok := obj.(string); ok {
internalVersion = s
}
}
return
return string(versionBytes), nil
}
func startEtcd() (*exec.Cmd, error) {
@@ -516,7 +502,7 @@ func startEtcd() (*exec.Cmd, error) {
}
func NewEtcdClientStartServerIfNecessary(server string) (EtcdClient, error) {
_, _, err := GetEtcdVersion(server)
_, err := GetEtcdVersion(server)
if err != nil {
glog.Infof("Failed to find etcd, attempting to start.")
_, err := startEtcd()

35
pkg/tools/etcd_helper_test.go
View File

@@ -685,35 +685,16 @@ func TestGuaranteedUpdate_CreateCollision(t *testing.T) {
func TestGetEtcdVersion_ValidVersion(t *testing.T) {
testServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintln(w, "{\"releaseVersion\":\"2.0.3\",\"internalVersion\":\"2\"}")
fmt.Fprint(w, "etcd 2.0.9")
}))
defer testServer.Close()
var relVersion string
var intVersion string
var version string
var err error
if relVersion, intVersion, err = GetEtcdVersion(testServer.URL); err != nil {
if version, err = GetEtcdVersion(testServer.URL); err != nil {
t.Errorf("Unexpected error: %v", err)
}
assert.Equal(t, "2.0.3", relVersion, "Unexpected external version")
assert.Equal(t, "2", intVersion, "Unexpected internal version")
assert.Nil(t, err)
}
func TestGetEtcdVersion_UnknownVersion(t *testing.T) {
testServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintln(w, "{\"unknownAttribute\":\"foobar\",\"internalVersion\":\"2\"}")
}))
defer testServer.Close()
var relVersion string
var intVersion string
var err error
if relVersion, intVersion, err = GetEtcdVersion(testServer.URL); err != nil {
t.Errorf("Unexpected error: %v", err)
}
assert.Equal(t, "", relVersion, "Unexpected external version")
assert.Equal(t, "2", intVersion, "Unexpected internal version")
assert.Equal(t, "etcd 2.0.9", version, "Unexpected version")
assert.Nil(t, err)
}
@@ -723,8 +704,12 @@ func TestGetEtcdVersion_ErrorStatus(t *testing.T) {
}))
defer testServer.Close()
var err error
_, _, err = GetEtcdVersion(testServer.URL)
_, err := GetEtcdVersion(testServer.URL)
assert.NotNil(t, err)
}
func TestGetEtcdVersion_NotListening(t *testing.T) {
_, err := GetEtcdVersion("http://127.0.0.1:4001")
assert.NotNil(t, err)
}