swagger-api/protoc-gen-openapiv2/generator/generator_test.go

package generator

import (
	"fmt"
	"os"
	"testing"

	"git.ikuban.com/server/swagger-api/protoc-gen-openapiv2/internal/codegenerator"
)

func Test_Gen(t *testing.T) {
	var err error
	f, err := os.Open("./req.bin")
	if err != nil {
		t.Fatal(err)
	}
	req, err := codegenerator.ParseRequest(f)
	if err != nil {
		t.Fatal(err)
	}
	var g Generator

	resp, err := g.Gen(req, false)
	if err != nil {
		t.Fatal(err)
	}
	for _, file := range resp.File {
		fmt.Println(*file.Content)
	}
}
func Test_NewGenerator(t *testing.T) {
	var err error
	f, err := os.Open("./req.bin")
	if err != nil {
		t.Fatal(err)
	}
	req, err := codegenerator.ParseRequest(f)
	if err != nil {
		t.Fatal(err)
	}

	g := NewGenerator(
		DisableDefaultErrors(true),
		UseJSONNamesForFields(false),
		EnumsAsInts(true),
		RecursiveDepth(2048),
	)

	resp, err := g.Gen(req, false)
	if err != nil {
		t.Fatal(err)
	}
	for _, file := range resp.File {
		fmt.Println(*file.Content)
	}
}