Merge pull request #752 from smarterclayton/remove_expect_no_error_util
Remove expectNoError from client, kubelet, and util
This commit is contained in:
commit
d4860c21a9
@ -30,19 +30,15 @@ import (
|
|||||||
"github.com/fsouza/go-dockerclient"
|
"github.com/fsouza/go-dockerclient"
|
||||||
)
|
)
|
||||||
|
|
||||||
// TODO: This doesn't reduce typing enough to make it worth the less readable errors. Remove.
|
|
||||||
func expectNoError(t *testing.T, err error) {
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("Unexpected error: %#v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestHTTPPodInfoGetter(t *testing.T) {
|
func TestHTTPPodInfoGetter(t *testing.T) {
|
||||||
expectObj := api.PodInfo{
|
expectObj := api.PodInfo{
|
||||||
"myID": docker.Container{ID: "myID"},
|
"myID": docker.Container{ID: "myID"},
|
||||||
}
|
}
|
||||||
body, err := json.Marshal(expectObj)
|
body, err := json.Marshal(expectObj)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
fakeHandler := util.FakeHandler{
|
fakeHandler := util.FakeHandler{
|
||||||
StatusCode: 200,
|
StatusCode: 200,
|
||||||
ResponseBody: string(body),
|
ResponseBody: string(body),
|
||||||
@ -50,17 +46,25 @@ func TestHTTPPodInfoGetter(t *testing.T) {
|
|||||||
testServer := httptest.NewServer(&fakeHandler)
|
testServer := httptest.NewServer(&fakeHandler)
|
||||||
|
|
||||||
hostURL, err := url.Parse(testServer.URL)
|
hostURL, err := url.Parse(testServer.URL)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
parts := strings.Split(hostURL.Host, ":")
|
parts := strings.Split(hostURL.Host, ":")
|
||||||
|
|
||||||
port, err := strconv.Atoi(parts[1])
|
port, err := strconv.Atoi(parts[1])
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
podInfoGetter := &HTTPPodInfoGetter{
|
podInfoGetter := &HTTPPodInfoGetter{
|
||||||
Client: http.DefaultClient,
|
Client: http.DefaultClient,
|
||||||
Port: uint(port),
|
Port: uint(port),
|
||||||
}
|
}
|
||||||
gotObj, err := podInfoGetter.GetPodInfo(parts[0], "foo")
|
gotObj, err := podInfoGetter.GetPodInfo(parts[0], "foo")
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
// reflect.DeepEqual(expectObj, gotObj) doesn't handle blank times well
|
// reflect.DeepEqual(expectObj, gotObj) doesn't handle blank times well
|
||||||
if len(gotObj) != len(expectObj) || expectObj["myID"].ID != gotObj["myID"].ID {
|
if len(gotObj) != len(expectObj) || expectObj["myID"].ID != gotObj["myID"].ID {
|
||||||
@ -73,7 +77,10 @@ func TestHTTPPodInfoGetterNotFound(t *testing.T) {
|
|||||||
"myID": docker.Container{ID: "myID"},
|
"myID": docker.Container{ID: "myID"},
|
||||||
}
|
}
|
||||||
_, err := json.Marshal(expectObj)
|
_, err := json.Marshal(expectObj)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
fakeHandler := util.FakeHandler{
|
fakeHandler := util.FakeHandler{
|
||||||
StatusCode: 404,
|
StatusCode: 404,
|
||||||
ResponseBody: "Pod not found",
|
ResponseBody: "Pod not found",
|
||||||
@ -81,11 +88,17 @@ func TestHTTPPodInfoGetterNotFound(t *testing.T) {
|
|||||||
testServer := httptest.NewServer(&fakeHandler)
|
testServer := httptest.NewServer(&fakeHandler)
|
||||||
|
|
||||||
hostURL, err := url.Parse(testServer.URL)
|
hostURL, err := url.Parse(testServer.URL)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
parts := strings.Split(hostURL.Host, ":")
|
parts := strings.Split(hostURL.Host, ":")
|
||||||
|
|
||||||
port, err := strconv.Atoi(parts[1])
|
port, err := strconv.Atoi(parts[1])
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
podInfoGetter := &HTTPPodInfoGetter{
|
podInfoGetter := &HTTPPodInfoGetter{
|
||||||
Client: http.DefaultClient,
|
Client: http.DefaultClient,
|
||||||
Port: uint(port),
|
Port: uint(port),
|
||||||
|
@ -153,11 +153,19 @@ func TestDoRequestNewWayObj(t *testing.T) {
|
|||||||
func TestDoRequestNewWayFile(t *testing.T) {
|
func TestDoRequestNewWayFile(t *testing.T) {
|
||||||
reqObj := &api.Pod{JSONBase: api.JSONBase{ID: "foo"}}
|
reqObj := &api.Pod{JSONBase: api.JSONBase{ID: "foo"}}
|
||||||
reqBodyExpected, err := api.Encode(reqObj)
|
reqBodyExpected, err := api.Encode(reqObj)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
file, err := ioutil.TempFile("", "foo")
|
file, err := ioutil.TempFile("", "foo")
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
_, err = file.Write(reqBodyExpected)
|
_, err = file.Write(reqBodyExpected)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
expectedObj := &api.Service{Port: 12345}
|
expectedObj := &api.Service{Port: 12345}
|
||||||
expectedBody, _ := api.Encode(expectedObj)
|
expectedBody, _ := api.Encode(expectedObj)
|
||||||
|
@ -32,25 +32,6 @@ import (
|
|||||||
"github.com/stretchr/testify/mock"
|
"github.com/stretchr/testify/mock"
|
||||||
)
|
)
|
||||||
|
|
||||||
// TODO: This doesn't reduce typing enough to make it worth the less readable errors. Remove.
|
|
||||||
func expectNoError(t *testing.T, err error) {
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("Unexpected error: %#v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func verifyNoError(t *testing.T, e error) {
|
|
||||||
if e != nil {
|
|
||||||
t.Errorf("Expected no error, found %#v", e)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func verifyError(t *testing.T, e error) {
|
|
||||||
if e == nil {
|
|
||||||
t.Errorf("Expected error, found nil")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func makeTestKubelet(t *testing.T) (*Kubelet, *tools.FakeEtcdClient, *FakeDockerClient) {
|
func makeTestKubelet(t *testing.T) (*Kubelet, *tools.FakeEtcdClient, *FakeDockerClient) {
|
||||||
fakeEtcdClient := tools.MakeFakeEtcdClient(t)
|
fakeEtcdClient := tools.MakeFakeEtcdClient(t)
|
||||||
fakeDocker := &FakeDockerClient{
|
fakeDocker := &FakeDockerClient{
|
||||||
@ -166,7 +147,9 @@ func TestKillContainerWithError(t *testing.T) {
|
|||||||
kubelet, _, _ := makeTestKubelet(t)
|
kubelet, _, _ := makeTestKubelet(t)
|
||||||
kubelet.dockerClient = fakeDocker
|
kubelet.dockerClient = fakeDocker
|
||||||
err := kubelet.killContainer(fakeDocker.containerList[0])
|
err := kubelet.killContainer(fakeDocker.containerList[0])
|
||||||
verifyError(t, err)
|
if err == nil {
|
||||||
|
t.Errorf("expected error, found nil")
|
||||||
|
}
|
||||||
verifyCalls(t, fakeDocker, []string{"stop"})
|
verifyCalls(t, fakeDocker, []string{"stop"})
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -187,7 +170,9 @@ func TestKillContainer(t *testing.T) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
err := kubelet.killContainer(fakeDocker.containerList[0])
|
err := kubelet.killContainer(fakeDocker.containerList[0])
|
||||||
verifyNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
verifyCalls(t, fakeDocker, []string{"stop"})
|
verifyCalls(t, fakeDocker, []string{"stop"})
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -246,7 +231,10 @@ func TestSyncPodsDoesNothing(t *testing.T) {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
verifyCalls(t, fakeDocker, []string{"list", "list"})
|
verifyCalls(t, fakeDocker, []string{"list", "list"})
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -269,7 +257,10 @@ func TestSyncPodsDeletes(t *testing.T) {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
err := kubelet.SyncPods([]Pod{})
|
err := kubelet.SyncPods([]Pod{})
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
verifyCalls(t, fakeDocker, []string{"list", "list", "stop", "stop"})
|
verifyCalls(t, fakeDocker, []string{"list", "list", "stop", "stop"})
|
||||||
|
|
||||||
// A map iteration is used to delete containers, so must not depend on
|
// A map iteration is used to delete containers, so must not depend on
|
||||||
@ -319,7 +310,10 @@ func TestSyncPodDeletesDuplicate(t *testing.T) {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
}, dockerContainers)
|
}, dockerContainers)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
verifyCalls(t, fakeDocker, []string{"list", "stop"})
|
verifyCalls(t, fakeDocker, []string{"list", "stop"})
|
||||||
|
|
||||||
// Expect one of the duplicates to be killed.
|
// Expect one of the duplicates to be killed.
|
||||||
@ -364,7 +358,10 @@ func TestSyncPodUnhealthy(t *testing.T) {
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
}, dockerContainers)
|
}, dockerContainers)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
verifyCalls(t, fakeDocker, []string{"list", "stop", "create", "start"})
|
verifyCalls(t, fakeDocker, []string{"list", "stop", "create", "start"})
|
||||||
|
|
||||||
// A map interation is used to delete containers, so must not depend on
|
// A map interation is used to delete containers, so must not depend on
|
||||||
@ -387,15 +384,24 @@ func TestEventWriting(t *testing.T) {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
err := kubelet.LogEvent(&expectedEvent)
|
err := kubelet.LogEvent(&expectedEvent)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
if fakeEtcd.Ix != 1 {
|
if fakeEtcd.Ix != 1 {
|
||||||
t.Errorf("Unexpected number of children added: %d, expected 1", fakeEtcd.Ix)
|
t.Errorf("Unexpected number of children added: %d, expected 1", fakeEtcd.Ix)
|
||||||
}
|
}
|
||||||
response, err := fakeEtcd.Get("/events/foo/1", false, false)
|
response, err := fakeEtcd.Get("/events/foo/1", false, false)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
var event api.Event
|
var event api.Event
|
||||||
err = json.Unmarshal([]byte(response.Node.Value), &event)
|
err = json.Unmarshal([]byte(response.Node.Value), &event)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
if event.Event != expectedEvent.Event ||
|
if event.Event != expectedEvent.Event ||
|
||||||
event.Container.Name != expectedEvent.Container.Name {
|
event.Container.Name != expectedEvent.Container.Name {
|
||||||
t.Errorf("Event's don't match. Expected: %#v Saw: %#v", expectedEvent, event)
|
t.Errorf("Event's don't match. Expected: %#v Saw: %#v", expectedEvent, event)
|
||||||
|
@ -23,12 +23,6 @@ import (
|
|||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
func expectNoError(t *testing.T, err error) {
|
|
||||||
if err != nil {
|
|
||||||
t.Errorf("Unexpected error: %#v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestFakeHandlerPath(t *testing.T) {
|
func TestFakeHandlerPath(t *testing.T) {
|
||||||
handler := FakeHandler{}
|
handler := FakeHandler{}
|
||||||
server := httptest.NewServer(&handler)
|
server := httptest.NewServer(&handler)
|
||||||
@ -37,10 +31,15 @@ func TestFakeHandlerPath(t *testing.T) {
|
|||||||
body := "somebody"
|
body := "somebody"
|
||||||
|
|
||||||
req, err := http.NewRequest(method, server.URL+path, bytes.NewBufferString(body))
|
req, err := http.NewRequest(method, server.URL+path, bytes.NewBufferString(body))
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
client := http.Client{}
|
client := http.Client{}
|
||||||
_, err = client.Do(req)
|
_, err = client.Do(req)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
handler.ValidateRequest(t, path, method, &body)
|
handler.ValidateRequest(t, path, method, &body)
|
||||||
}
|
}
|
||||||
@ -52,10 +51,15 @@ func TestFakeHandlerPathNoBody(t *testing.T) {
|
|||||||
path := "/foo/bar"
|
path := "/foo/bar"
|
||||||
|
|
||||||
req, err := http.NewRequest(method, server.URL+path, nil)
|
req, err := http.NewRequest(method, server.URL+path, nil)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
client := http.Client{}
|
client := http.Client{}
|
||||||
_, err = client.Do(req)
|
_, err = client.Do(req)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
handler.ValidateRequest(t, path, method, nil)
|
handler.ValidateRequest(t, path, method, nil)
|
||||||
}
|
}
|
||||||
@ -76,10 +80,15 @@ func TestFakeHandlerWrongPath(t *testing.T) {
|
|||||||
fakeT := fakeError{}
|
fakeT := fakeError{}
|
||||||
|
|
||||||
req, err := http.NewRequest(method, server.URL+"/foo/baz", nil)
|
req, err := http.NewRequest(method, server.URL+"/foo/baz", nil)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
client := http.Client{}
|
client := http.Client{}
|
||||||
_, err = client.Do(req)
|
_, err = client.Do(req)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
handler.ValidateRequest(&fakeT, path, method, nil)
|
handler.ValidateRequest(&fakeT, path, method, nil)
|
||||||
if len(fakeT.errors) != 1 {
|
if len(fakeT.errors) != 1 {
|
||||||
@ -95,10 +104,15 @@ func TestFakeHandlerWrongMethod(t *testing.T) {
|
|||||||
fakeT := fakeError{}
|
fakeT := fakeError{}
|
||||||
|
|
||||||
req, err := http.NewRequest("PUT", server.URL+path, nil)
|
req, err := http.NewRequest("PUT", server.URL+path, nil)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
client := http.Client{}
|
client := http.Client{}
|
||||||
_, err = client.Do(req)
|
_, err = client.Do(req)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
handler.ValidateRequest(&fakeT, path, method, nil)
|
handler.ValidateRequest(&fakeT, path, method, nil)
|
||||||
if len(fakeT.errors) != 1 {
|
if len(fakeT.errors) != 1 {
|
||||||
@ -115,10 +129,15 @@ func TestFakeHandlerWrongBody(t *testing.T) {
|
|||||||
fakeT := fakeError{}
|
fakeT := fakeError{}
|
||||||
|
|
||||||
req, err := http.NewRequest(method, server.URL+path, bytes.NewBufferString(body))
|
req, err := http.NewRequest(method, server.URL+path, bytes.NewBufferString(body))
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
client := http.Client{}
|
client := http.Client{}
|
||||||
_, err = client.Do(req)
|
_, err = client.Do(req)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
otherbody := "otherbody"
|
otherbody := "otherbody"
|
||||||
handler.ValidateRequest(&fakeT, path, method, &otherbody)
|
handler.ValidateRequest(&fakeT, path, method, &otherbody)
|
||||||
@ -136,10 +155,15 @@ func TestFakeHandlerNilBody(t *testing.T) {
|
|||||||
fakeT := fakeError{}
|
fakeT := fakeError{}
|
||||||
|
|
||||||
req, err := http.NewRequest(method, server.URL+path, nil)
|
req, err := http.NewRequest(method, server.URL+path, nil)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
client := http.Client{}
|
client := http.Client{}
|
||||||
_, err = client.Do(req)
|
_, err = client.Do(req)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
handler.ValidateRequest(&fakeT, path, method, &body)
|
handler.ValidateRequest(&fakeT, path, method, &body)
|
||||||
if len(fakeT.errors) != 1 {
|
if len(fakeT.errors) != 1 {
|
||||||
|
@ -47,7 +47,10 @@ func TestMakeJSONString(t *testing.T) {
|
|||||||
body := MakeJSONString(pod)
|
body := MakeJSONString(pod)
|
||||||
|
|
||||||
expectedBody, err := json.Marshal(pod)
|
expectedBody, err := json.Marshal(pod)
|
||||||
expectNoError(t, err)
|
if err != nil {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
if string(expectedBody) != body {
|
if string(expectedBody) != body {
|
||||||
t.Errorf("JSON doesn't match. Expected %s, saw %s", expectedBody, body)
|
t.Errorf("JSON doesn't match. Expected %s, saw %s", expectedBody, body)
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user