mirror of
https://github.com/astaxie/beego.git
synced 2024-11-19 06:50:57 +00:00
250 lines
7.6 KiB
Go
250 lines
7.6 KiB
Go
package web
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/astaxie/beego/pkg/infrastructure/governor"
|
|
)
|
|
|
|
type SampleDatabaseCheck struct {
|
|
}
|
|
|
|
type SampleCacheCheck struct {
|
|
}
|
|
|
|
func (dc *SampleDatabaseCheck) Check() error {
|
|
return nil
|
|
}
|
|
|
|
func (cc *SampleCacheCheck) Check() error {
|
|
return errors.New("no cache detected")
|
|
}
|
|
|
|
func TestList_01(t *testing.T) {
|
|
m := make(M)
|
|
list("BConfig", BConfig, m)
|
|
t.Log(m)
|
|
om := oldMap()
|
|
for k, v := range om {
|
|
if fmt.Sprint(m[k]) != fmt.Sprint(v) {
|
|
t.Log(k, "old-key", v, "new-key", m[k])
|
|
t.FailNow()
|
|
}
|
|
}
|
|
}
|
|
|
|
func oldMap() M {
|
|
m := make(M)
|
|
m["BConfig.AppName"] = BConfig.AppName
|
|
m["BConfig.RunMode"] = BConfig.RunMode
|
|
m["BConfig.RouterCaseSensitive"] = BConfig.RouterCaseSensitive
|
|
m["BConfig.ServerName"] = BConfig.ServerName
|
|
m["BConfig.RecoverPanic"] = BConfig.RecoverPanic
|
|
m["BConfig.CopyRequestBody"] = BConfig.CopyRequestBody
|
|
m["BConfig.EnableGzip"] = BConfig.EnableGzip
|
|
m["BConfig.MaxMemory"] = BConfig.MaxMemory
|
|
m["BConfig.EnableErrorsShow"] = BConfig.EnableErrorsShow
|
|
m["BConfig.Listen.Graceful"] = BConfig.Listen.Graceful
|
|
m["BConfig.Listen.ServerTimeOut"] = BConfig.Listen.ServerTimeOut
|
|
m["BConfig.Listen.ListenTCP4"] = BConfig.Listen.ListenTCP4
|
|
m["BConfig.Listen.EnableHTTP"] = BConfig.Listen.EnableHTTP
|
|
m["BConfig.Listen.HTTPAddr"] = BConfig.Listen.HTTPAddr
|
|
m["BConfig.Listen.HTTPPort"] = BConfig.Listen.HTTPPort
|
|
m["BConfig.Listen.EnableHTTPS"] = BConfig.Listen.EnableHTTPS
|
|
m["BConfig.Listen.HTTPSAddr"] = BConfig.Listen.HTTPSAddr
|
|
m["BConfig.Listen.HTTPSPort"] = BConfig.Listen.HTTPSPort
|
|
m["BConfig.Listen.HTTPSCertFile"] = BConfig.Listen.HTTPSCertFile
|
|
m["BConfig.Listen.HTTPSKeyFile"] = BConfig.Listen.HTTPSKeyFile
|
|
m["BConfig.Listen.EnableAdmin"] = BConfig.Listen.EnableAdmin
|
|
m["BConfig.Listen.AdminAddr"] = BConfig.Listen.AdminAddr
|
|
m["BConfig.Listen.AdminPort"] = BConfig.Listen.AdminPort
|
|
m["BConfig.Listen.EnableFcgi"] = BConfig.Listen.EnableFcgi
|
|
m["BConfig.Listen.EnableStdIo"] = BConfig.Listen.EnableStdIo
|
|
m["BConfig.WebConfig.AutoRender"] = BConfig.WebConfig.AutoRender
|
|
m["BConfig.WebConfig.EnableDocs"] = BConfig.WebConfig.EnableDocs
|
|
m["BConfig.WebConfig.FlashName"] = BConfig.WebConfig.FlashName
|
|
m["BConfig.WebConfig.FlashSeparator"] = BConfig.WebConfig.FlashSeparator
|
|
m["BConfig.WebConfig.DirectoryIndex"] = BConfig.WebConfig.DirectoryIndex
|
|
m["BConfig.WebConfig.StaticDir"] = BConfig.WebConfig.StaticDir
|
|
m["BConfig.WebConfig.StaticExtensionsToGzip"] = BConfig.WebConfig.StaticExtensionsToGzip
|
|
m["BConfig.WebConfig.StaticCacheFileSize"] = BConfig.WebConfig.StaticCacheFileSize
|
|
m["BConfig.WebConfig.StaticCacheFileNum"] = BConfig.WebConfig.StaticCacheFileNum
|
|
m["BConfig.WebConfig.TemplateLeft"] = BConfig.WebConfig.TemplateLeft
|
|
m["BConfig.WebConfig.TemplateRight"] = BConfig.WebConfig.TemplateRight
|
|
m["BConfig.WebConfig.ViewsPath"] = BConfig.WebConfig.ViewsPath
|
|
m["BConfig.WebConfig.EnableXSRF"] = BConfig.WebConfig.EnableXSRF
|
|
m["BConfig.WebConfig.XSRFExpire"] = BConfig.WebConfig.XSRFExpire
|
|
m["BConfig.WebConfig.Session.SessionOn"] = BConfig.WebConfig.Session.SessionOn
|
|
m["BConfig.WebConfig.Session.SessionProvider"] = BConfig.WebConfig.Session.SessionProvider
|
|
m["BConfig.WebConfig.Session.SessionName"] = BConfig.WebConfig.Session.SessionName
|
|
m["BConfig.WebConfig.Session.SessionGCMaxLifetime"] = BConfig.WebConfig.Session.SessionGCMaxLifetime
|
|
m["BConfig.WebConfig.Session.SessionProviderConfig"] = BConfig.WebConfig.Session.SessionProviderConfig
|
|
m["BConfig.WebConfig.Session.SessionCookieLifeTime"] = BConfig.WebConfig.Session.SessionCookieLifeTime
|
|
m["BConfig.WebConfig.Session.SessionAutoSetCookie"] = BConfig.WebConfig.Session.SessionAutoSetCookie
|
|
m["BConfig.WebConfig.Session.SessionDomain"] = BConfig.WebConfig.Session.SessionDomain
|
|
m["BConfig.WebConfig.Session.SessionDisableHTTPOnly"] = BConfig.WebConfig.Session.SessionDisableHTTPOnly
|
|
m["BConfig.Log.AccessLogs"] = BConfig.Log.AccessLogs
|
|
m["BConfig.Log.EnableStaticLogs"] = BConfig.Log.EnableStaticLogs
|
|
m["BConfig.Log.AccessLogsFormat"] = BConfig.Log.AccessLogsFormat
|
|
m["BConfig.Log.FileLineNum"] = BConfig.Log.FileLineNum
|
|
m["BConfig.Log.Outputs"] = BConfig.Log.Outputs
|
|
return m
|
|
}
|
|
|
|
func TestWriteJSON(t *testing.T) {
|
|
t.Log("Testing the adding of JSON to the response")
|
|
|
|
w := httptest.NewRecorder()
|
|
originalBody := []int{1, 2, 3}
|
|
|
|
res, _ := json.Marshal(originalBody)
|
|
|
|
writeJSON(w, res)
|
|
|
|
decodedBody := []int{}
|
|
err := json.NewDecoder(w.Body).Decode(&decodedBody)
|
|
|
|
if err != nil {
|
|
t.Fatal("Could not decode response body into slice.")
|
|
}
|
|
|
|
for i := range decodedBody {
|
|
if decodedBody[i] != originalBody[i] {
|
|
t.Fatalf("Expected %d but got %d in decoded body slice", originalBody[i], decodedBody[i])
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestHealthCheckHandlerDefault(t *testing.T) {
|
|
endpointPath := "/healthcheck"
|
|
|
|
governor.AddHealthCheck("database", &SampleDatabaseCheck{})
|
|
governor.AddHealthCheck("cache", &SampleCacheCheck{})
|
|
|
|
req, err := http.NewRequest("GET", endpointPath, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
w := httptest.NewRecorder()
|
|
|
|
handler := http.HandlerFunc(healthcheck)
|
|
|
|
handler.ServeHTTP(w, req)
|
|
|
|
if status := w.Code; status != http.StatusOK {
|
|
t.Errorf("handler returned wrong status code: got %v want %v",
|
|
status, http.StatusOK)
|
|
}
|
|
if !strings.Contains(w.Body.String(), "database") {
|
|
t.Errorf("Expected 'database' in generated template.")
|
|
}
|
|
|
|
}
|
|
|
|
func TestBuildHealthCheckResponseList(t *testing.T) {
|
|
healthCheckResults := [][]string{
|
|
[]string{
|
|
"error",
|
|
"Database",
|
|
"Error occured whie starting the db",
|
|
},
|
|
[]string{
|
|
"success",
|
|
"Cache",
|
|
"Cache started successfully",
|
|
},
|
|
}
|
|
|
|
responseList := buildHealthCheckResponseList(&healthCheckResults)
|
|
|
|
if len(responseList) != len(healthCheckResults) {
|
|
t.Errorf("invalid response map length: got %d want %d",
|
|
len(responseList), len(healthCheckResults))
|
|
}
|
|
|
|
responseFields := []string{"name", "message", "status"}
|
|
|
|
for _, response := range responseList {
|
|
for _, field := range responseFields {
|
|
_, ok := response[field]
|
|
if !ok {
|
|
t.Errorf("expected %s to be in the response %v", field, response)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func TestHealthCheckHandlerReturnsJSON(t *testing.T) {
|
|
|
|
governor.AddHealthCheck("database", &SampleDatabaseCheck{})
|
|
governor.AddHealthCheck("cache", &SampleCacheCheck{})
|
|
|
|
req, err := http.NewRequest("GET", "/healthcheck?json=true", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
w := httptest.NewRecorder()
|
|
|
|
handler := http.HandlerFunc(healthcheck)
|
|
|
|
handler.ServeHTTP(w, req)
|
|
if status := w.Code; status != http.StatusOK {
|
|
t.Errorf("handler returned wrong status code: got %v want %v",
|
|
status, http.StatusOK)
|
|
}
|
|
|
|
decodedResponseBody := []map[string]interface{}{}
|
|
expectedResponseBody := []map[string]interface{}{}
|
|
|
|
expectedJSONString := []byte(`
|
|
[
|
|
{
|
|
"message":"database",
|
|
"name":"success",
|
|
"status":"OK"
|
|
},
|
|
{
|
|
"message":"cache",
|
|
"name":"error",
|
|
"status":"no cache detected"
|
|
}
|
|
]
|
|
`)
|
|
|
|
json.Unmarshal(expectedJSONString, &expectedResponseBody)
|
|
|
|
json.Unmarshal(w.Body.Bytes(), &decodedResponseBody)
|
|
|
|
if len(expectedResponseBody) != len(decodedResponseBody) {
|
|
t.Errorf("invalid response map length: got %d want %d",
|
|
len(decodedResponseBody), len(expectedResponseBody))
|
|
}
|
|
assert.Equal(t, len(expectedResponseBody), len(decodedResponseBody))
|
|
assert.Equal(t, 2, len(decodedResponseBody))
|
|
|
|
var database, cache map[string]interface{}
|
|
if decodedResponseBody[0]["message"] == "database" {
|
|
database = decodedResponseBody[0]
|
|
cache = decodedResponseBody[1]
|
|
} else {
|
|
database = decodedResponseBody[1]
|
|
cache = decodedResponseBody[0]
|
|
}
|
|
|
|
assert.Equal(t, expectedResponseBody[0], database)
|
|
assert.Equal(t, expectedResponseBody[1], cache)
|
|
|
|
}
|