supermarket-api/internal/http/produce_test.go

package http

import (
	"context"
	"errors"
	"fmt"
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"

	"github.com/Rhymond/go-money"
	"github.com/davidlick/supermarket-api/internal/produce"
	"github.com/go-chi/chi"
	"github.com/golang/mock/gomock"
	"github.com/sirupsen/logrus"
	"github.com/stretchr/testify/assert"
)

func TestServer_handleAddProduce(t *testing.T) {
	tests := []struct {
		test       string
		body       string
		expectFunc func(mockProduceSvc *MockProduceService)
		assertFunc func(t *testing.T, w *httptest.ResponseRecorder)
	}{
		{
			test: "it should successfully add valid produce",
			body: `[{"code":"test","Name":"test","price":{"amount":101,"currency":"USD"}}]`,
			expectFunc: func(mockProduceSvc *MockProduceService) {
				mockProduceSvc.EXPECT().Add([]produce.Item{
					{
						Code:  "test",
						Name:  "test",
						Price: money.New(101, "USD"),
					},
				}).Return(nil)
			},
			assertFunc: func(t *testing.T, w *httptest.ResponseRecorder) {
				assert.Equal(t, http.StatusCreated, w.Code)
			},
		},
		{
			test:       "it should respond bad request if body isn't an array of produce items",
			body:       ``,
			expectFunc: func(mockProduceSvc *MockProduceService) {},
			assertFunc: func(t *testing.T, w *httptest.ResponseRecorder) {
				assert.Equal(t, http.StatusBadRequest, w.Code)
			},
		},
		{
			test: "it should respond internal server error if adding to service fails",
			body: `[{"code":"test","Name":"test","price":{"amount":101,"currency":"USD"}}]`,
			expectFunc: func(mockProduceSvc *MockProduceService) {
				mockProduceSvc.EXPECT().Add(gomock.Any()).Return(errors.New("test error"))
			},
			assertFunc: func(t *testing.T, w *httptest.ResponseRecorder) {
				assert.Equal(t, http.StatusInternalServerError, w.Code)
			},
		},
	}

	for _, tc := range tests {
		t.Run(tc.test, func(t *testing.T) {
			ctrl := gomock.NewController(t)
			defer ctrl.Finish()

			r := httptest.NewRequest(http.MethodPost, "/v1/produce", strings.NewReader(tc.body))
			w := httptest.NewRecorder()

			noopLogger := logrus.New()
			noopLogger.SetOutput(ioutil.Discard)

			mockProduceSvc := NewMockProduceService(ctrl)
			tc.expectFunc(mockProduceSvc)

			s := NewServer(3000, noopLogger, "test", mockProduceSvc)

			handler := http.HandlerFunc(s.handleAddProduce)
			handler.ServeHTTP(w, r)

			tc.assertFunc(t, w)
		})
	}
}

