github.com/SmartMeshFoundation/Spectrum@v0.0.0-20220621030607-452a266fee1e/rpc/http_test.go (about)

     1  // Copyright 2017 The Spectrum Authors
     2  // This file is part of the Spectrum library.
     3  //
     4  // The Spectrum 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 Spectrum 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 Spectrum 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, maxHTTPRequestContentLength+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  }