go-zero/tools/goctl/api/format/format_test.go

package format

import (
	"testing"

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

const (
	notFormattedStr = `
type Request struct {
  Name string 
}

type Response struct {
  Message string
}

service A-api {
@server(
handler: GreetHandler
  )
  get /greet/from/:name(Request) returns (Response)
}
`

	formattedStr = `type Request {
	Name string
}

type Response {
	Message string
}

service A-api {
	@server(
		handler: GreetHandler
	)
	get /greet/from/:name(Request) returns (Response)
}`
)

func TestFormat(t *testing.T) {
	r, err := apiFormat(notFormattedStr)
	assert.Nil(t, err)
	assert.Equal(t, r, formattedStr)
}