refactor testing files
This commit is contained in:
45
internal/setup/setup.go
Normal file
45
internal/setup/setup.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package setup
|
||||
|
||||
import (
|
||||
"github.com/gin-contrib/cors"
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"git.artlef.fr/PersonalLibraryManager/internal/api"
|
||||
"git.artlef.fr/PersonalLibraryManager/internal/config"
|
||||
"git.artlef.fr/PersonalLibraryManager/internal/db"
|
||||
"git.artlef.fr/PersonalLibraryManager/internal/jwtauth"
|
||||
"git.artlef.fr/PersonalLibraryManager/internal/middleware"
|
||||
)
|
||||
|
||||
func Setup(config *config.Config) *gin.Engine {
|
||||
db := db.Initdb(config.DatabaseFilePath, config.DemoDataPath)
|
||||
err := jwtauth.InitKey(config.JWTKey)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
r := gin.Default()
|
||||
r.Use(cors.New(configureCors())) // All origins allowed by default
|
||||
r.Use(middleware.Auth())
|
||||
r.GET("/mybooks", func(c *gin.Context) {
|
||||
api.GetMyBooksHanderl(c, db)
|
||||
})
|
||||
r.POST("/book", func(c *gin.Context) {
|
||||
api.PostBookHandler(c, db)
|
||||
})
|
||||
r.POST("/auth/signup", func(c *gin.Context) {
|
||||
api.PostSignupHandler(c, db)
|
||||
})
|
||||
r.POST("/auth/login", func(c *gin.Context) {
|
||||
api.PostLoginHandler(c, db)
|
||||
})
|
||||
return r
|
||||
}
|
||||
|
||||
func configureCors() cors.Config {
|
||||
config := cors.DefaultConfig()
|
||||
config.AllowOrigins = []string{"http://localhost:5173"}
|
||||
config.AllowPrivateNetwork = true
|
||||
config.AllowCredentials = true
|
||||
config.AllowHeaders = []string{"Authorization", "Content-Type"}
|
||||
return config
|
||||
}
|
||||
53
internal/testutils/testutils.go
Normal file
53
internal/testutils/testutils.go
Normal file
@@ -0,0 +1,53 @@
|
||||
package testutils
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"log"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"strings"
|
||||
|
||||
"git.artlef.fr/PersonalLibraryManager/internal/config"
|
||||
"git.artlef.fr/PersonalLibraryManager/internal/setup"
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
func TestSetup() *gin.Engine {
|
||||
c := config.LoadConfig("config_test/test.toml")
|
||||
return setup.Setup(&c)
|
||||
}
|
||||
|
||||
type loginResponse struct {
|
||||
Message string `json:"message"`
|
||||
Token string `json:"token"`
|
||||
}
|
||||
|
||||
func ConnectDemoUser(router *gin.Engine) string {
|
||||
loginJson :=
|
||||
`{
|
||||
"username": "demo",
|
||||
"password":"demopw"
|
||||
}`
|
||||
return connectUser(router, loginJson)
|
||||
}
|
||||
|
||||
func ConnectDemo2User(router *gin.Engine) string {
|
||||
loginJson :=
|
||||
`{
|
||||
"username": "demo2",
|
||||
"password":"demopw"
|
||||
}`
|
||||
return connectUser(router, loginJson)
|
||||
}
|
||||
|
||||
func connectUser(router *gin.Engine, loginJson string) string {
|
||||
w := httptest.NewRecorder()
|
||||
req, _ := http.NewRequest("POST", "/auth/login", strings.NewReader(loginJson))
|
||||
router.ServeHTTP(w, req)
|
||||
var parsedResponse loginResponse
|
||||
err := json.Unmarshal(w.Body.Bytes(), &parsedResponse)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
return parsedResponse.Token
|
||||
}
|
||||
Reference in New Issue
Block a user