mirror of
https://github.com/openfaas/faas.git
synced 2025-06-09 00:36:46 +00:00
8e1c34e222
Sets a new default maximum scale limit of 5 replicas out of the box for CE users, CE meaning "Community" rather than "Commercial". The increment factor of 10 vs 25 should not make a difference to genuine community and hobbyist users. Tested and verified with unit tests and hey with a CE cluster where the maximum limit was reached over several minutes, finally going back to 1 replica. Signed-off-by: Alex Ellis (OpenFaaS Ltd) <alexellis2@gmail.com>
139 lines
3.4 KiB
Go
139 lines
3.4 KiB
Go
package plugin
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"strings"
|
|
"testing"
|
|
|
|
middleware "github.com/openfaas/faas/gateway/pkg/middleware"
|
|
"github.com/openfaas/faas/gateway/scaling"
|
|
)
|
|
|
|
const fallbackValue = 120
|
|
|
|
func TestLabelValueWasEmpty(t *testing.T) {
|
|
extractedValue := extractLabelValue("", fallbackValue)
|
|
|
|
if extractedValue != fallbackValue {
|
|
t.Log("Expected extractedValue to equal the fallbackValue")
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestLabelValueWasValid(t *testing.T) {
|
|
extractedValue := extractLabelValue("42", fallbackValue)
|
|
|
|
if extractedValue != 42 {
|
|
t.Log("Expected extractedValue to equal answer to life (42)")
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestLabelValueWasInValid(t *testing.T) {
|
|
extractedValue := extractLabelValue("InvalidValue", fallbackValue)
|
|
|
|
if extractedValue != fallbackValue {
|
|
t.Log("Expected extractedValue to equal the fallbackValue")
|
|
t.Fail()
|
|
}
|
|
}
|
|
func TestGetReplicasNonExistentFn(t *testing.T) {
|
|
|
|
testServer := httptest.NewServer(
|
|
http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
|
|
res.WriteHeader(http.StatusNotFound)
|
|
}))
|
|
defer testServer.Close()
|
|
|
|
var injector middleware.AuthInjector
|
|
url, _ := url.Parse(testServer.URL + "/")
|
|
|
|
esq := NewExternalServiceQuery(*url, injector)
|
|
|
|
svcQryResp, err := esq.GetReplicas("figlet", "")
|
|
|
|
if err == nil {
|
|
t.Logf("Error was nil, expected non-nil - the service query response value was %+v ", svcQryResp)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestGetReplicasExistentFn(t *testing.T) {
|
|
|
|
testServer := httptest.NewServer(
|
|
http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
|
|
res.WriteHeader(http.StatusOK)
|
|
res.Write([]byte(`{"json":"body"}`))
|
|
}))
|
|
defer testServer.Close()
|
|
|
|
expectedSvcQryResp := scaling.ServiceQueryResponse{
|
|
Replicas: 0,
|
|
MaxReplicas: uint64(scaling.DefaultMaxReplicas),
|
|
MinReplicas: uint64(scaling.DefaultMinReplicas),
|
|
ScalingFactor: uint64(scaling.DefaultScalingFactor),
|
|
AvailableReplicas: 0,
|
|
}
|
|
|
|
var injector middleware.AuthInjector
|
|
url, _ := url.Parse(testServer.URL + "/")
|
|
|
|
esq := NewExternalServiceQuery(*url, injector)
|
|
|
|
svcQryResp, err := esq.GetReplicas("figlet", "")
|
|
|
|
if err != nil {
|
|
t.Logf("Expected err to be nil got: %s ", err.Error())
|
|
t.Fail()
|
|
}
|
|
if svcQryResp != expectedSvcQryResp {
|
|
t.Logf("Unexpected return values - wanted\n%+v\ngot\n%+v ", expectedSvcQryResp, svcQryResp)
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSetReplicasNonExistentFn(t *testing.T) {
|
|
|
|
testServer := httptest.NewServer(
|
|
http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
|
|
res.WriteHeader(http.StatusInternalServerError)
|
|
}))
|
|
defer testServer.Close()
|
|
|
|
var injector middleware.AuthInjector
|
|
url, _ := url.Parse(testServer.URL + "/")
|
|
esq := NewExternalServiceQuery(*url, injector)
|
|
|
|
err := esq.SetReplicas("figlet", "", 1)
|
|
|
|
expectedErrStr := "error scaling HTTP code 500"
|
|
|
|
if !strings.Contains(err.Error(), expectedErrStr) {
|
|
t.Logf("Wanted string containing %s, got %s", expectedErrStr, err.Error())
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestSetReplicasExistentFn(t *testing.T) {
|
|
|
|
testServer := httptest.NewServer(
|
|
http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
|
|
res.WriteHeader(http.StatusOK)
|
|
}))
|
|
defer testServer.Close()
|
|
|
|
var injector middleware.AuthInjector
|
|
|
|
url, _ := url.Parse(testServer.URL + "/")
|
|
esq := NewExternalServiceQuery(*url, injector)
|
|
|
|
err := esq.SetReplicas("figlet", "", 1)
|
|
|
|
if err != nil {
|
|
t.Logf("Expected err to be nil got: %s ", err.Error())
|
|
t.Fail()
|
|
}
|
|
}
|