Add integration tests for HTTP layer
- Create test server setup with in-memory SQLite - Add auth flow tests (register, login, validation) - Add habit CRUD tests (create, read, update, archive) - Add habit entries tests (mark, unmark, get entries) - Test authorization and cross-user access control
This commit is contained in:
@@ -0,0 +1,160 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestHabitCRUDFlow(t *testing.T) {
|
||||
ts := setupTestServer(t)
|
||||
defer ts.Close()
|
||||
|
||||
registerBody := RegisterRequest{
|
||||
Email: "habituser@example.com",
|
||||
Password: "password123",
|
||||
Timezone: "UTC",
|
||||
}
|
||||
rr := makeRequest(t, *ts.Router, "POST", "/api/v1/auth/register", registerBody, "")
|
||||
var authResp AuthResponse
|
||||
decodeResponse(t, rr, &authResp)
|
||||
token := authResp.Token
|
||||
|
||||
var habitID string
|
||||
|
||||
t.Run("Create habit", func(t *testing.T) {
|
||||
reqBody := CreateHabitRequest{
|
||||
Name: "Exercise",
|
||||
Description: "Daily workout",
|
||||
Type: "BOOLEAN",
|
||||
Frequency: "DAILY",
|
||||
CarryOver: false,
|
||||
}
|
||||
|
||||
rr := makeRequest(t, *ts.Router, "POST", "/api/v1/habits", reqBody, token)
|
||||
|
||||
if rr.Code != http.StatusCreated {
|
||||
t.Fatalf("Expected status 201, got %d. Body: %s", rr.Code, rr.Body.String())
|
||||
}
|
||||
|
||||
var resp map[string]string
|
||||
decodeResponse(t, rr, &resp)
|
||||
|
||||
habitID = resp["id"]
|
||||
if habitID == "" {
|
||||
t.Fatal("Expected habit ID in response")
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("Create habit without auth", func(t *testing.T) {
|
||||
reqBody := CreateHabitRequest{
|
||||
Name: "No Auth",
|
||||
Type: "BOOLEAN",
|
||||
Frequency: "DAILY",
|
||||
}
|
||||
|
||||
rr := makeRequest(t, *ts.Router, "POST", "/api/v1/habits", reqBody, "")
|
||||
|
||||
if rr.Code != http.StatusUnauthorized {
|
||||
t.Errorf("Expected status 401, got %d", rr.Code)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("Get all user habits", func(t *testing.T) {
|
||||
rr := makeRequest(t, *ts.Router, "GET", "/api/v1/habits", nil, token)
|
||||
|
||||
if rr.Code != http.StatusOK {
|
||||
t.Fatalf("Expected status 200, got %d", rr.Code)
|
||||
}
|
||||
|
||||
var habits []UserHabitResponse
|
||||
decodeResponse(t, rr, &habits)
|
||||
|
||||
if len(habits) != 1 {
|
||||
t.Errorf("Expected 1 habit, got %d", len(habits))
|
||||
}
|
||||
|
||||
if habits[0].Name != "Exercise" {
|
||||
t.Errorf("Expected habit name 'Exercise', got '%s'", habits[0].Name)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("Get habit by ID", func(t *testing.T) {
|
||||
rr := makeRequest(t, *ts.Router, "GET", "/api/v1/habits/"+habitID, nil, token)
|
||||
|
||||
if rr.Code != http.StatusOK {
|
||||
t.Fatalf("Expected status 200, got %d", rr.Code)
|
||||
}
|
||||
|
||||
var habit UserHabitResponse
|
||||
decodeResponse(t, rr, &habit)
|
||||
|
||||
if habit.Name != "Exercise" {
|
||||
t.Errorf("Expected habit name 'Exercise', got '%s'", habit.Name)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("Update habit", func(t *testing.T) {
|
||||
reqBody := UpdateHabitRequest{
|
||||
Name: "Morning Exercise",
|
||||
Description: "Updated description",
|
||||
}
|
||||
|
||||
rr := makeRequest(t, *ts.Router, "PUT", "/api/v1/habits/"+habitID, reqBody, token)
|
||||
|
||||
if rr.Code != http.StatusOK {
|
||||
t.Fatalf("Expected status 200, got %d. Body: %s", rr.Code, rr.Body.String())
|
||||
}
|
||||
|
||||
rr = makeRequest(t, *ts.Router, "GET", "/api/v1/habits/"+habitID, nil, token)
|
||||
var habit UserHabitResponse
|
||||
decodeResponse(t, rr, &habit)
|
||||
|
||||
if habit.Name != "Morning Exercise" {
|
||||
t.Errorf("Expected updated name 'Morning Exercise', got '%s'", habit.Name)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("Archive habit", func(t *testing.T) {
|
||||
rr := makeRequest(t, *ts.Router, "DELETE", "/api/v1/habits/"+habitID, nil, token)
|
||||
|
||||
if rr.Code != http.StatusOK {
|
||||
t.Fatalf("Expected status 200, got %d", rr.Code)
|
||||
}
|
||||
|
||||
rr = makeRequest(t, *ts.Router, "GET", "/api/v1/habits", nil, token)
|
||||
var habits []UserHabitResponse
|
||||
decodeResponse(t, rr, &habits)
|
||||
|
||||
if len(habits) != 0 {
|
||||
t.Errorf("Expected 0 active habits after archive, got %d", len(habits))
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("Access other user's habit", func(t *testing.T) {
|
||||
registerBody := RegisterRequest{
|
||||
Email: "otheruser@example.com",
|
||||
Password: "password123",
|
||||
Timezone: "UTC",
|
||||
}
|
||||
rr := makeRequest(t, *ts.Router, "POST", "/api/v1/auth/register", registerBody, "")
|
||||
var authResp AuthResponse
|
||||
decodeResponse(t, rr, &authResp)
|
||||
otherToken := authResp.Token
|
||||
|
||||
reqBody := CreateHabitRequest{
|
||||
Name: "Other User Habit",
|
||||
Type: "BOOLEAN",
|
||||
Frequency: "DAILY",
|
||||
}
|
||||
rr = makeRequest(t, *ts.Router, "POST", "/api/v1/habits", reqBody, otherToken)
|
||||
var createResp map[string]string
|
||||
decodeResponse(t, rr, &createResp)
|
||||
otherHabitID := createResp["id"]
|
||||
|
||||
rr = makeRequest(t, *ts.Router, "GET", "/api/v1/habits/"+otherHabitID, nil, token)
|
||||
|
||||
if rr.Code != http.StatusForbidden {
|
||||
t.Errorf("Expected status 403, got %d", rr.Code)
|
||||
}
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user