response_test.go

package easycodefgo

import (
	"fmt"
	"testing"

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

// newResponseByMessage 테스트
func TestNewResponseByMessage(t *testing.T) {
	ast := assert.New(t)

	res := newResponseByMessage(messageNotFound)
	data := res.GetData()
	ast.NotNil(data)

	result := res.GetResult()
	ast.NotNil(result)

	testResponseGetMessageInfo(ast, res, messageNotFound)
}

// Response.GetMessageInfo 테스트
func testResponseGetMessageInfo(
	ast *assert.Assertions,
	res *Response,
	m *messageConstant,
) {
	code, msg, extraMsg := res.GetMessageInfo()
	ast.Equal(m.Code, code)
	ast.Equal(m.Message, msg)
	ast.Equal(m.ExtraMessage, extraMsg)
}

// newResponseByMap 테스트
func TestNewResponseByMap(t *testing.T) {
	ast := assert.New(t)

	m := map[string]interface{}{
		KeyResult: make(map[string]interface{}),
		KeyData:   make([]map[string]interface{}, 1),
	}
	res := newResponseByMap(m)
	ast.NotNil(res)
}

// Response 인스턴스를 JSON string으로 변환 테스트
func TestWriteValueAsString(t *testing.T) {
	ast := assert.New(t)

	res := newResponseByMessage(messageOK)
	str := res.WriteValueAsString()

	// 원하는 결과
	s := fmt.Sprintf(
		`{"data":{},"result":{"code":"%s","extraMessage":"%s","message":"%s"}}`,
		messageOK.Code,
		messageOK.ExtraMessage,
		messageOK.Message,
	)
	ast.Equal(str, s)
}