【Golang】Goのテストを書いてみよう

func checkIntValueExample(i int64) bool {
	if 0 < i && i < 10 {
		return true
	}
	return false
}

import (
	"testing"
	"github.com/stretchr/testify/suite"
)

// test suite
type ExampleSuite struct {
	suite.Suite
}

// run test suite
func TestExampleSuite(t *testing.T) {
	suite.Run(t, new(ExampleSuite))
}

func (s *ExampleSuite) TestCheckIntValueExample() {
	cases := map[string]struct {
		i    int64
		want bool
	}{
		"0":  {0, false},
		"1":  {1, true},
		"9":  {9, true},
		"10": {10, false},
	}

	a := s.Assert()
	for name, v := range cases {
		s.Run(name, func() {
			matched := checkIntValueExample(v.i)
			a.Equal(v.want, matched)
		})
	}
}

// 指定のURLへリクエストしてレスポンス構造体を書き換える
func helloExample(url string, respbody interface{}) (int, error) {
	req, _ := http.NewRequest(http.MethodGet, url, nil)
	client := &http.Client{}
	res, err := client.Do(req)
	if err != nil {
		return http.StatusInternalServerError, err
	}
	defer res.Body.Close()

	buf := &bytes.Buffer{}
	tee := io.TeeReader(res.Body, buf)
	if err = json.NewDecoder(tee).Decode(&respbody); err != nil {
		return http.StatusInternalServerError, err
	}

	return res.StatusCode, nil
}

import (
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/gin-gonic/gin"
	"github.com/stretchr/testify/suite"
)

// test suite
type HelloExampleSuite struct {
	suite.Suite
	testServer *httptest.Server
	ctx        *gin.Context
	router     *gin.Engine
}

// run test suite
func TestHelloExampleSuite(t *testing.T) {
	suite.Run(t, new(HelloExampleSuite))
}

// handler
type helloResp struct {
	Message string `json:"message"`
}

func testHelloHandler(c *gin.Context) {
	resp := helloResp{
		Message: "hello",
	}
	c.JSON(http.StatusOK, resp)
}


// setUp, tearDown
func (s *HelloExampleSuite) SetupSuite() {
	s.ctx, s.router = gin.CreateTestContext(httptest.NewRecorder())
	s.ctx.Request = httptest.NewRequest("GET", "/", nil)
	s.router.GET("/get", testHelloHandler)
	s.testServer = httptest.NewServer(s.router)
}

func (s *HelloExampleSuite) TearDownSuite() {
	s.testServer.Close()
}

// test
func (s *HelloExampleSuite) TestHelloExample() {
	cases := map[string]struct {
		url string
		status int
		message string
	}{
		"statusOK":  {"/get", 200, "hello"},
	}

	a := s.Assert()
	for name, v := range cases {
		s.Run(name, func() {
			var resp helloResp
			httpStatus, err := helloExample(s.testServer.URL+v.url, &resp)
			if err != nil {
				s.T().Fatal(err.Error())
			}
			a.Equal(v.status, httpStatus)
			a.Equal(v.message, resp.Message)
		})
	}
}