mirror of
https://github.com/openfaas/faas.git
synced 2025-06-09 16:56:47 +00:00
Extend the health endpoint and add gateway version information Resolves: #733 Signed-off-by: Edward Wilde <ewilde@gmail.com>
126 lines
2.9 KiB
Go
126 lines
2.9 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/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/echoit", http.MethodPost, reqBody, headers)
|
|
|
|
if err != nil {
|
|
t.Log(err)
|
|
t.Fail()
|
|
}
|
|
|
|
if code != http.StatusOK {
|
|
t.Logf("Failed, code: %d, body:%s", code, body)
|
|
t.Fail()
|
|
}
|
|
|
|
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/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()
|
|
}
|
|
}
|
|
|
|
// Test suppressed due to X-Header deprecation.
|
|
// 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.Logf("statusCode - want: %d, got: %d", http.StatusOK, code)
|
|
// }
|
|
// if body != reqBody {
|
|
// t.Logf("Expected body from echo function to be equal to input, but was: %s", body)
|
|
// t.Fail()
|
|
// }
|
|
// }
|