merge upstream

cl/builtin_test.go

@@ -21,11 +21,42 @@ import (
 	"go/constant"
 	"go/types"
 	"testing"
+	"unsafe"
 
 	llssa "github.com/goplus/llgo/ssa"
 	"golang.org/x/tools/go/ssa"
 )
 
+func TestIsAllocVargs(t *testing.T) {
+	if isAllocVargs(nil, ssaAlloc(&ssa.Return{})) {
+		t.Fatal("isVargs?")
+	}
+	if isAllocVargs(nil, ssaAlloc(ssaSlice(&ssa.Go{}))) {
+		t.Fatal("isVargs?")
+	}
+	if isAllocVargs(nil, ssaAlloc(ssaSlice(&ssa.Return{}))) {
+		t.Fatal("isVargs?")
+	}
+}
+
+func ssaSlice(refs ...ssa.Instruction) *ssa.Slice {
+	a := &ssa.Slice{}
+	setRefs(unsafe.Pointer(a), refs...)
+	return a
+}
+
+func ssaAlloc(refs ...ssa.Instruction) *ssa.Alloc {
+	a := &ssa.Alloc{}
+	setRefs(unsafe.Pointer(a), refs...)
+	return a
+}
+
+func setRefs(v unsafe.Pointer, refs ...ssa.Instruction) {
+	off := unsafe.Offsetof(ssa.Alloc{}.Comment) - unsafe.Sizeof([]int(nil))
+	ptr := uintptr(v) + off
+	*(*[]ssa.Instruction)(unsafe.Pointer(ptr)) = refs
+}
+
 func TestRecvTypeName(t *testing.T) {
 	if ret := recvTypeName(&ast.IndexExpr{
 		X:     &ast.Ident{Name: "Pointer"},
@@ -121,6 +152,12 @@ func TestPkgKind(t *testing.T) {
 	if v, _ := pkgKind(""); v != PkgLLGo {
 		t.Fatal("pkgKind:", v)
 	}
+	if v, _ := pkgKind("decl"); v != PkgDeclOnly {
+		t.Fatal("pkgKind:", v)
+	}
+	if v, _ := pkgKind("decl: test.ll"); v != PkgDeclOnly {
+		t.Fatal("pkgKind:", v)
+	}
 }
 
 func TestPkgKindOf(t *testing.T) {