github.com/vugu/vugu@v0.3.6-0.20240430171613-3f6f402e014b/gen/parser-go_test.go

package gen

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"github.com/vugu/html"
)

func TestEmitForExpr(t *testing.T) {
	tests := []struct {
		name           string
		node           *html.Node
		expectedError  string
		expectedResult string
	}{
		{
			name:          "no vg-for attributes",
			node:          &html.Node{},
			expectedError: "no for expression, code should not be calling emitForExpr when no vg-for is present",
		},
		{
			name: "no iteration vars",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for", Val: "c.Items"},
				},
			},
			expectedResult: `for key, value := range c.Items {
var vgiterkey interface{} = key
_ = vgiterkey
key := key
_ = key
value := value
_ = value
`,
		},
		{
			name: "no iteration vars noshadow",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for.noshadow", Val: "c.Items"},
				},
			},
			expectedResult: `for key, value := range c.Items {
var vgiterkey interface{} = key
_ = vgiterkey
`,
		},
		{
			name: "no iteration vars with vg-key",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for", Val: "c.Items"},
					{Key: "vg-key", Val: "1"},
				},
			},
			expectedResult: `for key, value := range c.Items {
var vgiterkey interface{} = 1
_ = vgiterkey
key := key
_ = key
value := value
_ = value
`,
		},
		{
			name: "key and value vars",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for", Val: "k, v := range c.Items"},
				},
			},
			expectedResult: `for k, v := range c.Items {
var vgiterkey interface{} = k
_ = vgiterkey
k := k
_ = k
v := v
_ = v
`,
		},
		{
			name: "key and value vars noshadow",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for.noshadow", Val: "k, v := range c.Items"},
				},
			},
			expectedResult: `for k, v := range c.Items {
var vgiterkey interface{} = k
_ = vgiterkey
`,
		},
		{
			name: "only key var",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for", Val: "k := range c.Items"},
				},
			},
			expectedResult: `for k := range c.Items {
var vgiterkey interface{} = k
_ = vgiterkey
k := k
_ = k
`,
		},
		{
			name: "only key var noshadow",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for.noshadow", Val: "k := range c.Items"},
				},
			},
			expectedResult: `for k := range c.Items {
var vgiterkey interface{} = k
_ = vgiterkey
`,
		},
		{
			name: "only key var with vg-key",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for", Val: "k := range c.Items"},
					{Key: "vg-key", Val: "1"},
				},
			},
			expectedResult: `for k := range c.Items {
var vgiterkey interface{} = 1
_ = vgiterkey
k := k
_ = k
`,
		},
		{
			name: "only value var",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for", Val: "_, v := range c.Items"},
				},
			},
			expectedResult: `for vgiterkeyt , v := range c.Items {
var vgiterkey interface{} = vgiterkeyt
_ = vgiterkey
v := v
_ = v
`,
		},
		{
			name: "only value var noshadow",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for.noshadow", Val: "_, v := range c.Items"},
				},
			},
			expectedResult: `for vgiterkeyt , v := range c.Items {
var vgiterkey interface{} = vgiterkeyt
_ = vgiterkey
`,
		},
		{
			name: "only value var with vg-key",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for", Val: "_, v := range c.Items"},
					{Key: "vg-key", Val: "1"},
				},
			},
			expectedResult: `for _, v := range c.Items {
var vgiterkey interface{} = 1
_ = vgiterkey
v := v
_ = v
`,
		},
		{
			name: "iteration with for clause",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for", Val: "i:= 0; i < 5; i++"},
				},
			},
			expectedResult: `for i:= 0; i < 5; i++ {
var vgiterkey interface{} = i
_ = vgiterkey
i := i
_ = i
`,
		},
		{
			name: "iteration with for clause noshadow",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for.noshadow", Val: "i:= 0; i < 5; i++"},
				},
			},
			expectedResult: `for i:= 0; i < 5; i++ {
var vgiterkey interface{} = i
_ = vgiterkey
`,
		},
		{
			name: "iteration with for clause with vg-key",
			node: &html.Node{
				Attr: []html.Attribute{
					{Key: "vg-for", Val: "i:= 0; i < 5; i++"},
					{Key: "vg-key", Val: "1"},
				},
			},
			expectedResult: `for i:= 0; i < 5; i++ {
var vgiterkey interface{} = 1
_ = vgiterkey
i := i
_ = i
`,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			assert := assert.New(t)
			require := require.New(t)
			pg := &ParserGo{}
			state := &parseGoState{}

			err := pg.emitForExpr(state, tt.node)

			if tt.expectedError != "" {
				require.EqualError(err, tt.expectedError)
				return
			}
			require.NoError(err)
			assert.Exactly(tt.expectedResult, state.buildBuf.String())
		})
	}
}