test: Add unit and integration tests for API and storage
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Jan Novak
517
backend/internal/api/sessions_test.go
Normal file
517
backend/internal/api/sessions_test.go
Normal file
@@ -0,0 +1,517 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"training-tracker/internal/models"
|
||||
)
|
||||
|
||||
func TestSessionHandler_List(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
now := time.Now()
|
||||
tests := []struct {
|
||||
name string
|
||||
mock *mockSessionRepo
|
||||
wantStatus int
|
||||
wantLen int
|
||||
}{
|
||||
{
|
||||
name: "empty list",
|
||||
mock: &mockSessionRepo{sessions: nil},
|
||||
wantStatus: http.StatusOK,
|
||||
wantLen: 0,
|
||||
},
|
||||
{
|
||||
name: "multiple sessions",
|
||||
mock: &mockSessionRepo{
|
||||
sessions: []models.Session{
|
||||
{ID: 1, Date: now, Notes: "Morning workout"},
|
||||
{ID: 2, Date: now.AddDate(0, 0, -1), Notes: "Evening cardio"},
|
||||
},
|
||||
},
|
||||
wantStatus: http.StatusOK,
|
||||
wantLen: 2,
|
||||
},
|
||||
{
|
||||
name: "session with plan",
|
||||
mock: &mockSessionRepo{
|
||||
sessions: []models.Session{
|
||||
{
|
||||
ID: 1,
|
||||
Date: now,
|
||||
PlanID: ptr(int64(1)),
|
||||
Plan: &models.TrainingPlan{ID: 1, Name: "Push Day"},
|
||||
},
|
||||
},
|
||||
},
|
||||
wantStatus: http.StatusOK,
|
||||
wantLen: 1,
|
||||
},
|
||||
{
|
||||
name: "repository error",
|
||||
mock: &mockSessionRepo{err: errors.New("db error")},
|
||||
wantStatus: http.StatusInternalServerError,
|
||||
wantLen: 0,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
handler := NewSessionHandler(tt.mock)
|
||||
req := httptest.NewRequest(http.MethodGet, "/api/sessions", nil)
|
||||
rec := httptest.NewRecorder()
|
||||
|
||||
handler.ServeHTTP(rec, req)
|
||||
|
||||
if rec.Code != tt.wantStatus {
|
||||
t.Errorf("status = %d, want %d", rec.Code, tt.wantStatus)
|
||||
}
|
||||
|
||||
if tt.wantStatus == http.StatusOK {
|
||||
var sessions []models.Session
|
||||
if err := json.NewDecoder(rec.Body).Decode(&sessions); err != nil {
|
||||
t.Fatalf("failed to decode response: %v", err)
|
||||
}
|
||||
if len(sessions) != tt.wantLen {
|
||||
t.Errorf("len = %d, want %d", len(sessions), tt.wantLen)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSessionHandler_GetByID(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
now := time.Now()
|
||||
tests := []struct {
|
||||
name string
|
||||
url string
|
||||
mock *mockSessionRepo
|
||||
wantStatus int
|
||||
wantNotes string
|
||||
}{
|
||||
{
|
||||
name: "found",
|
||||
url: "/api/sessions/1",
|
||||
mock: &mockSessionRepo{
|
||||
session: &models.Session{
|
||||
ID: 1,
|
||||
Date: now,
|
||||
Notes: "Great workout",
|
||||
CreatedAt: now,
|
||||
},
|
||||
},
|
||||
wantStatus: http.StatusOK,
|
||||
wantNotes: "Great workout",
|
||||
},
|
||||
{
|
||||
name: "found with entries",
|
||||
url: "/api/sessions/1",
|
||||
mock: &mockSessionRepo{
|
||||
session: &models.Session{
|
||||
ID: 1,
|
||||
Date: now,
|
||||
Notes: "Complete session",
|
||||
Entries: []models.SessionEntry{
|
||||
{ID: 1, SessionID: 1, ExerciseID: 1, Weight: 100, Reps: 10},
|
||||
{ID: 2, SessionID: 1, ExerciseID: 2, Duration: 1800},
|
||||
},
|
||||
CreatedAt: now,
|
||||
},
|
||||
},
|
||||
wantStatus: http.StatusOK,
|
||||
wantNotes: "Complete session",
|
||||
},
|
||||
{
|
||||
name: "not found",
|
||||
url: "/api/sessions/999",
|
||||
mock: &mockSessionRepo{session: nil},
|
||||
wantStatus: http.StatusNotFound,
|
||||
},
|
||||
{
|
||||
name: "repository error",
|
||||
url: "/api/sessions/1",
|
||||
mock: &mockSessionRepo{err: errors.New("db error")},
|
||||
wantStatus: http.StatusInternalServerError,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
handler := NewSessionHandler(tt.mock)
|
||||
req := httptest.NewRequest(http.MethodGet, tt.url, nil)
|
||||
rec := httptest.NewRecorder()
|
||||
|
||||
handler.ServeHTTP(rec, req)
|
||||
|
||||
if rec.Code != tt.wantStatus {
|
||||
t.Errorf("status = %d, want %d", rec.Code, tt.wantStatus)
|
||||
}
|
||||
|
||||
if tt.wantStatus == http.StatusOK {
|
||||
var session models.Session
|
||||
if err := json.NewDecoder(rec.Body).Decode(&session); err != nil {
|
||||
t.Fatalf("failed to decode response: %v", err)
|
||||
}
|
||||
if session.Notes != tt.wantNotes {
|
||||
t.Errorf("notes = %q, want %q", session.Notes, tt.wantNotes)
|
||||
}
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSessionHandler_Create(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
now := time.Now()
|
||||
dateStr := now.Format(time.RFC3339)
|
||||
tests := []struct {
|
||||
name string
|
||||
body string
|
||||
mock *mockSessionRepo
|
||||
wantStatus int
|
||||
}{
|
||||
{
|
||||
name: "valid session without plan",
|
||||
body: `{"date":"` + dateStr + `","notes":"Morning run"}`,
|
||||
mock: &mockSessionRepo{
|
||||
session: &models.Session{
|
||||
ID: 1,
|
||||
Date: now,
|
||||
Notes: "Morning run",
|
||||
CreatedAt: now,
|
||||
},
|
||||
},
|
||||
wantStatus: http.StatusCreated,
|
||||
},
|
||||
{
|
||||
name: "valid session with plan",
|
||||
body: `{"date":"` + dateStr + `","plan_id":1,"notes":"Following plan"}`,
|
||||
mock: &mockSessionRepo{
|
||||
session: &models.Session{
|
||||
ID: 2,
|
||||
Date: now,
|
||||
PlanID: ptr(int64(1)),
|
||||
Notes: "Following plan",
|
||||
CreatedAt: now,
|
||||
},
|
||||
},
|
||||
wantStatus: http.StatusCreated,
|
||||
},
|
||||
{
|
||||
name: "missing date",
|
||||
body: `{"notes":"No date"}`,
|
||||
mock: &mockSessionRepo{},
|
||||
wantStatus: http.StatusBadRequest,
|
||||
},
|
||||
{
|
||||
name: "invalid JSON",
|
||||
body: `{invalid}`,
|
||||
mock: &mockSessionRepo{},
|
||||
wantStatus: http.StatusBadRequest,
|
||||
},
|
||||
{
|
||||
name: "repository error",
|
||||
body: `{"date":"` + dateStr + `"}`,
|
||||
mock: &mockSessionRepo{err: errors.New("db error")},
|
||||
wantStatus: http.StatusInternalServerError,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
handler := NewSessionHandler(tt.mock)
|
||||
req := httptest.NewRequest(http.MethodPost, "/api/sessions", bytes.NewBufferString(tt.body))
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
rec := httptest.NewRecorder()
|
||||
|
||||
handler.ServeHTTP(rec, req)
|
||||
|
||||
if rec.Code != tt.wantStatus {
|
||||
t.Errorf("status = %d, want %d", rec.Code, tt.wantStatus)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSessionHandler_Update(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
now := time.Now()
|
||||
dateStr := now.Format(time.RFC3339)
|
||||
tests := []struct {
|
||||
name string
|
||||
url string
|
||||
body string
|
||||
mock *mockSessionRepo
|
||||
wantStatus int
|
||||
}{
|
||||
{
|
||||
name: "valid update",
|
||||
url: "/api/sessions/1",
|
||||
body: `{"date":"` + dateStr + `","notes":"Updated notes"}`,
|
||||
mock: &mockSessionRepo{
|
||||
session: &models.Session{
|
||||
ID: 1,
|
||||
Date: now,
|
||||
Notes: "Updated notes",
|
||||
CreatedAt: now,
|
||||
},
|
||||
},
|
||||
wantStatus: http.StatusOK,
|
||||
},
|
||||
{
|
||||
name: "update with plan",
|
||||
url: "/api/sessions/1",
|
||||
body: `{"date":"` + dateStr + `","plan_id":2}`,
|
||||
mock: &mockSessionRepo{
|
||||
session: &models.Session{
|
||||
ID: 1,
|
||||
Date: now,
|
||||
PlanID: ptr(int64(2)),
|
||||
CreatedAt: now,
|
||||
},
|
||||
},
|
||||
wantStatus: http.StatusOK,
|
||||
},
|
||||
{
|
||||
name: "not found",
|
||||
url: "/api/sessions/999",
|
||||
body: `{"date":"` + dateStr + `"}`,
|
||||
mock: &mockSessionRepo{session: nil},
|
||||
wantStatus: http.StatusNotFound,
|
||||
},
|
||||
{
|
||||
name: "missing date",
|
||||
url: "/api/sessions/1",
|
||||
body: `{"notes":"No date"}`,
|
||||
mock: &mockSessionRepo{},
|
||||
wantStatus: http.StatusBadRequest,
|
||||
},
|
||||
{
|
||||
name: "no ID in URL",
|
||||
url: "/api/sessions/",
|
||||
body: `{"date":"` + dateStr + `"}`,
|
||||
mock: &mockSessionRepo{},
|
||||
wantStatus: http.StatusBadRequest,
|
||||
},
|
||||
{
|
||||
name: "repository error",
|
||||
url: "/api/sessions/1",
|
||||
body: `{"date":"` + dateStr + `"}`,
|
||||
mock: &mockSessionRepo{err: errors.New("db error")},
|
||||
wantStatus: http.StatusInternalServerError,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
handler := NewSessionHandler(tt.mock)
|
||||
req := httptest.NewRequest(http.MethodPut, tt.url, bytes.NewBufferString(tt.body))
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
rec := httptest.NewRecorder()
|
||||
|
||||
handler.ServeHTTP(rec, req)
|
||||
|
||||
if rec.Code != tt.wantStatus {
|
||||
t.Errorf("status = %d, want %d", rec.Code, tt.wantStatus)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSessionHandler_Delete(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
url string
|
||||
mock *mockSessionRepo
|
||||
wantStatus int
|
||||
}{
|
||||
{
|
||||
name: "successful delete",
|
||||
url: "/api/sessions/1",
|
||||
mock: &mockSessionRepo{deleteErr: nil},
|
||||
wantStatus: http.StatusNoContent,
|
||||
},
|
||||
{
|
||||
name: "not found",
|
||||
url: "/api/sessions/999",
|
||||
mock: &mockSessionRepo{deleteErr: sql.ErrNoRows},
|
||||
wantStatus: http.StatusNotFound,
|
||||
},
|
||||
{
|
||||
name: "no ID in URL",
|
||||
url: "/api/sessions/",
|
||||
mock: &mockSessionRepo{},
|
||||
wantStatus: http.StatusBadRequest,
|
||||
},
|
||||
{
|
||||
name: "repository error",
|
||||
url: "/api/sessions/1",
|
||||
mock: &mockSessionRepo{deleteErr: errors.New("db error")},
|
||||
wantStatus: http.StatusInternalServerError,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
handler := NewSessionHandler(tt.mock)
|
||||
req := httptest.NewRequest(http.MethodDelete, tt.url, nil)
|
||||
rec := httptest.NewRecorder()
|
||||
|
||||
handler.ServeHTTP(rec, req)
|
||||
|
||||
if rec.Code != tt.wantStatus {
|
||||
t.Errorf("status = %d, want %d", rec.Code, tt.wantStatus)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSessionHandler_AddEntry(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
now := time.Now()
|
||||
tests := []struct {
|
||||
name string
|
||||
url string
|
||||
body string
|
||||
mock *mockSessionRepo
|
||||
wantStatus int
|
||||
}{
|
||||
{
|
||||
name: "valid strength entry",
|
||||
url: "/api/sessions/1/entries",
|
||||
body: `{"exercise_id":1,"weight":100,"reps":10,"sets_completed":3}`,
|
||||
mock: &mockSessionRepo{
|
||||
entry: &models.SessionEntry{
|
||||
ID: 1,
|
||||
SessionID: 1,
|
||||
ExerciseID: 1,
|
||||
Weight: 100,
|
||||
Reps: 10,
|
||||
SetsCompleted: 3,
|
||||
CreatedAt: now,
|
||||
},
|
||||
},
|
||||
wantStatus: http.StatusCreated,
|
||||
},
|
||||
{
|
||||
name: "valid cardio entry",
|
||||
url: "/api/sessions/1/entries",
|
||||
body: `{"exercise_id":2,"duration":1800,"distance":5.5,"heart_rate":145}`,
|
||||
mock: &mockSessionRepo{
|
||||
entry: &models.SessionEntry{
|
||||
ID: 2,
|
||||
SessionID: 1,
|
||||
ExerciseID: 2,
|
||||
Duration: 1800,
|
||||
Distance: 5.5,
|
||||
HeartRate: 145,
|
||||
CreatedAt: now,
|
||||
},
|
||||
},
|
||||
wantStatus: http.StatusCreated,
|
||||
},
|
||||
{
|
||||
name: "missing exercise_id",
|
||||
url: "/api/sessions/1/entries",
|
||||
body: `{"weight":100,"reps":10}`,
|
||||
mock: &mockSessionRepo{},
|
||||
wantStatus: http.StatusBadRequest,
|
||||
},
|
||||
{
|
||||
name: "invalid JSON",
|
||||
url: "/api/sessions/1/entries",
|
||||
body: `{invalid}`,
|
||||
mock: &mockSessionRepo{},
|
||||
wantStatus: http.StatusBadRequest,
|
||||
},
|
||||
{
|
||||
name: "no session ID",
|
||||
url: "/api/sessions//entries",
|
||||
body: `{"exercise_id":1}`,
|
||||
mock: &mockSessionRepo{},
|
||||
wantStatus: http.StatusBadRequest,
|
||||
},
|
||||
{
|
||||
name: "repository error",
|
||||
url: "/api/sessions/1/entries",
|
||||
body: `{"exercise_id":1}`,
|
||||
mock: &mockSessionRepo{entryErr: errors.New("db error")},
|
||||
wantStatus: http.StatusInternalServerError,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
handler := NewSessionHandler(tt.mock)
|
||||
req := httptest.NewRequest(http.MethodPost, tt.url, bytes.NewBufferString(tt.body))
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
rec := httptest.NewRecorder()
|
||||
|
||||
handler.ServeHTTP(rec, req)
|
||||
|
||||
if rec.Code != tt.wantStatus {
|
||||
t.Errorf("status = %d, want %d", rec.Code, tt.wantStatus)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSessionHandler_MethodNotAllowed(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
handler := NewSessionHandler(&mockSessionRepo{})
|
||||
req := httptest.NewRequest(http.MethodPatch, "/api/sessions/1", nil)
|
||||
rec := httptest.NewRecorder()
|
||||
|
||||
handler.ServeHTTP(rec, req)
|
||||
|
||||
if rec.Code != http.StatusMethodNotAllowed {
|
||||
t.Errorf("status = %d, want %d", rec.Code, http.StatusMethodNotAllowed)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSessionHandler_EntriesMethodNotAllowed(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
handler := NewSessionHandler(&mockSessionRepo{})
|
||||
req := httptest.NewRequest(http.MethodGet, "/api/sessions/1/entries", nil)
|
||||
rec := httptest.NewRecorder()
|
||||
|
||||
handler.ServeHTTP(rec, req)
|
||||
|
||||
if rec.Code != http.StatusMethodNotAllowed {
|
||||
t.Errorf("status = %d, want %d", rec.Code, http.StatusMethodNotAllowed)
|
||||
}
|
||||
}
|
||||
|
||||
// ptr is a helper to create a pointer to a value
|
||||
func ptr[T any](v T) *T {
|
||||
return &v
|
||||
}
|
||||
Reference in New Issue
Block a user