david
4290ee3455
- Replace mattn/go-sqlite3 with modernc.org/sqlite (pure-Go, no CGO required) - Update GoReleaser config to disable CGO for ARM64 builds - Change driver name from 'sqlite3' to 'sqlite' in all sql.Open calls - Enables successful cross-compilation for ARM64 without cross-compiler toolchain - All tests passing with new driver
107 lines
3.6 KiB
Go
107 lines
3.6 KiB
Go
package http
|
|
|
|
import (
|
|
"bytes"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"apocapoc-api/internal/application/commands"
|
|
"apocapoc-api/internal/application/queries"
|
|
"apocapoc-api/internal/infrastructure/auth"
|
|
"apocapoc-api/internal/infrastructure/crypto"
|
|
"apocapoc-api/internal/infrastructure/persistence/sqlite"
|
|
|
|
_ "modernc.org/sqlite"
|
|
)
|
|
|
|
type TestServer struct {
|
|
Router *http.Handler
|
|
DB *sql.DB
|
|
}
|
|
|
|
func setupTestServer(t *testing.T) *TestServer {
|
|
db, err := sql.Open("sqlite", ":memory:")
|
|
if err != nil {
|
|
t.Fatalf("Failed to open test database: %v", err)
|
|
}
|
|
|
|
if err := sqlite.RunMigrations(db); err != nil {
|
|
t.Fatalf("Failed to run migrations: %v", err)
|
|
}
|
|
|
|
jwtService := auth.NewJWTService("test-secret", 24)
|
|
passwordHasher := crypto.NewBcryptHasher()
|
|
|
|
userRepo := sqlite.NewUserRepository(db)
|
|
habitRepo := sqlite.NewHabitRepository(db)
|
|
entryRepo := sqlite.NewHabitEntryRepository(db)
|
|
refreshTokenRepo := sqlite.NewRefreshTokenRepository(db)
|
|
|
|
registerHandler := commands.NewRegisterUserHandler(userRepo, passwordHasher)
|
|
loginHandler := queries.NewLoginUserHandler(userRepo, passwordHasher)
|
|
refreshTokenHandler := queries.NewRefreshTokenHandler(refreshTokenRepo, userRepo)
|
|
revokeTokenHandler := commands.NewRevokeTokenHandler(refreshTokenRepo)
|
|
revokeAllTokensHandler := commands.NewRevokeAllTokensHandler(refreshTokenRepo)
|
|
createHandler := commands.NewCreateHabitHandler(habitRepo)
|
|
getTodaysHandler := queries.NewGetTodaysHabitsHandler(habitRepo, entryRepo)
|
|
getUserHabitsHandler := queries.NewGetUserHabitsHandler(habitRepo)
|
|
getHabitByIDHandler := queries.NewGetHabitByIDHandler(habitRepo)
|
|
getHabitEntriesHandler := queries.NewGetHabitEntriesHandler(habitRepo, entryRepo)
|
|
getHabitStatsHandler := queries.NewGetHabitStatsHandler(habitRepo, entryRepo)
|
|
updateHandler := commands.NewUpdateHabitHandler(habitRepo)
|
|
archiveHandler := commands.NewArchiveHabitHandler(habitRepo)
|
|
markHandler := commands.NewMarkHabitHandler(entryRepo, habitRepo)
|
|
unmarkHandler := commands.NewUnmarkHabitHandler(habitRepo, entryRepo)
|
|
|
|
refreshTokenExpiry := 7 * 24 * time.Hour
|
|
|
|
authHandlers := NewAuthHandlers(registerHandler, loginHandler, refreshTokenHandler, revokeTokenHandler, revokeAllTokensHandler, jwtService, refreshTokenRepo, refreshTokenExpiry)
|
|
habitHandlers := NewHabitHandlers(createHandler, getTodaysHandler, getUserHabitsHandler, getHabitByIDHandler, getHabitEntriesHandler, updateHandler, archiveHandler, markHandler, unmarkHandler)
|
|
statsHandlers := NewStatsHandlers(getHabitStatsHandler)
|
|
healthHandlers := NewHealthHandlers(db)
|
|
|
|
router := NewRouter("*", habitHandlers, authHandlers, statsHandlers, healthHandlers, jwtService)
|
|
|
|
handler := http.Handler(router)
|
|
return &TestServer{
|
|
Router: &handler,
|
|
DB: db,
|
|
}
|
|
}
|
|
|
|
func (ts *TestServer) Close() {
|
|
ts.DB.Close()
|
|
}
|
|
|
|
func makeRequest(t *testing.T, handler http.Handler, method, path string, body interface{}, authToken string) *httptest.ResponseRecorder {
|
|
var bodyBytes []byte
|
|
if body != nil {
|
|
var err error
|
|
bodyBytes, err = json.Marshal(body)
|
|
if err != nil {
|
|
t.Fatalf("Failed to marshal request body: %v", err)
|
|
}
|
|
}
|
|
|
|
req := httptest.NewRequest(method, path, bytes.NewReader(bodyBytes))
|
|
req.Header.Set("Content-Type", "application/json")
|
|
if authToken != "" {
|
|
req.Header.Set("Authorization", "Bearer "+authToken)
|
|
}
|
|
|
|
rr := httptest.NewRecorder()
|
|
handler.ServeHTTP(rr, req)
|
|
|
|
return rr
|
|
}
|
|
|
|
func decodeResponse(t *testing.T, rr *httptest.ResponseRecorder, target interface{}) {
|
|
if err := json.NewDecoder(rr.Body).Decode(target); err != nil {
|
|
t.Fatalf("Failed to decode response: %v", err)
|
|
}
|
|
}
|