tangmz
/
openbilibili


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295
							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)
		}
	}
}