response_test.go

package web

import (
	"testing"

	assertion "github.com/stretchr/testify/assert"
)

func TestResponse(t *testing.T) {
	assert := assertion.New(t)

	t.Run("successResponse", func(t *testing.T) {
		t.Run("should return success response", func(t *testing.T) {
			result := successResponse()

			assert.IsType(JSONResponse{}, result)
			assert.Equal(result.Success, true, "they should be equal")
			assert.Equal(result.Error, "", "they should be equal")
		})

		t.Run("should return success response with custom message", func(t *testing.T) {
			result := successResponse("test")

			assert.IsType(JSONResponse{}, result)
			assert.Equal(result.Success, true, "they should be equal")
			assert.Equal(result.Error, "", "they should be equal")
			assert.Equal(result.Message, "test", "they should be equal")
		})
	})

	t.Run("errorResponse", func(t *testing.T) {
		t.Run("should return error response", func(t *testing.T) {
			result := errorResponse()

			assert.IsType(JSONResponse{}, result)
			assert.Equal(result.Success, false, "they should be equal")
			assert.Equal(result.Error, "An error occurred", "they should be equal")
		})

		t.Run("should return error response with custom message", func(t *testing.T) {
			result := errorResponse("test")

			assert.IsType(JSONResponse{}, result)
			assert.Equal(result.Success, false, "they should be equal")
			assert.Equal(result.Error, "test", "they should be equal")
		})
	})
}