mirror of
https://github.com/openfaas/faas.git
synced 2025-06-10 01:06:47 +00:00
Get is a supported method so this test failed, I've now changed this to an unsupported method - i.e. HEAD to trigger the error and keep the test. Signed-off-by: Alex Ellis (VMware) <alexellis2@gmail.com>
123 lines
2.7 KiB
Go
123 lines
2.7 KiB
Go
// Copyright (c) Alex Ellis 2017. All rights reserved.
|
|
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
|
|
|
package inttests
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// Before running these tests do a Docker stack deploy.
|
|
|
|
func fireRequest(url string, method string, reqBody string) (string, int, error) {
|
|
headers := make(map[string]string)
|
|
return fireRequestWithHeaders(url, method, reqBody, headers)
|
|
}
|
|
|
|
func fireRequestWithHeaders(url string, method string, reqBody string, headers map[string]string) (string, int, error) {
|
|
httpClient := http.Client{
|
|
Timeout: time.Second * 2, // Maximum of 2 secs
|
|
}
|
|
|
|
req, err := http.NewRequest(method, url, bytes.NewBufferString(reqBody))
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
req.Header.Set("User-Agent", "go-integration")
|
|
for kk, vv := range headers {
|
|
req.Header.Set(kk, vv)
|
|
}
|
|
|
|
res, getErr := httpClient.Do(req)
|
|
if getErr != nil {
|
|
log.Fatal(getErr)
|
|
}
|
|
|
|
body, readErr := ioutil.ReadAll(res.Body)
|
|
defer req.Body.Close()
|
|
if readErr != nil {
|
|
log.Fatal(readErr)
|
|
}
|
|
return string(body), res.StatusCode, readErr
|
|
}
|
|
|
|
func TestGet_Rejected(t *testing.T) {
|
|
var reqBody string
|
|
unsupportedMethod := http.MethodHead
|
|
_, code, err := fireRequest("http://localhost:8080/function/func_echoit", unsupportedMethod, reqBody)
|
|
want := http.StatusMethodNotAllowed
|
|
if code != want {
|
|
t.Logf("Failed got: %d, wanted: %d", code, want)
|
|
t.Fail()
|
|
}
|
|
|
|
if err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestEchoIt_Post_Route_Handler_ForwardsClientHeaders(t *testing.T) {
|
|
reqBody := "test message"
|
|
headers := make(map[string]string, 0)
|
|
headers["X-Api-Key"] = "123"
|
|
|
|
body, code, err := fireRequestWithHeaders("http://localhost:8080/function/func_echoit", http.MethodPost, reqBody, headers)
|
|
|
|
if err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
if code != http.StatusOK {
|
|
t.Log("Failed")
|
|
}
|
|
if body != reqBody {
|
|
t.Log("Expected body returned")
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestEchoIt_Post_Route_Handler(t *testing.T) {
|
|
reqBody := "test message"
|
|
body, code, err := fireRequest("http://localhost:8080/function/func_echoit", http.MethodPost, reqBody)
|
|
|
|
if err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
if code != http.StatusOK {
|
|
t.Log("Failed")
|
|
}
|
|
if body != reqBody {
|
|
t.Log("Expected body returned")
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestEchoIt_Post_X_Header_Routing_Handler(t *testing.T) {
|
|
reqBody := "test message"
|
|
headers := make(map[string]string, 0)
|
|
headers["X-Function"] = "func_echoit"
|
|
|
|
body, code, err := fireRequestWithHeaders("http://localhost:8080/", http.MethodPost, reqBody, headers)
|
|
|
|
if err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
if code != http.StatusOK {
|
|
t.Log("Failed")
|
|
}
|
|
if body != reqBody {
|
|
t.Log("Expected body returned")
|
|
t.Fail()
|
|
}
|
|
|
|
}
|