github.com/m3shine/gochain@v2.2.26+incompatible/rpc/http_test.go (about) 1 // Copyright 2017 The go-ethereum Authors 2 // This file is part of the go-ethereum library. 3 // 4 // The go-ethereum library is free software: you can redistribute it and/or modify 5 // it under the terms of the GNU Lesser General Public License as published by 6 // the Free Software Foundation, either version 3 of the License, or 7 // (at your option) any later version. 8 // 9 // The go-ethereum library is distributed in the hope that it will be useful, 10 // but WITHOUT ANY WARRANTY; without even the implied warranty of 11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 // GNU Lesser General Public License for more details. 13 // 14 // You should have received a copy of the GNU Lesser General Public License 15 // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>. 16 17 package rpc 18 19 import ( 20 "net/http" 21 "net/http/httptest" 22 "strings" 23 "testing" 24 ) 25 26 func TestHTTPErrorResponseWithDelete(t *testing.T) { 27 testHTTPErrorResponse(t, http.MethodDelete, contentType, "", http.StatusMethodNotAllowed) 28 } 29 30 func TestHTTPErrorResponseWithPut(t *testing.T) { 31 testHTTPErrorResponse(t, http.MethodPut, contentType, "", http.StatusMethodNotAllowed) 32 } 33 34 func TestHTTPErrorResponseWithMaxContentLength(t *testing.T) { 35 body := make([]rune, maxRequestContentLength+1) 36 testHTTPErrorResponse(t, 37 http.MethodPost, contentType, string(body), http.StatusRequestEntityTooLarge) 38 } 39 40 func TestHTTPErrorResponseWithEmptyContentType(t *testing.T) { 41 testHTTPErrorResponse(t, http.MethodPost, "", "", http.StatusUnsupportedMediaType) 42 } 43 44 func TestHTTPErrorResponseWithValidRequest(t *testing.T) { 45 testHTTPErrorResponse(t, http.MethodPost, contentType, "", 0) 46 } 47 48 func testHTTPErrorResponse(t *testing.T, method, contentType, body string, expected int) { 49 request := httptest.NewRequest(method, "http://url.com", strings.NewReader(body)) 50 request.Header.Set("content-type", contentType) 51 if code, _ := validateRequest(request); code != expected { 52 t.Fatalf("response code should be %d not %d", expected, code) 53 } 54 } 55 56 func TestVirtualHostHandler_validHost(t *testing.T) { 57 for _, test := range []struct { 58 name string 59 vhosts []string 60 valid []string 61 invalid []string 62 }{ 63 { 64 name: "any", 65 vhosts: []string{"*"}, 66 valid: []string{"google.com", "example.com", "gochain.io"}, 67 }, 68 { 69 name: "literal", 70 vhosts: []string{"test.gochain.io"}, 71 valid: []string{"test.gochain.io"}, 72 invalid: []string{"a.gochain.io", "a.test.gochain.io"}, 73 }, 74 { 75 name: "subdomain", 76 vhosts: []string{"*.gochain.io"}, 77 valid: []string{"test.gochain.io", "a.gochain.io", "a.b.gochain.io"}, 78 invalid: []string{"gochain.io"}, 79 }, 80 { 81 name: "multi", 82 vhosts: []string{"*.e.d.c.b.a"}, 83 valid: []string{"f.e.d.c.b.a"}, 84 invalid: []string{"e.d.c.b.a", "c.b.a", "b.a"}, 85 }, 86 } { 87 test := test 88 t.Run(test.name, func(t *testing.T) { 89 h := newVHostHandler(test.vhosts, nil) 90 for _, host := range test.valid { 91 if !h.validHost(host) { 92 t.Errorf("expected %q to be valid", host) 93 } 94 } 95 for _, host := range test.invalid { 96 if h.validHost(host) { 97 t.Errorf("expected %q to be invalid", host) 98 } 99 } 100 }) 101 } 102 }