etcdserver: address style comments

This commit is contained in:
Jonathan Boulle 2014-09-10 14:14:14 -07:00
parent bed63cddf7
commit 2342402434
2 changed files with 27 additions and 52 deletions

View File

@ -195,8 +195,7 @@ func (h Handler) serveKeys(ctx context.Context, w http.ResponseWriter, r *http.R
case resp.Event != nil: case resp.Event != nil:
ev = resp.Event ev = resp.Event
case resp.Watcher != nil: case resp.Watcher != nil:
ev, err = waitForEvent(ctx, w, resp.Watcher) if ev, err = waitForEvent(ctx, w, resp.Watcher); err != nil {
if err != nil {
http.Error(w, err.Error(), http.StatusGatewayTimeout) http.Error(w, err.Error(), http.StatusGatewayTimeout)
return return
} }

View File

@ -26,6 +26,8 @@ func mustNewURL(t *testing.T, s string) *url.URL {
return u return u
} }
// mustNewRequest takes a path, appends it to the standard keysPrefix, and constructs
// an *http.Request referencing the resulting URL
func mustNewRequest(t *testing.T, p string) *http.Request { func mustNewRequest(t *testing.T, p string) *http.Request {
return &http.Request{ return &http.Request{
URL: mustNewURL(t, path.Join(keysPrefix, p)), URL: mustNewURL(t, path.Join(keysPrefix, p)),
@ -38,7 +40,7 @@ func TestParseBool(t *testing.T) {
t.Fatalf("got %t, want %t", got, false) t.Fatalf("got %t, want %t", got, false)
} }
if err != nil { if err != nil {
t.Fatalf("got err=%v, want err=%v", err, nil) t.Fatalf("err = %v, want %v", err, nil)
} }
} }
@ -48,7 +50,7 @@ func TestParseUint64(t *testing.T) {
t.Fatalf("got %d, want %d", got, 0) t.Fatalf("got %d, want %d", got, 0)
} }
if err != nil { if err != nil {
t.Fatalf("got err=%v, want err=%v", err, nil) t.Fatalf("err = %v, want %v", err, nil)
} }
} }
@ -70,46 +72,20 @@ func TestBadParseRequest(t *testing.T) {
}, },
}, },
// bad values for prevIndex, waitIndex, ttl // bad values for prevIndex, waitIndex, ttl
{ {mustNewRequest(t, "?prevIndex=foo")},
mustNewRequest(t, "?prevIndex=foo"), {mustNewRequest(t, "?prevIndex=1.5")},
}, {mustNewRequest(t, "?prevIndex=-1")},
{ {mustNewRequest(t, "?waitIndex=garbage")},
mustNewRequest(t, "?prevIndex=1.5"), {mustNewRequest(t, "?waitIndex=??")},
}, {mustNewRequest(t, "?ttl=-1")},
{
mustNewRequest(t, "?prevIndex=-1"),
},
{
mustNewRequest(t, "?waitIndex=garbage"),
},
{
mustNewRequest(t, "?waitIndex=??"),
},
{
mustNewRequest(t, "?ttl=-1"),
},
// bad values for recursive, sorted, wait // bad values for recursive, sorted, wait
{ {mustNewRequest(t, "?recursive=hahaha")},
mustNewRequest(t, "?recursive=hahaha"), {mustNewRequest(t, "?recursive=1234")},
}, {mustNewRequest(t, "?recursive=?")},
{ {mustNewRequest(t, "?sorted=hahaha")},
mustNewRequest(t, "?recursive=1234"), {mustNewRequest(t, "?sorted=!!")},
}, {mustNewRequest(t, "?wait=notreally")},
{ {mustNewRequest(t, "?wait=what!")},
mustNewRequest(t, "?recursive=?"),
},
{
mustNewRequest(t, "?sorted=hahaha"),
},
{
mustNewRequest(t, "?sorted=!!"),
},
{
mustNewRequest(t, "?wait=notreally"),
},
{
mustNewRequest(t, "?wait=what!"),
},
} }
for i, tt := range tests { for i, tt := range tests {
got, err := parseRequest(tt.in, 1234) got, err := parseRequest(tt.in, 1234)
@ -244,8 +220,8 @@ func TestWriteInternalError(t *testing.T) {
tests := []struct { tests := []struct {
err error err error
code int wcode int
idx string wi string
}{ }{
{ {
etcdErr.NewError(etcdErr.EcodeKeyNotFound, "/foo/bar", 123), etcdErr.NewError(etcdErr.EcodeKeyNotFound, "/foo/bar", 123),
@ -259,18 +235,18 @@ func TestWriteInternalError(t *testing.T) {
}, },
{ {
err: errors.New("something went wrong"), err: errors.New("something went wrong"),
code: http.StatusInternalServerError, wcode: http.StatusInternalServerError,
}, },
} }
for i, tt := range tests { for i, tt := range tests {
rw := httptest.NewRecorder() rw := httptest.NewRecorder()
writeInternalError(rw, tt.err) writeInternalError(rw, tt.err)
if code := rw.Code; code != tt.code { if code := rw.Code; code != tt.wcode {
t.Errorf("#%d: got %d, want %d", i, code, tt.code) t.Errorf("#%d: code=%d, want %d", i, code, tt.wcode)
} }
if idx := rw.Header().Get("X-Etcd-Index"); idx != tt.idx { if idx := rw.Header().Get("X-Etcd-Index"); idx != tt.wi {
t.Errorf("#%d: got %q, want %q", i, idx, tt.idx) t.Errorf("#%d: X-Etcd-Index=%q, want %q", i, idx, tt.wi)
} }
} }
} }