astree_test.go

package astree

import (
	"bytes"
	"go/parser"
	"go/token"
	"strings"
	"testing"
)

func TestInternalTree(t *testing.T) {
	src := `package main

import "fmt"

var foo int

func main() {
	foo = 1
	fmt.Println(foo)
}
`

	fset := token.NewFileSet() // positions are relative to fset
	f, err := parser.ParseFile(fset, "src.go", src, 0)
	if err != nil {
		t.Fatal(err)
	}
	b := &bytes.Buffer{}
	tree(b, fset, "", []string{"", ""}, f)
	// its ok if no panic by invalid memory address error
}

func TestTree_Generics(t *testing.T) {
	src := `package main

func main() {
	ints := map[string]int64{
		"first": 34,
		"second": 12,
	}
	v := SumIntsOrFloats[string, int64](ints)
	println(v)
}

func SumIntsOrFloats[K comparable, V int64 | float64](m map[K]V) V {
	var s V
	for _, v := range m {
		s += v
	}
	return s
}
`

	fset := token.NewFileSet() // positions are relative to fset
	f, err := parser.ParseFile(fset, "src.go", src, 0)
	if err != nil {
		t.Fatal(err)
	}
	b := &bytes.Buffer{}
	tree(b, fset, "", []string{"", ""}, f)
	// its ok if no panic by invalid memory address error
}

func TestFile_NilFileSet(t *testing.T) {
	src := `package main

import "fmt"

var foo int

func main() {
	foo = 1
	fmt.Println(foo)
}
`

	fset := token.NewFileSet() // positions are relative to fset
	f, err := parser.ParseFile(fset, "src.go", src, 0)
	if err != nil {
		t.Fatal(err)
	}
	b := &bytes.Buffer{}
	err = File(b, nil, f)
	if err == nil {
		t.Error("expected to cause error")
	}
	if !strings.Contains(err.Error(), "*token.FileSet is nil") {
		t.Errorf("error message is not expected. actual=%s", err.Error())
	}
}