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  }