dstepanov-yadro/neoneo-go
1
0
Fork 0
forked from TrueCloudLab/neoneo-go
Code Pull requests Activity

compiler: allow to use return with no arguments

Browse source
This commit is contained in:
Evgenii Stratonikov 2020-05-06 15:24:12 +03:00
parent 9afcd71be3
commit 770cff8b91
2 changed files with 14 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pkg/compiler/func_scope.go
View file

@ -65,10 +65,12 @@ func (c *funcScope) analyzeVoidCalls(node ast.Node) bool {
} }
} }
case *ast.ReturnStmt: case *ast.ReturnStmt:
if len(n.Results) > 0 {
switch n.Results[0].(type) { switch n.Results[0].(type) {
case *ast.CallExpr: case *ast.CallExpr:
return false return false
} }
}
case *ast.BinaryExpr: case *ast.BinaryExpr:
return false return false
case *ast.CallExpr: case *ast.CallExpr:

9
pkg/compiler/function_call_test.go
View file

@ -121,7 +121,14 @@ func TestFunctionWithVoidReturn(t *testing.T) {
return x + y return x + y
} }
func getSomeInteger() { } func getSomeInteger() { %s }
` `
t.Run("EmptyBody", func(t *testing.T) {
src := fmt.Sprintf(src, "")
eval(t, src, big.NewInt(6)) eval(t, src, big.NewInt(6))
})
t.Run("SingleReturn", func(t *testing.T) {
src := fmt.Sprintf(src, "return")
eval(t, src, big.NewInt(6))
})
} }