phoenixandphoenix
d34cad033d
Closes #30 Reviewed-on: phoenix/textsender-auth#33 Co-authored-by: phoenix <kundeng00@pm.me> Co-committed-by: phoenix <kundeng00@pm.me>
97 lines
2.9 KiB
Go
97 lines
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/joho/godotenv"
|
|
|
|
"git.kundeng.us/phoenix/textsender-auth/internal/config"
|
|
"git.kundeng.us/phoenix/textsender-auth/internal/db"
|
|
"git.kundeng.us/phoenix/textsender-auth/internal/handler"
|
|
"git.kundeng.us/phoenix/textsender-auth/internal/handler/endpoint"
|
|
mdleware "git.kundeng.us/phoenix/textsender-auth/internal/middleware"
|
|
"git.kundeng.us/phoenix/textsender-auth/internal/services"
|
|
"git.kundeng.us/phoenix/textsender-auth/internal/store"
|
|
)
|
|
|
|
var testRouter *chi.Mux
|
|
|
|
func TestMain(m *testing.M) {
|
|
cfg := load()
|
|
|
|
database, err := db.NewDatabase(cfg.GetDBConnString())
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
panic("Failed to initialize database")
|
|
}
|
|
defer database.Close()
|
|
|
|
ctx := context.Background()
|
|
err = database.ResetDatabase(ctx)
|
|
if err != nil {
|
|
fmt.Println(err.Error())
|
|
panic("Failed to initialize database")
|
|
}
|
|
|
|
userStore := store.NewUserStore(database.Pool)
|
|
serviceStore := store.NewServiceStore(database.Pool)
|
|
userHandler := handler.NewUserHandler(cfg, userStore)
|
|
loginHandler := handler.NewLoginHandler(cfg, userStore)
|
|
serviceHandler := handler.NewServiceHandler(cfg, serviceStore)
|
|
refreshHandler := handler.NewRefreshHandler(cfg, userStore, serviceStore)
|
|
|
|
testRouter = chi.NewRouter()
|
|
jwtService := services.NewJWTService(config.GetSecretKey())
|
|
testRouter.Method("POST", endpoint.Register, http.HandlerFunc(userHandler.Register))
|
|
testRouter.Method("POST", endpoint.Login, http.HandlerFunc(loginHandler.Login))
|
|
testRouter.Method("POST", endpoint.CreateServiceUser, http.HandlerFunc(serviceHandler.Register))
|
|
testRouter.Method("POST", endpoint.LoginServiceUser, http.HandlerFunc(serviceHandler.Login))
|
|
testRouter.Method("POST", endpoint.TokenRefresh, http.HandlerFunc(refreshHandler.Refresh))
|
|
testRouter.Method("PATCH", endpoint.UpdatePassword, mdleware.AuthMiddleware(jwtService)(http.HandlerFunc(loginHandler.UpdatePassword)))
|
|
|
|
code := m.Run()
|
|
os.Exit(code)
|
|
}
|
|
|
|
func load() *config.Config {
|
|
resetDb := flag.Bool("reset-db", false, "Reset the database schema and exit")
|
|
port := flag.String("port", config.Port, "Server port")
|
|
flag.Parse()
|
|
|
|
cwd, _ := os.Getwd()
|
|
envPath := path.Join(cwd, ".env")
|
|
|
|
err := godotenv.Load(envPath)
|
|
if err != nil {
|
|
envPath = path.Join(cwd, "../..", ".env")
|
|
if err := godotenv.Load(envPath); err != nil {
|
|
panic("Error loading .env file: " + err.Error())
|
|
}
|
|
}
|
|
|
|
unpackedConnString := config.UnpackDBConnString()
|
|
dbConnString := unpackedConnString.Parse()
|
|
|
|
return &config.Config{
|
|
DBConnString: dbConnString,
|
|
ServerPort: *port,
|
|
ResetDB: *resetDb,
|
|
EnableRegistration: config.CheckRegistration(),
|
|
}
|
|
}
|
|
|
|
func resetTestDB(t *testing.T) {
|
|
t.Helper()
|
|
_, err := db.Pool.Exec(context.Background(), "DELETE FROM users")
|
|
if err != nil {
|
|
t.Fatalf("Failed to reset test database: %v", err)
|
|
}
|
|
}
|