goscript 0.0.83 → 0.1.0

package/compiler/analysis_test.go

@@ -1,338 +0,0 @@
-package compiler
-
-import (
-	"go/ast"
-	"go/parser"
-	"go/token"
-	"go/types"
-	"os"
-	"slices"
-	"testing"
-
-	"golang.org/x/tools/go/packages"
-)
-
-// TestAnalysisVarRefLogic verifies that the analysis correctly identifies
-// which variables need variable references based on actual tests test cases
-func TestAnalysisVarRefLogic(t *testing.T) {
-	tests := []struct {
-		name     string
-		code     string
-		expected map[string]AnalysisExpectation
-	}{
-		{
-			name: "pointer_range_loop",
-			code: `package main
-func main() {
-	arr := [3]int{1, 2, 3}
-	arrPtr := &arr
-	for i, v := range arrPtr {
-		println("index:", i, "value:", v)
-	}
-}`,
-			expected: map[string]AnalysisExpectation{
-				"arr":    {NeedsVarRef: true, NeedsVarRefAccess: true},   // varrefed because &arr is taken
-				"arrPtr": {NeedsVarRef: false, NeedsVarRefAccess: true},  // NOT varrefed, but points to varrefed value
-				"i":      {NeedsVarRef: false, NeedsVarRefAccess: false}, // regular loop variable
-				"v":      {NeedsVarRef: false, NeedsVarRefAccess: false}, // regular loop variable
-			},
-		},
-		{
-			name: "simple_pointers",
-			code: `package main
-type MyStruct struct {
-	Val int
-}
-func main() {
-	s1 := MyStruct{Val: 1}
-	p1 := &s1
-	pp1 := &p1
-	p4 := &s1
-	_ = p4
-	_ = pp1
-}`,
-			expected: map[string]AnalysisExpectation{
-				"s1":  {NeedsVarRef: true, NeedsVarRefAccess: true},  // varrefed because &s1 is taken
-				"p1":  {NeedsVarRef: true, NeedsVarRefAccess: true},  // varrefed because &p1 is taken
-				"pp1": {NeedsVarRef: false, NeedsVarRefAccess: true}, // NOT varrefed, points to varrefed value
-				"p4":  {NeedsVarRef: false, NeedsVarRefAccess: true}, // NOT varrefed, points to varrefed value
-			},
-		},
-		{
-			name: "varref_deref_struct",
-			code: `package main
-type MyStruct struct {
-	MyInt int
-}
-func main() {
-	myStruct := &MyStruct{}
-	(*myStruct).MyInt = 5
-	println((*myStruct).MyInt)
-}`,
-			expected: map[string]AnalysisExpectation{
-				"myStruct": {NeedsVarRef: false, NeedsVarRefAccess: false}, // NOT varrefed, direct pointer to struct
-			},
-		},
-		{
-			name: "pointer_composite_literal_untyped",
-			code: `package main
-func main() {
-	var ptr *struct{ x int }
-	ptr = &struct{ x int }{42}
-	println("Pointer value x:", ptr.x)
-
-	data := []*struct{ x int }{{42}, {43}}
-	println("First element x:", data[0].x)
-	println("Second element x:", data[1].x)
-}`,
-			expected: map[string]AnalysisExpectation{
-				"ptr":  {NeedsVarRef: false, NeedsVarRefAccess: false}, // Should NOT be varrefed
-				"data": {NeedsVarRef: false, NeedsVarRefAccess: false}, // Should NOT be varrefed
-			},
-		},
-		{
-			name: "var_declaration_composite_literal",
-			code: `package main
-type MockInode struct {
-	Value int
-}
-func main() {
-	var childInode *MockInode = &MockInode{Value: 42}
-	println("childInode.Value:", childInode.Value)
-}`,
-			expected: map[string]AnalysisExpectation{
-				"childInode": {NeedsVarRef: false, NeedsVarRefAccess: false}, // Should NOT be varrefed
-			},
-		},
-	}
-
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			analysis, objects := parseAndAnalyze(t, tt.code)
-
-			for varName, expected := range tt.expected {
-				obj, exists := objects[varName]
-				if !exists {
-					t.Errorf("Variable %q not found in parsed objects", varName)
-					continue
-				}
-
-				actualNeedsVarRef := analysis.NeedsVarRef(obj)
-				actualNeedsVarRefAccess := analysis.NeedsVarRefAccess(obj)
-
-				if actualNeedsVarRef != expected.NeedsVarRef {
-					t.Errorf("Variable %q: NeedsVarRef = %v, want %v",
-						varName, actualNeedsVarRef, expected.NeedsVarRef)
-				}
-
-				if actualNeedsVarRefAccess != expected.NeedsVarRefAccess {
-					t.Errorf("Variable %q: NeedsVarRefAccess = %v, want %v",
-						varName, actualNeedsVarRefAccess, expected.NeedsVarRefAccess)
-				}
-
-				// Print debug info
-				t.Logf("Variable %q: NeedsVarRef=%v, NeedsVarRefAccess=%v",
-					varName, actualNeedsVarRef, actualNeedsVarRefAccess)
-			}
-		})
-	}
-}
-
-// AnalysisExpectation defines what we expect from the analysis for a variable
-type AnalysisExpectation struct {
-	NeedsVarRef       bool
-	NeedsVarRefAccess bool
-}
-
-// parseAndAnalyze parses Go code and runs the analysis, returning the analysis and variable objects
-func parseAndAnalyze(t *testing.T, code string) (*Analysis, map[string]types.Object) {
-	// Parse the code
-	fset := token.NewFileSet()
-	file, err := parser.ParseFile(fset, "test.go", code, parser.ParseComments)
-	if err != nil {
-		t.Fatalf("Failed to parse code: %v", err)
-	}
-
-	// Create a minimal package for type checking
-	pkg := &packages.Package{
-		Syntax: []*ast.File{file},
-		TypesInfo: &types.Info{
-			Types: make(map[ast.Expr]types.TypeAndValue),
-			Defs:  make(map[*ast.Ident]types.Object),
-			Uses:  make(map[*ast.Ident]types.Object),
-		},
-		Fset: fset,
-	}
-
-	// Type check the package
-	typeConfig := &types.Config{}
-	typePkg, err := typeConfig.Check("main", fset, pkg.Syntax, pkg.TypesInfo)
-	if err != nil {
-		t.Fatalf("Failed to type check: %v", err)
-	}
-	pkg.Types = typePkg
-
-	// Run package-level analysis
-	analysis := AnalyzePackageFiles(pkg, nil)
-
-	// Collect variable objects
-	objects := make(map[string]types.Object)
-	for ident, obj := range pkg.TypesInfo.Defs {
-		if obj != nil && ident.Name != "_" {
-			if _, isVar := obj.(*types.Var); isVar {
-				objects[ident.Name] = obj
-			}
-		}
-	}
-
-	return analysis, objects
-}
-
-// TestAnalysisDebugInfo prints debug information about variable usage for manual inspection
-func TestAnalysisDebugInfo(t *testing.T) {
-	code := `package main
-func main() {
-	arr := [3]int{1, 2, 3}
-	arrPtr := &arr
-	for i, v := range arrPtr {
-		println("index:", i, "value:", v)
-	}
-}`
-
-	analysis, objects := parseAndAnalyze(t, code)
-
-	t.Log("=== Analysis Debug Information ===")
-	for varName, obj := range objects {
-		needsVarRef := analysis.NeedsVarRef(obj)
-		needsVarRefAccess := analysis.NeedsVarRefAccess(obj)
-
-		t.Logf("Variable: %s", varName)
-		t.Logf("  Type: %s", obj.Type())
-		t.Logf("  NeedsVarRef: %v", needsVarRef)
-		t.Logf("  NeedsVarRefAccess: %v", needsVarRefAccess)
-
-		// Print usage info if available
-		if usage, exists := analysis.VariableUsage[obj]; exists {
-			t.Logf("  Sources: %d", len(usage.Sources))
-			for i, src := range usage.Sources {
-				srcName := "nil"
-				if src.Object != nil {
-					srcName = src.Object.Name()
-				}
-				t.Logf("    [%d] %s (type: %v)", i, srcName, src.Type)
-			}
-			t.Logf("  Destinations: %d", len(usage.Destinations))
-			for i, dst := range usage.Destinations {
-				dstName := "nil"
-				if dst.Object != nil {
-					dstName = dst.Object.Name()
-				}
-				t.Logf("    [%d] %s (type: %v)", i, dstName, dst.Type)
-			}
-		}
-		t.Log("")
-	}
-}
-
-// TestWrapperTypeDetection verifies that the analysis correctly identifies wrapper types
-func TestWrapperTypeDetection(t *testing.T) {
-	// Use the actual tests test case
-	testPath := "../tests/tests/wrapper_type_args"
-
-	// Load the package using the packages config like the main compiler does
-	cfg := &packages.Config{
-		Mode: packages.NeedName |
-			packages.NeedFiles |
-			packages.NeedCompiledGoFiles |
-			packages.NeedImports |
-			packages.NeedDeps |
-			packages.NeedExportFile |
-			packages.NeedTypes |
-			packages.NeedSyntax |
-			packages.NeedTypesInfo |
-			packages.NeedTypesSizes,
-		Tests: false,
-		Env:   append(os.Environ(), "GOOS=js", "GOARCH=wasm"),
-	}
-
-	pkgs, err := packages.Load(cfg, testPath)
-	if err != nil {
-		t.Fatalf("Failed to load test package: %v", err)
-	}
-
-	if len(pkgs) != 1 {
-		t.Fatalf("Expected 1 package, got %d", len(pkgs))
-	}
-
-	pkg := pkgs[0]
-	if len(pkg.Errors) > 0 {
-		t.Fatalf("Package has errors: %v", pkg.Errors[0])
-	}
-
-	// Run package-level analysis
-	if len(pkg.Syntax) == 0 {
-		t.Fatal("No syntax files found")
-	}
-
-	analysis := AnalyzePackageFiles(pkg, nil)
-
-	// Verify the NamedBasicTypes map was initialized
-	if analysis.NamedBasicTypes == nil {
-		t.Error("NamedBasicTypes map was not initialized")
-	}
-
-	// Test some type lookups to verify wrapper type detection works
-	// We'll check if MyMode (which has methods) is detected as a wrapper type
-	scope := pkg.Types.Scope()
-
-	// Check if MyMode is detected as a wrapper type
-	if obj := scope.Lookup("MyMode"); obj != nil {
-		if typeName, ok := obj.(*types.TypeName); ok {
-			isWrapper := analysis.IsNamedBasicType(typeName.Type())
-			if !isWrapper {
-				t.Errorf("MyMode should be detected as wrapper type, got %v", isWrapper)
-			}
-			t.Logf("MyMode wrapper detection: %v (correct)", isWrapper)
-		}
-	}
-
-	// Test that regular struct types are not detected as wrapper types
-	if obj := scope.Lookup("MyDir"); obj != nil {
-		if typeName, ok := obj.(*types.TypeName); ok {
-			isWrapper := analysis.IsNamedBasicType(typeName.Type())
-			if isWrapper {
-				t.Errorf("MyDir should not be detected as wrapper type, got %v", isWrapper)
-			}
-			t.Logf("MyDir wrapper detection: %v (correct)", isWrapper)
-		}
-	}
-
-	t.Logf("Analysis completed successfully with %d named basic types tracked", len(analysis.NamedBasicTypes))
-}
-
-// TestDiscoverGsPackages verifies that the discoverEmbeddedGsPackages function
-// can find packages in the embedded gs/ directory
-func TestDiscoverGsPackages(t *testing.T) {
-	analysis := NewAnalysis(nil)
-
-	// Test package discovery using the embedded filesystem
-	packages := analysis.discoverEmbeddedGsPackages()
-	t.Logf("Discovered %d packages:", len(packages))
-	for _, pkg := range packages {
-		t.Logf("  - %s", pkg)
-	}
-
-	// We should find at least some packages
-	if len(packages) == 0 {
-		t.Errorf("Expected to find at least one package in gs/ directory")
-	}
-
-	// Check for some known packages that should exist
-	expectedPackages := []string{"sync", "bytes", "strings"}
-	for _, expected := range expectedPackages {
-		found := slices.Contains(packages, expected)
-		if !found {
-			t.Logf("Expected package '%s' not found in discovered packages: %v", expected, packages)
-		}
-	}
-}