forked from goffee/core
initial commits 2
This commit is contained in:
parent
5475b7dd26
commit
7f38826b9c
39 changed files with 4525 additions and 0 deletions
525
core_test.go
Normal file
525
core_test.go
Normal file
|
|
@ -0,0 +1,525 @@
|
|||
// Copyright 2021 Harran Ali <harran.m@gmail.com>. All rights reserved.
|
||||
// Copyright (c) 2024 Zeni Kim <zenik@smarteching.com>
|
||||
// Use of this source code is governed by MIT-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package core
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"reflect"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"git.smarteching.com/goffee/core/env"
|
||||
"git.smarteching.com/goffee/core/logger"
|
||||
"github.com/google/uuid"
|
||||
"github.com/joho/godotenv"
|
||||
"github.com/julienschmidt/httprouter"
|
||||
)
|
||||
|
||||
func TestNew(t *testing.T) {
|
||||
app := createNewApp(t)
|
||||
if fmt.Sprintf("%T", app) != "*core.App" {
|
||||
t.Errorf("failed testing new core")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSetEnv(t *testing.T) {
|
||||
envVars, err := godotenv.Read("./testingdata/.env")
|
||||
if err != nil {
|
||||
t.Errorf("failed reading .env file")
|
||||
}
|
||||
|
||||
env.SetEnvVars(envVars)
|
||||
|
||||
if os.Getenv("KEY_ONE") != "VAL_ONE" || os.Getenv("KEY_TWO") != "VAL_TWO" {
|
||||
t.Errorf("failed to set env vars")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMakeHTTPHandlerFunc(t *testing.T) {
|
||||
app := createNewApp(t)
|
||||
tmpFile := filepath.Join(t.TempDir(), uuid.NewString())
|
||||
app.SetLogsDriver(&logger.LogFileDriver{
|
||||
FilePath: filepath.Join(t.TempDir(), uuid.NewString()),
|
||||
})
|
||||
hdlr := Handler(func(c *Context) *Response {
|
||||
f, _ := os.Create(tmpFile)
|
||||
f.WriteString("DFT2V56H")
|
||||
c.Response.SetHeader("header-key", "header-val")
|
||||
return c.Response.Text("DFT2V56H")
|
||||
})
|
||||
h := app.makeHTTPRouterHandlerFunc(hdlr, nil)
|
||||
r := httptest.NewRequest(GET, LOCALHOST, nil)
|
||||
w := httptest.NewRecorder()
|
||||
h(w, r, []httprouter.Param{{Key: "tkey", Value: "tvalue"}})
|
||||
rsp := w.Result()
|
||||
if rsp.StatusCode != http.StatusOK {
|
||||
t.Errorf("failed testing make http handler func")
|
||||
}
|
||||
s, _ := os.ReadFile(tmpFile)
|
||||
if string(s) != "DFT2V56H" {
|
||||
t.Errorf("failed testing make http handler func")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMakeHTTPHandlerFuncVerifyJson(t *testing.T) {
|
||||
app := createNewApp(t)
|
||||
tmpFile := filepath.Join(t.TempDir(), uuid.NewString())
|
||||
app.SetLogsDriver(&logger.LogFileDriver{
|
||||
FilePath: filepath.Join(t.TempDir(), uuid.NewString()),
|
||||
})
|
||||
hdlr := Handler(func(c *Context) *Response {
|
||||
f, _ := os.Create(tmpFile)
|
||||
f.WriteString("DFT2V56H")
|
||||
c.Response.SetHeader("header-key", "header-val")
|
||||
return c.Response.Json("{\"testKey\": \"testVal\"}")
|
||||
})
|
||||
h := app.makeHTTPRouterHandlerFunc(hdlr, nil)
|
||||
r := httptest.NewRequest(GET, LOCALHOST, nil)
|
||||
w := httptest.NewRecorder()
|
||||
h(w, r, []httprouter.Param{{Key: "tkey", Value: "tvalue"}})
|
||||
rsp := w.Result()
|
||||
if rsp.StatusCode != http.StatusOK {
|
||||
t.Errorf("failed testing make http handler func with json verify")
|
||||
}
|
||||
b, err := io.ReadAll(rsp.Body)
|
||||
if err != nil {
|
||||
t.Errorf("failed testing make http handler func with json verify")
|
||||
}
|
||||
var j map[string]interface{}
|
||||
err = json.Unmarshal(b, &j)
|
||||
if err != nil {
|
||||
t.Errorf("failed testing make http handler func with json verify: %v", err)
|
||||
}
|
||||
if j["testKey"] != "testVal" {
|
||||
t.Errorf("failed testing make http handler func with json verify")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMethodNotAllowedHandler(t *testing.T) {
|
||||
app := createNewApp(t)
|
||||
app.SetLogsDriver(logger.LogNullDriver{})
|
||||
app.Bootstrap()
|
||||
m := &methodNotAllowed{}
|
||||
r := httptest.NewRequest(GET, LOCALHOST, nil)
|
||||
w := httptest.NewRecorder()
|
||||
m.ServeHTTP(w, r)
|
||||
rsp := w.Result()
|
||||
if rsp.StatusCode != 405 {
|
||||
t.Errorf("failed testing method not allowed")
|
||||
}
|
||||
}
|
||||
|
||||
func TestNotFoundHandler(t *testing.T) {
|
||||
n := ¬FoundHandler{}
|
||||
r := httptest.NewRequest(GET, LOCALHOST, nil)
|
||||
w := httptest.NewRecorder()
|
||||
n.ServeHTTP(w, r)
|
||||
rsp := w.Result()
|
||||
if rsp.StatusCode != 404 {
|
||||
t.Errorf("failed testing not found handler")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestUseMiddleware(t *testing.T) {
|
||||
app := createNewApp(t)
|
||||
UseMiddleware(Middleware(func(c *Context) { c.GetLogger().Info("Testing!") }))
|
||||
if len(app.middlewares.GetMiddlewares()) != 1 {
|
||||
t.Errorf("failed testing use middleware")
|
||||
}
|
||||
}
|
||||
|
||||
func TestChainReset(t *testing.T) {
|
||||
c := &chain{}
|
||||
c.nodes = append(c.nodes, Middleware(func(c *Context) { c.GetLogger().Info("Testing1!") }))
|
||||
c.nodes = append(c.nodes, Middleware(func(c *Context) { c.GetLogger().Info("Testing2!") }))
|
||||
|
||||
c.reset()
|
||||
if len(c.nodes) != 0 {
|
||||
t.Errorf("failed testing reset chain")
|
||||
}
|
||||
}
|
||||
|
||||
func TestNext(t *testing.T) {
|
||||
app := createNewApp(t)
|
||||
app.t = 0
|
||||
tfPath := filepath.Join(t.TempDir(), uuid.NewString())
|
||||
var hs []interface{}
|
||||
hs = append(hs, Middleware(func(c *Context) { c.Next() }))
|
||||
hs = append(hs, Handler(func(c *Context) *Response {
|
||||
f, _ := os.Create(tfPath)
|
||||
f.WriteString("DFT2V56H")
|
||||
return nil
|
||||
}))
|
||||
app.prepareChain(hs)
|
||||
|
||||
app.chain.execute(makeCTX(t))
|
||||
cnt, _ := os.ReadFile(tfPath)
|
||||
if string(cnt) != "DFT2V56H" {
|
||||
// t.Errorf("failed testing next")
|
||||
}
|
||||
}
|
||||
|
||||
func TestChainGetByIndex(t *testing.T) {
|
||||
c := &chain{}
|
||||
tf := filepath.Join(t.TempDir(), uuid.NewString())
|
||||
var hs []interface{}
|
||||
hs = append(hs, Middleware(func(c *Context) { c.GetLogger().Info("testing!") }))
|
||||
hs = append(hs, Middleware(func(c *Context) {
|
||||
f, _ := os.Create(tf)
|
||||
f.WriteString("DFT2V56H")
|
||||
}))
|
||||
c.nodes = hs
|
||||
pf := c.getByIndex(1)
|
||||
f, ok := pf.(Middleware)
|
||||
if ok {
|
||||
f(makeCTX(t))
|
||||
}
|
||||
d, _ := os.ReadFile(tf)
|
||||
if string(d) != "DFT2V56H" {
|
||||
t.Errorf("failed testing chain get by index")
|
||||
}
|
||||
}
|
||||
|
||||
func TestPrepareChain(t *testing.T) {
|
||||
app := createNewApp(t)
|
||||
UseMiddleware(Middleware(func(c *Context) { c.GetLogger().Info("Testing!") }))
|
||||
var hs []interface{}
|
||||
hs = append(hs, Middleware(func(c *Context) { c.GetLogger().Info("testing1!") }))
|
||||
hs = append(hs, Middleware(func(c *Context) { c.GetLogger().Info("testing2!") }))
|
||||
app.prepareChain(hs)
|
||||
if len(app.chain.nodes) != 3 {
|
||||
t.Errorf("failed preparing chain")
|
||||
}
|
||||
}
|
||||
|
||||
func TestChainExecute(t *testing.T) {
|
||||
tmpDir := t.TempDir()
|
||||
f1Path := filepath.Join(tmpDir, uuid.NewString())
|
||||
c := &chain{}
|
||||
c.nodes = []interface{}{
|
||||
Handler(func(c *Context) *Response {
|
||||
tf, _ := os.Create(f1Path)
|
||||
defer tf.Close()
|
||||
tf.WriteString("DFT2V56H")
|
||||
return nil
|
||||
}),
|
||||
}
|
||||
ctx := makeCTX(t)
|
||||
c.execute(ctx)
|
||||
cnt, _ := os.ReadFile(f1Path)
|
||||
if string(cnt) != "DFT2V56H" {
|
||||
t.Errorf("failed testing execute chain")
|
||||
}
|
||||
}
|
||||
|
||||
func makeCTX(t *testing.T) *Context {
|
||||
t.Helper()
|
||||
return &Context{
|
||||
Request: &Request{
|
||||
httpRequest: httptest.NewRequest(GET, LOCALHOST, nil),
|
||||
httpPathParams: nil,
|
||||
},
|
||||
Response: &Response{
|
||||
headers: []header{},
|
||||
body: nil,
|
||||
HttpResponseWriter: httptest.NewRecorder(),
|
||||
},
|
||||
GetValidator: nil,
|
||||
GetJWT: nil,
|
||||
}
|
||||
}
|
||||
|
||||
func TestcombHndlers(t *testing.T) {
|
||||
app := createNewApp(t)
|
||||
t1 := Handler(func(c *Context) *Response { c.GetLogger().Info("Testing1!"); return nil })
|
||||
t2 := Middleware(func(c *Context) { c.GetLogger().Info("Testing2!") })
|
||||
|
||||
mw := []Middleware{t2}
|
||||
comb := app.combHandlers(t1, mw)
|
||||
if reflect.ValueOf(t1).Pointer() != reflect.ValueOf(comb[0]).Pointer() {
|
||||
t.Errorf("failed testing reverse handlers")
|
||||
}
|
||||
|
||||
if reflect.ValueOf(t2).Pointer() != reflect.ValueOf(comb[1]).Pointer() {
|
||||
t.Errorf("failed testing reverse handlers")
|
||||
}
|
||||
}
|
||||
|
||||
func TestRegisterGetRoute(t *testing.T) {
|
||||
app := New()
|
||||
hr := httprouter.New()
|
||||
gcr := NewRouter()
|
||||
gcr.Get("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
gcr.Post("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
gcr.Delete("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
gcr.Patch("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
gcr.Put("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
gcr.Options("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
gcr.Head("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
hr = app.RegisterRoutes(gcr.GetRoutes(), hr)
|
||||
s := httptest.NewServer(hr)
|
||||
defer s.Close()
|
||||
clt := &http.Client{}
|
||||
req, err := http.NewRequest("GET", s.URL+"?param=valget", nil)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register routes")
|
||||
}
|
||||
rsp, err := clt.Do(req)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register routes")
|
||||
}
|
||||
b, err := io.ReadAll(rsp.Body)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register routes")
|
||||
}
|
||||
if strings.TrimSpace(string(b)) != "valget" {
|
||||
t.Errorf("failed test register routes")
|
||||
}
|
||||
rsp.Body.Close()
|
||||
}
|
||||
|
||||
func TestRegisterPostRoute(t *testing.T) {
|
||||
app := New()
|
||||
hr := httprouter.New()
|
||||
gcr := NewRouter()
|
||||
gcr.Post("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
hr = app.RegisterRoutes(gcr.GetRoutes(), hr)
|
||||
s := httptest.NewServer(hr)
|
||||
defer s.Close()
|
||||
clt := &http.Client{}
|
||||
req, err := http.NewRequest("POST", s.URL+"?param=valpost", nil)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register post route")
|
||||
}
|
||||
rsp, err := clt.Do(req)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register post route")
|
||||
}
|
||||
b, err := io.ReadAll(rsp.Body)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register post route")
|
||||
}
|
||||
if strings.TrimSpace(string(b)) != "valpost" {
|
||||
t.Errorf("failed test register post route")
|
||||
}
|
||||
rsp.Body.Close()
|
||||
}
|
||||
|
||||
func TestRegisterDeleteRoute(t *testing.T) {
|
||||
app := New()
|
||||
hr := httprouter.New()
|
||||
gcr := NewRouter()
|
||||
gcr.Delete("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
hr = app.RegisterRoutes(gcr.GetRoutes(), hr)
|
||||
s := httptest.NewServer(hr)
|
||||
defer s.Close()
|
||||
clt := &http.Client{}
|
||||
req, err := http.NewRequest("DELETE", s.URL+"?param=valdelete", nil)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register delete route")
|
||||
}
|
||||
rsp, err := clt.Do(req)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register delete route")
|
||||
}
|
||||
b, err := io.ReadAll(rsp.Body)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register delete route")
|
||||
}
|
||||
if strings.TrimSpace(string(b)) != "valdelete" {
|
||||
t.Errorf("failed test register delete route")
|
||||
}
|
||||
rsp.Body.Close()
|
||||
}
|
||||
|
||||
func TestRegisterPatchRoute(t *testing.T) {
|
||||
app := New()
|
||||
hr := httprouter.New()
|
||||
gcr := NewRouter()
|
||||
gcr.Patch("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
hr = app.RegisterRoutes(gcr.GetRoutes(), hr)
|
||||
s := httptest.NewServer(hr)
|
||||
defer s.Close()
|
||||
clt := &http.Client{}
|
||||
req, err := http.NewRequest("PATCH", s.URL+"?param=valpatch", nil)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register patch route")
|
||||
}
|
||||
rsp, err := clt.Do(req)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register patch route")
|
||||
}
|
||||
b, err := io.ReadAll(rsp.Body)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register patch route")
|
||||
}
|
||||
if strings.TrimSpace(string(b)) != "valpatch" {
|
||||
t.Errorf("failed test register patch route")
|
||||
}
|
||||
rsp.Body.Close()
|
||||
}
|
||||
|
||||
func TestRegisterPutRoute(t *testing.T) {
|
||||
app := New()
|
||||
hr := httprouter.New()
|
||||
gcr := NewRouter()
|
||||
gcr.Put("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
hr = app.RegisterRoutes(gcr.GetRoutes(), hr)
|
||||
s := httptest.NewServer(hr)
|
||||
defer s.Close()
|
||||
clt := &http.Client{}
|
||||
req, err := http.NewRequest("PUT", s.URL+"?param=valput", nil)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register put route")
|
||||
}
|
||||
rsp, err := clt.Do(req)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register put route")
|
||||
}
|
||||
b, err := io.ReadAll(rsp.Body)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register put route")
|
||||
}
|
||||
if strings.TrimSpace(string(b)) != "valput" {
|
||||
t.Errorf("failed test register put route")
|
||||
}
|
||||
rsp.Body.Close()
|
||||
}
|
||||
|
||||
func TestRegisterOptionsRoute(t *testing.T) {
|
||||
app := New()
|
||||
hr := httprouter.New()
|
||||
gcr := NewRouter()
|
||||
gcr.Options("/", Handler(func(c *Context) *Response {
|
||||
fmt.Fprintln(c.Response.HttpResponseWriter, c.GetRequestParam("param"))
|
||||
return nil
|
||||
}))
|
||||
hr = app.RegisterRoutes(gcr.GetRoutes(), hr)
|
||||
s := httptest.NewServer(hr)
|
||||
defer s.Close()
|
||||
clt := &http.Client{}
|
||||
req, err := http.NewRequest("OPTIONS", s.URL+"?param=valoptions", nil)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register options route")
|
||||
}
|
||||
rsp, err := clt.Do(req)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register options route")
|
||||
}
|
||||
b, err := io.ReadAll(rsp.Body)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register options route")
|
||||
}
|
||||
if strings.TrimSpace(string(b)) != "valoptions" {
|
||||
t.Errorf("failed test register options route")
|
||||
}
|
||||
rsp.Body.Close()
|
||||
}
|
||||
|
||||
func TestRegisterHeadRoute(t *testing.T) {
|
||||
app := New()
|
||||
hr := httprouter.New()
|
||||
gcr := NewRouter()
|
||||
tfp := filepath.Join(t.TempDir(), uuid.NewString())
|
||||
gcr.Head("/", Handler(func(c *Context) *Response {
|
||||
param := c.GetRequestParam("param")
|
||||
p, _ := param.(string)
|
||||
f, err := os.OpenFile(p, os.O_CREATE|os.O_RDWR, 777)
|
||||
if err != nil {
|
||||
fmt.Println(err.Error())
|
||||
}
|
||||
defer f.Close()
|
||||
f.WriteString("fromhead")
|
||||
return nil
|
||||
}))
|
||||
hr = app.RegisterRoutes(gcr.GetRoutes(), hr)
|
||||
s := httptest.NewServer(hr)
|
||||
defer s.Close()
|
||||
clt := &http.Client{}
|
||||
req, err := http.NewRequest("HEAD", s.URL+"?param="+tfp, nil)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register head route")
|
||||
}
|
||||
rsp, err := clt.Do(req)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register head route")
|
||||
}
|
||||
f, err := os.Open(tfp)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register head route: %v", err.Error())
|
||||
}
|
||||
b, err := io.ReadAll(f)
|
||||
if err != nil {
|
||||
t.Errorf("failed test register head route: %v", err.Error())
|
||||
}
|
||||
f.Close()
|
||||
if strings.TrimSpace(string(b)) != "fromhead" {
|
||||
t.Errorf("failed test register head route")
|
||||
}
|
||||
rsp.Body.Close()
|
||||
}
|
||||
|
||||
func TestPanicHandler(t *testing.T) {
|
||||
os.Setenv("APP_DEBUG_MODE", "true")
|
||||
loggr = logger.NewLogger(&logger.LogNullDriver{})
|
||||
r := httptest.NewRequest(GET, LOCALHOST, nil)
|
||||
w := httptest.NewRecorder()
|
||||
panicHandler(w, r, "")
|
||||
rsp := w.Result()
|
||||
b, _ := io.ReadAll(rsp.Body)
|
||||
if !strings.Contains(string(b), "stack trace") {
|
||||
t.Errorf("failed test panic handler")
|
||||
}
|
||||
}
|
||||
|
||||
func createNewApp(t *testing.T) *App {
|
||||
t.Helper()
|
||||
a := New()
|
||||
a.SetLogsDriver(&logger.LogNullDriver{})
|
||||
a.SetRequestConfig(testingRequestC)
|
||||
|
||||
return a
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue