feat: adjust fiber route generation

pkg/swag/format/format_test.go

@@ -0,0 +1,151 @@
+package format
+
+import (
+	"bytes"
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestFormat_Format(t *testing.T) {
+	fx := setup(t)
+	assert.NoError(t, New().Build(&Config{SearchDir: fx.basedir}))
+	assert.True(t, fx.isFormatted("main.go"))
+	assert.True(t, fx.isFormatted("api/api.go"))
+}
+
+func TestFormat_PermissionsPreserved(t *testing.T) {
+	fx := setup(t)
+
+	originalFileInfo, err := os.Stat(filepath.Join(fx.basedir, "main.go"))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	assert.NoError(t, New().Build(&Config{SearchDir: fx.basedir}))
+	assert.True(t, permissionsEqual(t, filepath.Join(fx.basedir, "main.go"), originalFileInfo.Mode()))
+	assert.True(t, permissionsEqual(t, filepath.Join(fx.basedir, "api/api.go"), originalFileInfo.Mode()))
+}
+
+func TestFormat_ExcludeDir(t *testing.T) {
+	fx := setup(t)
+	assert.NoError(t, New().Build(&Config{
+		SearchDir: fx.basedir,
+		Excludes:  filepath.Join(fx.basedir, "api"),
+	}))
+	assert.False(t, fx.isFormatted("api/api.go"))
+}
+
+func TestFormat_ExcludeFile(t *testing.T) {
+	fx := setup(t)
+	assert.NoError(t, New().Build(&Config{
+		SearchDir: fx.basedir,
+		Excludes:  filepath.Join(fx.basedir, "main.go"),
+	}))
+	assert.False(t, fx.isFormatted("main.go"))
+}
+
+func TestFormat_DefaultExcludes(t *testing.T) {
+	fx := setup(t)
+	assert.NoError(t, New().Build(&Config{SearchDir: fx.basedir}))
+	assert.False(t, fx.isFormatted("api/api_test.go"))
+	assert.False(t, fx.isFormatted("docs/docs.go"))
+}
+
+func TestFormat_ParseError(t *testing.T) {
+	fx := setup(t)
+	os.WriteFile(filepath.Join(fx.basedir, "parse_error.go"), []byte(`package main
+		func invalid() {`), 0644)
+	assert.Error(t, New().Build(&Config{SearchDir: fx.basedir}))
+}
+
+func TestFormat_ReadError(t *testing.T) {
+	fx := setup(t)
+	os.Chmod(filepath.Join(fx.basedir, "main.go"), 0)
+	assert.Error(t, New().Build(&Config{SearchDir: fx.basedir}))
+}
+
+func TestFormat_WriteError(t *testing.T) {
+	fx := setup(t)
+	os.Chmod(fx.basedir, 0555)
+	assert.Error(t, New().Build(&Config{SearchDir: fx.basedir}))
+	os.Chmod(fx.basedir, 0755)
+}
+
+func TestFormat_InvalidSearchDir(t *testing.T) {
+	formatter := New()
+	assert.Error(t, formatter.Build(&Config{SearchDir: "no_such_dir"}))
+}
+
+type fixture struct {
+	t       *testing.T
+	basedir string
+}
+
+func setup(t *testing.T) *fixture {
+	fx := &fixture{
+		t:       t,
+		basedir: t.TempDir(),
+	}
+	for filename, contents := range testFiles {
+		fullpath := filepath.Join(fx.basedir, filepath.Clean(filename))
+		if err := os.MkdirAll(filepath.Dir(fullpath), 0755); err != nil {
+			t.Fatal(err)
+		}
+		if err := os.WriteFile(fullpath, contents, 0644); err != nil {
+			t.Fatal(err)
+		}
+	}
+	return fx
+}
+
+func (fx *fixture) isFormatted(file string) bool {
+	contents, err := os.ReadFile(filepath.Join(fx.basedir, file))
+	if err != nil {
+		fx.t.Fatal(err)
+	}
+	return !bytes.Equal(testFiles[file], contents)
+}
+
+func permissionsEqual(t *testing.T, path string, expectedMode os.FileMode) bool {
+	fileInfo, err := os.Stat(path)
+	if err != nil {
+		t.Fatal(err)
+	}
+	return expectedMode == fileInfo.Mode()
+}
+
+var testFiles = map[string][]byte{
+	"api/api.go": []byte(`package api
+
+		import "net/http"
+
+		// @Summary		Add a new pet to the store
+		// @Description	get string by ID
+		func GetStringByInt(w http.ResponseWriter, r *http.Request) {
+			//write your code
+		}`),
+	"api/api_test.go": []byte(`package api
+		// @Summary API Test
+		// @Description Should not be formatted
+		func TestApi(t *testing.T) {}`),
+	"docs/docs.go": []byte(`package docs
+		// @Summary Documentation package
+		// @Description Should not be formatted`),
+	"main.go": []byte(`package main
+
+		import (
+			"net/http"
+
+			"git.ipao.vip/rogeecn/atomctl/pkg/swag/format/testdata/api"
+		)
+
+		// @title Swagger Example API
+		// @version 1.0
+		func main() {
+			http.HandleFunc("/testapi/get-string-by-int/", api.GetStringByInt)
+		}`),
+	"README.md": []byte(`# Format test`),
+}