func TestServer_handleGetAllProduce(t *testing.T) {
	tests := []struct {
		test       string
		expectFunc func(mockProduceSvc *MockProduceService)
		assertFunc func(t *testing.T, w *httptest.ResponseRecorder)
	}{
		{
			test: "it should successfully get all produce",
			expectFunc: func(mockProduceSvc *MockProduceService) {
				mockProduceSvc.EXPECT().All().Return([]produce.Item{
					{Code: "code-1", Name: "name-1", Price: money.New(101, "USD")},
					{Code: "code-2", Name: "name-2", Price: money.New(202, "USD")},
					{Code: "code-3", Name: "name-3", Price: money.New(303, "USD")},
				}, nil)
			},
			assertFunc: func(t *testing.T, w *httptest.ResponseRecorder) {
				assert.Equal(t, http.StatusOK, w.Code)

				b, err := ioutil.ReadAll(w.Body)
				if err != nil {
					t.Error(err)
				}

				assert.Equal(t, "[{\"code\":\"code-1\",\"name\":\"name-1\",\"price\":{\"amount\":101,\"currency\":\"USD\"}},{\"code\":\"code-2\",\"name\":\"name-2\",\"price\":{\"amount\":202,\"currency\":\"USD\"}},{\"code\":\"code-3\",\"name\":\"name-3\",\"price\":{\"amount\":303,\"currency\":\"USD\"}}]\n", string(b))
			},
		},
		{
			test: "it should respond internal server error if adding to service fails",
			expectFunc: func(mockProduceSvc *MockProduceService) {
				mockProduceSvc.EXPECT().All().Return(nil, errors.New("test error"))
			},
			assertFunc: func(t *testing.T, w *httptest.ResponseRecorder) {
				assert.Equal(t, http.StatusInternalServerError, w.Code)
			},
		},
	}

	for _, tc := range tests {
		t.Run(tc.test, func(t *testing.T) {
			ctrl := gomock.NewController(t)
			defer ctrl.Finish()

			r := httptest.NewRequest(http.MethodGet, "/v1/produce", nil)
			w := httptest.NewRecorder()

			noopLogger := logrus.New()
			noopLogger.SetOutput(ioutil.Discard)

			mockProduceSvc := NewMockProduceService(ctrl)
			tc.expectFunc(mockProduceSvc)

			s := NewServer(3000, noopLogger, "test", mockProduceSvc)

			handler := http.HandlerFunc(s.handleGetAllProduce)
			handler.ServeHTTP(w, r)

			tc.assertFunc(t, w)
		})
	}
}

func TestServer_handleDeleteProduce(t *testing.T) {
	tests := []struct {
		test        string
		produceCode string
		expectFunc  func(mockProduceSvc *MockProduceService)
		assertFunc  func(t *testing.T, w *httptest.ResponseRecorder)
	}{
		{
			test:        "it should respond no content when successful",
			produceCode: "test-code",
			expectFunc: func(mockProduceSvc *MockProduceService) {
				mockProduceSvc.EXPECT().Remove(produce.Item{Code: "test-code"}).Return(nil)
			},
			assertFunc: func(t *testing.T, w *httptest.ResponseRecorder) {
				assert.Equal(t, http.StatusNoContent, w.Code)
			},
		},
		{
			test:        "it should respond internal server error if adding to service fails",
			produceCode: "test-code",
			expectFunc: func(mockProduceSvc *MockProduceService) {
				mockProduceSvc.EXPECT().Remove(produce.Item{Code: "test-code"}).Return(errors.New("test error"))
			},
			assertFunc: func(t *testing.T, w *httptest.ResponseRecorder) {
				assert.Equal(t, http.StatusInternalServerError, w.Code)
			},
		},
		{
			test:       "it should respond bad request if no produce code is supplied",
			expectFunc: func(mockProduceSvc *MockProduceService) {},
			assertFunc: func(t *testing.T, w *httptest.ResponseRecorder) {
				assert.Equal(t, http.StatusBadRequest, w.Code)
			},
		},
	}

	for _, tc := range tests {
		t.Run(tc.test, func(t *testing.T) {
			ctrl := gomock.NewController(t)
			defer ctrl.Finish()

			r := httptest.NewRequest(http.MethodDelete, fmt.Sprintf("/v1/produce/%s", tc.produceCode), nil)
			rctx := chi.NewRouteContext()
			rctx.URLParams.Add("produceCode", tc.produceCode)
			ctx := context.WithValue(context.Background(), chi.RouteCtxKey, rctx)

			r = r.WithContext(ctx)
			w := httptest.NewRecorder()

			noopLogger := logrus.New()
			noopLogger.SetOutput(ioutil.Discard)

			mockProduceSvc := NewMockProduceService(ctrl)
			tc.expectFunc(mockProduceSvc)

			s := NewServer(3000, noopLogger, "test", mockProduceSvc)

			handler := http.HandlerFunc(s.handleDeleteProduce)
			handler.ServeHTTP(w, r)

			tc.assertFunc(t, w)
		})
	}
}