wp-plugin-go/docs/docs_test.go

package docs

import (
	"bytes"
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/urfave/cli/v2"
)

func testApp() *cli.App {
	app := &cli.App{
		Name:        "test",
		Description: "test description",
		Flags: []cli.Flag{
			&cli.Int64Flag{
				Name:     "dummy-flag-int",
				Usage:    "dummy int flag desc",
				EnvVars:  []string{"PLUGIN_DUMMY_FLAG_INT"},
				Value:    10,
				Required: true,
			},
			&cli.StringFlag{
				Name:     "dummy-flag",
				Usage:    "Dummy flag desc.",
				EnvVars:  []string{"PLUGIN_DUMMY_FLAG"},
				Value:    "test",
				Required: true,
			},
			&cli.StringFlag{
				Name:    "simpe-flag",
				EnvVars: []string{"PLUGIN_X_SIMPLE_FLAG"},
			},
			&cli.StringFlag{
				Name:    "other.flag",
				Usage:   "other flag with desc",
				EnvVars: []string{"PLUGIN_Z_OTHER_FLAG"},
			},
			&cli.StringSliceFlag{
				Name:    "slice.flag",
				Usage:   "slice flag",
				EnvVars: []string{"PLUGIN_SLICE_FLAG"},
			},
			&cli.StringFlag{
				Name:    "hidden.flag",
				Usage:   "hidden flag",
				EnvVars: []string{"HIDDEN_FLAG", "PLUGIN_HIDDEN_FLAG"},
			},
		},
	}

	return app
}

func testFileContent(t *testing.T, file string) string {
	t.Helper()

	data, err := os.ReadFile(file)
	if err != nil {
		t.Error(err)
	}

	data = bytes.ReplaceAll(data, []byte("\r\n"), []byte("\n"))

	return string(data)
}

func TestToMarkdownFull(t *testing.T) {
	tests := []struct {
		name string
		app  *cli.App
		want string
	}{
		{
			"normal branch",
			testApp(),
			"testdata/expected-doc-full.md",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			want := testFileContent(t, tt.want)
			got, _ := ToMarkdown(tt.app)
			assert.Equal(t, want, got)
		})
	}
}

func TestToData(t *testing.T) {
	tests := []struct {
		name string
		app  *cli.App
		want *CliTemplate
	}{
		{
			"normal branch",
			testApp(),
			&CliTemplate{
				Name:        "test",
				Description: "test description",
				GlobalArgs: []*PluginArg{
					{
						Name:        "dummy_flag",
						Description: "Dummy flag desc.",
						Default:     "\"test\"",
						Type:        "string",
						Required:    true,
					},
					{
						Name:        "dummy_flag_int",
						Description: "dummy int flag desc",
						Default:     "10",
						Type:        "integer",
						Required:    true,
					},
					{
						Name:        "slice_flag",
						Description: "slice flag",
						Default:     "",
						Type:        "list",
						Required:    false,
					},
					{
						Name:     "x_simple_flag",
						Type:     "string",
						Required: false,
					},
					{
						Name:        "z_other_flag",
						Description: "other flag with desc",
						Type:        "string",
						Required:    false,
					},
				},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := GetTemplateData(tt.app)
			assert.Equal(t, tt.want, got)
		})
	}
}