assign_stmt_test.go

package goast

import (
	"go/ast"
	"go/parser"
	"go/token"
	"testing"

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

func TestAssignStmt(t *testing.T) {
	fset := token.NewFileSet()
	f, err := parser.ParseFile(fset, "./assign_stmt.go", nil, parser.ParseComments)
	if err != nil {
		t.Fatal(err)
	}

	actuals := []*ast.AssignStmt{}

	ast.Inspect(f, func(n ast.Node) bool {
		a, ok := n.(*ast.AssignStmt)
		if ok {
			actuals = append(actuals, a)
		}

		return true
	})

	// a, b := 10, input

	// lhs
	assert.Equal(t, "a", actuals[0].Lhs[0].(*ast.Ident).Name)
	assert.Equal(t, "b", actuals[0].Lhs[1].(*ast.Ident).Name)

	// token
	assert.Equal(t, token.DEFINE, actuals[0].Tok)

	// rhs
	assert.Equal(t, token.INT, actuals[0].Rhs[0].(*ast.BasicLit).Kind)
	assert.Equal(t, "10", actuals[0].Rhs[0].(*ast.BasicLit).Value)
	assert.Equal(t, "input", actuals[0].Rhs[1].(*ast.Ident).Name)

	// c, d = a, b
	// lhs
	assert.Equal(t, "c", actuals[1].Lhs[0].(*ast.Ident).Name)
	assert.Equal(t, "d", actuals[1].Lhs[1].(*ast.Ident).Name)

	// token
	assert.Equal(t, token.ASSIGN, actuals[1].Tok)

	// rhs
	assert.Equal(t, "a", actuals[1].Rhs[0].(*ast.Ident).Name)
	assert.Equal(t, "b", actuals[1].Rhs[1].(*ast.Ident).Name)
}