apocapoc-api/internal/application/commands/create_habit_test.go

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)
	}
}