Implement Application Layer with TDD

- Add date utilities for habit scheduling logic - Implement CreateHabitHandler with full validation - Implement GetTodaysHabitsHandler with carry-over support - Implement MarkHabitHandler for completing habits - All components developed following TDD methodology - Complete test coverage for commands and queries
2025-11-25 23:54:49 +01:00
parent 0977d3a58b
commit 34ea1f718e
8 changed files with 1002 additions and 0 deletions
@@ -0,0 +1,197 @@
+package commands
+
+import (
+	"context"
+	"testing"
+
+	"habit-tracker-api/internal/domain/entities"
+	"habit-tracker-api/internal/shared/errors"
+)
+
+type mockHabitRepo struct {
+	createFunc func(ctx context.Context, habit *entities.Habit) error
+}
+
+func (m *mockHabitRepo) Create(ctx context.Context, habit *entities.Habit) error {
+	return m.createFunc(ctx, habit)
+}
+
+func (m *mockHabitRepo) FindByID(ctx context.Context, id string) (*entities.Habit, error) {
+	return nil, nil
+}
+
+func (m *mockHabitRepo) FindByUserID(ctx context.Context, userID string) ([]*entities.Habit, error) {
+	return nil, nil
+}
+
+func (m *mockHabitRepo) FindActiveByUserID(ctx context.Context, userID string) ([]*entities.Habit, error) {
+	return nil, nil
+}
+
+func (m *mockHabitRepo) Update(ctx context.Context, habit *entities.Habit) error {
+	return nil
+}
+
+func (m *mockHabitRepo) Delete(ctx context.Context, id string) error {
+	return nil
+}
+
+func TestCreateHabitHandler_Success(t *testing.T) {
+	mock := &mockHabitRepo{
+		createFunc: func(ctx context.Context, habit *entities.Habit) error {
+			habit.ID = "habit-123"
+			return nil
+		},
+	}
+
+	handler := NewCreateHabitHandler(mock)
+
+	cmd := CreateHabitCommand{
+		UserID:      "user-123",
+		Name:        "Exercise",
+		Description: "Daily workout",
+		Type:        "BOOLEAN",
+		Frequency:   "DAILY",
+		CarryOver:   true,
+	}
+
+	habitID, err := handler.Handle(context.Background(), cmd)
+
+	if err != nil {
+		t.Fatalf("Expected no error, got %v", err)
+	}
+
+	if habitID == "" {
+		t.Error("Expected habit ID to be returned")
+	}
+}
+
+func TestCreateHabitHandler_InvalidType(t *testing.T) {
+	mock := &mockHabitRepo{}
+	handler := NewCreateHabitHandler(mock)
+
+	cmd := CreateHabitCommand{
+		UserID:    "user-123",
+		Name:      "Exercise",
+		Type:      "INVALID",
+		Frequency: "DAILY",
+	}
+
+	_, err := handler.Handle(context.Background(), cmd)
+
+	if err != errors.ErrInvalidInput {
+		t.Errorf("Expected ErrInvalidInput, got %v", err)
+	}
+}
+
+func TestCreateHabitHandler_InvalidFrequency(t *testing.T) {
+	mock := &mockHabitRepo{}
+	handler := NewCreateHabitHandler(mock)
+
+	cmd := CreateHabitCommand{
+		UserID:    "user-123",
+		Name:      "Exercise",
+		Type:      "BOOLEAN",
+		Frequency: "INVALID",
+	}
+
+	_, err := handler.Handle(context.Background(), cmd)
+
+	if err != errors.ErrInvalidInput {
+		t.Errorf("Expected ErrInvalidInput, got %v", err)
+	}
+}
+
+func TestCreateHabitHandler_WeeklyWithoutSpecificDays(t *testing.T) {
+	mock := &mockHabitRepo{}
+	handler := NewCreateHabitHandler(mock)
+
+	cmd := CreateHabitCommand{
+		UserID:       "user-123",
+		Name:         "Exercise",
+		Type:         "BOOLEAN",
+		Frequency:    "WEEKLY",
+		SpecificDays: []int{},
+	}
+
+	_, err := handler.Handle(context.Background(), cmd)
+
+	if err != errors.ErrInvalidInput {
+		t.Errorf("Expected ErrInvalidInput, got %v", err)
+	}
+}
+
+func TestCreateHabitHandler_MonthlyWithoutSpecificDates(t *testing.T) {
+	mock := &mockHabitRepo{}
+	handler := NewCreateHabitHandler(mock)
+
+	cmd := CreateHabitCommand{
+		UserID:        "user-123",
+		Name:          "Exercise",
+		Type:          "BOOLEAN",
+		Frequency:     "MONTHLY",
+		SpecificDates: []int{},
+	}
+
+	_, err := handler.Handle(context.Background(), cmd)
+
+	if err != errors.ErrInvalidInput {
+		t.Errorf("Expected ErrInvalidInput, got %v", err)
+	}
+}
+
+func TestCreateHabitHandler_WeeklyWithSpecificDays(t *testing.T) {
+	mock := &mockHabitRepo{
+		createFunc: func(ctx context.Context, habit *entities.Habit) error {
+			habit.ID = "habit-123"
+			if len(habit.SpecificDays) != 3 {
+				t.Errorf("Expected 3 specific days, got %d", len(habit.SpecificDays))
+			}
+			return nil
+		},
+	}
+
+	handler := NewCreateHabitHandler(mock)
+
+	cmd := CreateHabitCommand{
+		UserID:       "user-123",
+		Name:         "Exercise",
+		Type:         "BOOLEAN",
+		Frequency:    "WEEKLY",
+		SpecificDays: []int{1, 3, 5},
+	}
+
+	_, err := handler.Handle(context.Background(), cmd)
+
+	if err != nil {
+		t.Fatalf("Expected no error, got %v", err)
+	}
+}
+
+func TestCreateHabitHandler_MonthlyWithSpecificDates(t *testing.T) {
+	mock := &mockHabitRepo{
+		createFunc: func(ctx context.Context, habit *entities.Habit) error {
+			habit.ID = "habit-123"
+			if len(habit.SpecificDates) != 2 {
+				t.Errorf("Expected 2 specific dates, got %d", len(habit.SpecificDates))
+			}
+			return nil
+		},
+	}
+
+	handler := NewCreateHabitHandler(mock)
+
+	cmd := CreateHabitCommand{
+		UserID:        "user-123",
+		Name:          "Pay bills",
+		Type:          "BOOLEAN",
+		Frequency:     "MONTHLY",
+		SpecificDates: []int{1, 15},
+	}
+
+	_, err := handler.Handle(context.Background(), cmd)
+
+	if err != nil {
+		t.Fatalf("Expected no error, got %v", err)
+	}
+}