handler_test.go

package main

import (
	"context"
	"errors"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	"github.com/go-chi/chi/v5"
)

type MockStorage struct {
	shouldFail bool // должна ли быть ошибка: true/false
	tasks      []Task
}

func (m *MockStorage) UpdateTask(ctx context.Context, id int, title string, done bool) (Task, error) {
	if m.shouldFail {
		return Task{}, errors.New("ошибка в базе данных")
	}

	return Task{}, nil
}

func (m *MockStorage) PrintAllTasks(ctx context.Context) ([]Task, error) {
	if m.shouldFail {
		return nil, errors.New("ошибка в базе данных")
	}

	return m.tasks, nil
}

func (m *MockStorage) DeleteTask(ctx context.Context, id int) error {
	if m.shouldFail {
		return errors.New("ошибка в базе данных")
	}

	return nil
}

func (m *MockStorage) CreateTask(ctx context.Context, title string) (Task, error) {
	if m.shouldFail {
		return Task{}, errors.New("ошибка в базе данных")
	}

	return Task{ID: 1, Title: title, Done: false}, nil
}

func (m *MockStorage) PrintTask(ctx context.Context, id int) (Task, error) {
	if m.shouldFail {
		return Task{}, errors.New("ошибка в базе данных")
	}

	return Task{}, nil
}

func TestHandlerUpdateTask(t *testing.T) {
	tests := []struct {
		name       string
		id         string
		shouldFail bool
		wantStatus int
	}{
		{
			name:       "успешное обновление",
			id:         "1",
			shouldFail: false,
			wantStatus: http.StatusNoContent,
		},
		{
			name:       "ошибка хранилища",
			id:         "1",
			shouldFail: true,
			wantStatus: http.StatusBadRequest,
		},
		{
			name:       "невалидный id",
			id:         "abc",
			shouldFail: false,
			wantStatus: http.StatusBadRequest,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			mock := &MockStorage{shouldFail: tt.shouldFail}
			handler := &Handler{storage: mock}

			req := httptest.NewRequest("DELETE", "/tasks/"+tt.id, nil)
			rctx := chi.NewRouteContext()
			rctx.URLParams.Add("id", tt.id)
			req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))

			w := httptest.NewRecorder()
			handler.handlerUpdateTask(w, req)

			if w.Code != tt.wantStatus {
				t.Errorf("ожидал %d, получил %d", tt.wantStatus, w.Code)
			}
		})
	}
}

func TestHandlerPrintAllTasks(t *testing.T) {
	tests := []struct {
		name       string
		tasks      []Task
		shouldFail bool
		wantStatus int
	}{
		{
			name:       "успешное получение списка",
			tasks:      []Task{{ID: 1, Title: "купить молоко", Done: false}},
			shouldFail: false,
			wantStatus: http.StatusOK,
		},
		{
			name:       "пустой список",
			tasks:      []Task{},
			shouldFail: false,
			wantStatus: http.StatusOK,
		},
		{
			name:       "ошибка хранилища",
			tasks:      nil,
			shouldFail: true,
			wantStatus: http.StatusBadRequest,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			mock := &MockStorage{shouldFail: tt.shouldFail, tasks: tt.tasks}
			handler := &Handler{storage: mock}

			req := httptest.NewRequest("GET", "/tasks", nil)
			w := httptest.NewRecorder()

			handler.handlerPrintAllTasks(w, req)

			if w.Code != tt.wantStatus {
				t.Errorf("ожидал %d, получил %d", tt.wantStatus, w.Code)
			}
		})
	}
}

func TestHandlerDeleteTask(t *testing.T) {
	tests := []struct {
		name       string
		id         string
		shouldFail bool
		wantStatus int
	}{
		{
			name:       "успешное удаление",
			id:         "1",
			shouldFail: false,
			wantStatus: http.StatusNoContent,
		},
		{
			name:       "ошибка хранилища",
			id:         "1",
			shouldFail: true,
			wantStatus: http.StatusBadRequest,
		},
		{
			name:       "невалидный id",
			id:         "abc",
			shouldFail: false,
			wantStatus: http.StatusBadRequest,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			mock := &MockStorage{shouldFail: tt.shouldFail}
			handler := &Handler{storage: mock}

			req := httptest.NewRequest("DELETE", "/tasks/"+tt.id, nil)
			rctx := chi.NewRouteContext()
			rctx.URLParams.Add("id", tt.id)
			req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))

			w := httptest.NewRecorder()
			handler.handlerDeleteTask(w, req)

			if w.Code != tt.wantStatus {
				t.Errorf("ожидал %d, получил %d", tt.wantStatus, w.Code)
			}
		})
	}
}

func TestHandlerCreateTask(t *testing.T) {
	tests := []struct {
		name       string
		body       string
		shouldFail bool
		wantStatus int
	}{
		{
			name:       "успешное создание",
			body:       `{"title": "купить молоко"}`,
			shouldFail: false,
			wantStatus: http.StatusCreated,
		},
		{
			name:       "невалидный json",
			body:       `{not json}`,
			shouldFail: false,
			wantStatus: http.StatusBadRequest,
		},
		{
			name:       "ошибка хранилища",
			body:       `{"title": "купить молоко"}`,
			shouldFail: true,
			wantStatus: http.StatusInternalServerError,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			mock := &MockStorage{shouldFail: tt.shouldFail}
			handler := &Handler{storage: mock}

			req := httptest.NewRequest("POST", "/tasks", strings.NewReader(tt.body))
			w := httptest.NewRecorder()

			handler.handlerCreateTask(w, req)

			if w.Code != tt.wantStatus {
				t.Errorf("ожидал %d, получил %d", tt.wantStatus, w.Code)
			}
		})
	}
}

func TestHandlerPrintTask(t *testing.T) {
	tests := []struct {
		name       string
		id         string
		shouldFail bool
		wantStatus int
	}{
		{
			name:       "успешное получение",
			id:         "1",
			shouldFail: false,
			wantStatus: http.StatusOK,
		},
		{
			name:       "задача не найдена",
			id:         "1",
			shouldFail: true,
			wantStatus: http.StatusBadRequest,
		},
		{
			name:       "невалидный id",
			id:         "abc",
			shouldFail: false,
			wantStatus: http.StatusBadRequest,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			mock := &MockStorage{shouldFail: tt.shouldFail}
			handler := &Handler{storage: mock}

			req := httptest.NewRequest("GET", "/tasks/"+tt.id, nil)
			rctx := chi.NewRouteContext()
			rctx.URLParams.Add("id", tt.id)
			req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))

			w := httptest.NewRecorder()
			handler.handlerPrintTask(w, req)

			if w.Code != tt.wantStatus {
				t.Errorf("ожидал %d, получил %d", tt.wantStatus, w.Code)
			}
		})
	}
}