diff --git a/pkg/compiler/codegen.go b/pkg/compiler/codegen.go index 90f4d62c9..93ee4c711 100644 --- a/pkg/compiler/codegen.go +++ b/pkg/compiler/codegen.go @@ -565,6 +565,9 @@ func (c *codegen) Visit(node ast.Node) ast.Visitor { lElse := c.newLabel() lElseEnd := c.newLabel() + if n.Init != nil { + ast.Walk(c, n.Init) + } if n.Cond != nil { c.emitBoolExpr(n.Cond, true, false, lElse) } diff --git a/pkg/compiler/if_test.go b/pkg/compiler/if_test.go index 23077ecf9..038c8fe92 100644 --- a/pkg/compiler/if_test.go +++ b/pkg/compiler/if_test.go @@ -1,6 +1,7 @@ package compiler_test import ( + "fmt" "math/big" "testing" ) @@ -91,3 +92,32 @@ func TestNestedIF(t *testing.T) { ` eval(t, src, big.NewInt(0)) } + +func TestInitIF(t *testing.T) { + t.Run("Simple", func(t *testing.T) { + src := `package foo + func Main() int { + if a := 42; true { + return a + } + return 0 + }` + eval(t, src, big.NewInt(42)) + }) + t.Run("Shadow", func(t *testing.T) { + srcTmpl := `package foo + func Main() int { + a := 11 + if a := 42; %v { + return a + } + return a + }` + t.Run("True", func(t *testing.T) { + eval(t, fmt.Sprintf(srcTmpl, true), big.NewInt(42)) + }) + t.Run("False", func(t *testing.T) { + eval(t, fmt.Sprintf(srcTmpl, false), big.NewInt(11)) + }) + }) +}