You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
295 lines
8.1 KiB
295 lines
8.1 KiB
package elastic |
|
|
|
import ( |
|
"bufio" |
|
"fmt" |
|
"net/http" |
|
"strings" |
|
"testing" |
|
) |
|
|
|
func TestResponseError(t *testing.T) { |
|
raw := "HTTP/1.1 404 Not Found\r\n" + |
|
"\r\n" + |
|
`{"error":{"root_cause":[{"type":"index_missing_exception","reason":"no such index","index":"elastic-test"}],"type":"index_missing_exception","reason":"no such index","index":"elastic-test"},"status":404}` + "\r\n" |
|
r := bufio.NewReader(strings.NewReader(raw)) |
|
|
|
req, err := http.NewRequest("GET", "/", nil) |
|
if err != nil { |
|
t.Fatal(err) |
|
} |
|
|
|
resp, err := http.ReadResponse(r, nil) |
|
if err != nil { |
|
t.Fatal(err) |
|
} |
|
err = checkResponse(req, resp) |
|
if err == nil { |
|
t.Fatalf("expected error; got: %v", err) |
|
} |
|
|
|
// Check for correct error message |
|
expected := fmt.Sprintf("elastic: Error %d (%s): no such index [type=index_missing_exception]", resp.StatusCode, http.StatusText(resp.StatusCode)) |
|
got := err.Error() |
|
if got != expected { |
|
t.Fatalf("expected %q; got: %q", expected, got) |
|
} |
|
|
|
// Check that error is of type *elastic.Error, which contains additional information |
|
e, ok := err.(*Error) |
|
if !ok { |
|
t.Fatal("expected error to be of type *elastic.Error") |
|
} |
|
if e.Status != resp.StatusCode { |
|
t.Fatalf("expected status code %d; got: %d", resp.StatusCode, e.Status) |
|
} |
|
if e.Details == nil { |
|
t.Fatalf("expected error details; got: %v", e.Details) |
|
} |
|
if got, want := e.Details.Index, "elastic-test"; got != want { |
|
t.Fatalf("expected error details index %q; got: %q", want, got) |
|
} |
|
if got, want := e.Details.Type, "index_missing_exception"; got != want { |
|
t.Fatalf("expected error details type %q; got: %q", want, got) |
|
} |
|
if got, want := e.Details.Reason, "no such index"; got != want { |
|
t.Fatalf("expected error details reason %q; got: %q", want, got) |
|
} |
|
if got, want := len(e.Details.RootCause), 1; got != want { |
|
t.Fatalf("expected %d error details root causes; got: %d", want, got) |
|
} |
|
|
|
if got, want := e.Details.RootCause[0].Index, "elastic-test"; got != want { |
|
t.Fatalf("expected root cause index %q; got: %q", want, got) |
|
} |
|
if got, want := e.Details.RootCause[0].Type, "index_missing_exception"; got != want { |
|
t.Fatalf("expected root cause type %q; got: %q", want, got) |
|
} |
|
if got, want := e.Details.RootCause[0].Reason, "no such index"; got != want { |
|
t.Fatalf("expected root cause reason %q; got: %q", want, got) |
|
} |
|
} |
|
|
|
func TestResponseErrorHTML(t *testing.T) { |
|
raw := "HTTP/1.1 413 Request Entity Too Large\r\n" + |
|
"\r\n" + |
|
`<html> |
|
<head><title>413 Request Entity Too Large</title></head> |
|
<body bgcolor="white"> |
|
<center><h1>413 Request Entity Too Large</h1></center> |
|
<hr><center>nginx/1.6.2</center> |
|
</body> |
|
</html>` + "\r\n" |
|
r := bufio.NewReader(strings.NewReader(raw)) |
|
|
|
req, err := http.NewRequest("GET", "/", nil) |
|
if err != nil { |
|
t.Fatal(err) |
|
} |
|
|
|
resp, err := http.ReadResponse(r, nil) |
|
if err != nil { |
|
t.Fatal(err) |
|
} |
|
err = checkResponse(req, resp) |
|
if err == nil { |
|
t.Fatalf("expected error; got: %v", err) |
|
} |
|
|
|
// Check for correct error message |
|
expected := fmt.Sprintf("elastic: Error %d (%s)", http.StatusRequestEntityTooLarge, http.StatusText(http.StatusRequestEntityTooLarge)) |
|
got := err.Error() |
|
if got != expected { |
|
t.Fatalf("expected %q; got: %q", expected, got) |
|
} |
|
} |
|
|
|
func TestResponseErrorWithIgnore(t *testing.T) { |
|
raw := "HTTP/1.1 404 Not Found\r\n" + |
|
"\r\n" + |
|
`{"some":"response"}` + "\r\n" |
|
r := bufio.NewReader(strings.NewReader(raw)) |
|
|
|
req, err := http.NewRequest("HEAD", "/", nil) |
|
if err != nil { |
|
t.Fatal(err) |
|
} |
|
|
|
resp, err := http.ReadResponse(r, nil) |
|
if err != nil { |
|
t.Fatal(err) |
|
} |
|
err = checkResponse(req, resp) |
|
if err == nil { |
|
t.Fatalf("expected error; got: %v", err) |
|
} |
|
err = checkResponse(req, resp, 404) // ignore 404 errors |
|
if err != nil { |
|
t.Fatalf("expected no error; got: %v", err) |
|
} |
|
} |
|
|
|
func TestIsNotFound(t *testing.T) { |
|
if got, want := IsNotFound(nil), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsNotFound(""), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsNotFound(200), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsNotFound(404), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
|
|
if got, want := IsNotFound(&Error{Status: 404}), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsNotFound(&Error{Status: 200}), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
|
|
if got, want := IsNotFound(Error{Status: 404}), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsNotFound(Error{Status: 200}), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
|
|
if got, want := IsNotFound(&http.Response{StatusCode: 404}), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsNotFound(&http.Response{StatusCode: 200}), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
} |
|
|
|
func TestIsTimeout(t *testing.T) { |
|
if got, want := IsTimeout(nil), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsTimeout(""), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsTimeout(200), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsTimeout(408), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
|
|
if got, want := IsTimeout(&Error{Status: 408}), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsTimeout(&Error{Status: 200}), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
|
|
if got, want := IsTimeout(Error{Status: 408}), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsTimeout(Error{Status: 200}), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
|
|
if got, want := IsTimeout(&http.Response{StatusCode: 408}), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsTimeout(&http.Response{StatusCode: 200}), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
} |
|
|
|
func TestIsConflict(t *testing.T) { |
|
if got, want := IsConflict(nil), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsConflict(""), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsConflict(200), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsConflict(http.StatusConflict), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
|
|
if got, want := IsConflict(&Error{Status: 409}), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsConflict(&Error{Status: 200}), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
|
|
if got, want := IsConflict(Error{Status: 409}), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsConflict(Error{Status: 200}), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
|
|
if got, want := IsConflict(&http.Response{StatusCode: 409}), true; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
if got, want := IsConflict(&http.Response{StatusCode: 200}), false; got != want { |
|
t.Errorf("expected %v; got: %v", want, got) |
|
} |
|
} |
|
|
|
func TestIsStatusCode(t *testing.T) { |
|
tests := []struct { |
|
Error interface{} |
|
Code int |
|
Want bool |
|
}{ |
|
// #0 |
|
{ |
|
Error: nil, |
|
Code: 200, |
|
Want: false, |
|
}, |
|
// #1 |
|
{ |
|
Error: "", |
|
Code: 200, |
|
Want: false, |
|
}, |
|
// #2 |
|
{ |
|
Error: http.StatusConflict, |
|
Code: 409, |
|
Want: true, |
|
}, |
|
// #3 |
|
{ |
|
Error: http.StatusConflict, |
|
Code: http.StatusInternalServerError, |
|
Want: false, |
|
}, |
|
// #4 |
|
{ |
|
Error: &Error{Status: http.StatusConflict}, |
|
Code: 409, |
|
Want: true, |
|
}, |
|
// #5 |
|
{ |
|
Error: Error{Status: http.StatusConflict}, |
|
Code: 409, |
|
Want: true, |
|
}, |
|
// #6 |
|
{ |
|
Error: &http.Response{StatusCode: http.StatusConflict}, |
|
Code: 409, |
|
Want: true, |
|
}, |
|
} |
|
|
|
for i, tt := range tests { |
|
if have, want := IsStatusCode(tt.Error, tt.Code), tt.Want; have != want { |
|
t.Errorf("#%d: have %v, want %v", i, have, want) |
|
} |
|
} |
|
}
|
|
|