SimpleCloudNotifier/server/test/common_test.go
2022-11-24 12:53:27 +01:00

198 lines
4.7 KiB
Go

package test
import (
scn "blackforestbytes.com/simplecloudnotifier"
"blackforestbytes.com/simplecloudnotifier/api"
"blackforestbytes.com/simplecloudnotifier/common/ginext"
"blackforestbytes.com/simplecloudnotifier/db"
"blackforestbytes.com/simplecloudnotifier/jobs"
"blackforestbytes.com/simplecloudnotifier/logic"
"blackforestbytes.com/simplecloudnotifier/push"
"bytes"
"encoding/json"
"fmt"
"github.com/gin-gonic/gin"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
"gogs.mikescher.com/BlackForestBytes/goext/langext"
"io"
"net/http"
"os"
"path/filepath"
"time"
)
type Void = struct{}
func NewSimpleWebserver() (*logic.Application, func()) {
cw := zerolog.ConsoleWriter{
Out: os.Stdout,
TimeFormat: "2006-01-02 15:04:05 Z07:00",
}
zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
multi := zerolog.MultiLevelWriter(cw)
logger := zerolog.New(multi).With().
Timestamp().
Caller().
Logger()
log.Logger = logger
gin.SetMode(gin.TestMode)
zerolog.SetGlobalLevel(zerolog.DebugLevel)
uuid1, _ := langext.NewHexUUID()
uuid2, _ := langext.NewHexUUID()
dbdir := filepath.Join(os.TempDir(), uuid1)
dbfile := filepath.Join(dbdir, uuid2+".sqlite3")
err := os.MkdirAll(dbdir, os.ModePerm)
if err != nil {
panic(err)
}
f, err := os.Create(dbfile)
if err != nil {
panic(err)
}
err = f.Close()
if err != nil {
panic(err)
}
err = os.Chmod(dbfile, 0777)
if err != nil {
panic(err)
}
//dbfile := "/home/mike/Code/private/SimpleCloudNotifier/server/.run-data/db_test.sqlite3"
fmt.Println("DatabaseFile: " + dbfile)
conf := scn.Config{
Namespace: "test",
GinDebug: true,
ServerIP: "0.0.0.0",
ServerPort: "0", // simply choose a free port
DBFile: dbfile,
RequestTimeout: 500 * time.Millisecond,
ReturnRawErrors: true,
DummyFirebase: true,
}
sqlite, err := db.NewDatabase(dbfile)
if err != nil {
panic(err)
}
app := logic.NewApp(sqlite)
if err := app.Migrate(); err != nil {
panic(err)
}
ginengine := ginext.NewEngine(conf)
router := api.NewRouter(app)
nc := push.NewTestSink()
jobRetry := jobs.NewDeliveryRetryJob(app)
app.Init(conf, ginengine, nc, []logic.Job{jobRetry})
router.Init(ginengine)
return app, func() { app.Stop(); _ = os.Remove(dbfile) }
}
func requestGet[TResult any](baseURL string, prefix string) TResult {
return requestAny[TResult]("", "GET", baseURL, prefix, nil)
}
func requestAuthGet[TResult any](akey string, baseURL string, prefix string) TResult {
return requestAny[TResult](akey, "GET", baseURL, prefix, nil)
}
func requestPost[TResult any](baseURL string, prefix string, body any) TResult {
return requestAny[TResult]("", "POST", baseURL, prefix, body)
}
func requestAuthPost[TResult any](akey string, baseURL string, prefix string, body any) TResult {
return requestAny[TResult](akey, "POST", baseURL, prefix, body)
}
func requestPut[TResult any](baseURL string, prefix string, body any) TResult {
return requestAny[TResult]("", "PUT", baseURL, prefix, body)
}
func requestAuthPUT[TResult any](akey string, baseURL string, prefix string, body any) TResult {
return requestAny[TResult](akey, "PUT", baseURL, prefix, body)
}
func requestPatch[TResult any](baseURL string, prefix string, body any) TResult {
return requestAny[TResult]("", "PATCH", baseURL, prefix, body)
}
func requestAuthPatch[TResult any](akey string, baseURL string, prefix string, body any) TResult {
return requestAny[TResult](akey, "PATCH", baseURL, prefix, body)
}
func requestDelete[TResult any](baseURL string, prefix string, body any) TResult {
return requestAny[TResult]("", "DELETE", baseURL, prefix, body)
}
func requestAuthDelete[TResult any](akey string, baseURL string, prefix string, body any) TResult {
return requestAny[TResult](akey, "DELETE", baseURL, prefix, body)
}
func requestAny[TResult any](akey string, method string, baseURL string, prefix string, body any) TResult {
client := http.Client{}
bytesbody := make([]byte, 0)
if body != nil {
bjson, err := json.Marshal(body)
if err != nil {
panic(err)
}
bytesbody = bjson
}
req, err := http.NewRequest(method, baseURL+prefix, bytes.NewReader(bytesbody))
if err != nil {
panic(err)
}
if body != nil {
req.Header.Set("Content-Type", "application/json")
}
if akey != "" {
req.Header.Set("Authorization", "SCN "+akey)
}
resp, err := client.Do(req)
if err != nil {
panic(err)
}
defer func() { _ = resp.Body.Close() }()
respBodyBin, err := io.ReadAll(resp.Body)
if err != nil {
panic(err)
}
if resp.StatusCode != 200 {
fmt.Println("Request: " + method + " :: " + baseURL + prefix)
fmt.Println(string(respBodyBin))
panic("Statuscode != 200")
}
var data TResult
if err := json.Unmarshal(respBodyBin, &data); err != nil {
panic(err)
}
return data
}