parser: enable deferred stms for void and optional functions

compiler/tests/defer_test.v

@@ -1,16 +1,51 @@
 fn foo() string {
-	println('foo()') 
-	return 'foo' 
-} 
+	println('foo()')
+	return 'foo'
+}
 
-fn foo2() string { 
-	println('start') 
-	defer { println('defer') } 
-	defer { println('defer2') } 
-	println('end') 
-	return foo() 
-} 
+fn foo2() string {
+	println('start')
+	defer { println('defer') }
+	defer { println('defer2') }
+	println('end')
+	return foo()
+}
 
-fn test_defer() { 
-	assert foo2() == 'foo' 
-} 
+fn test_defer() {
+	assert foo2() == 'foo'
+}
+
+fn set_num(i int, n mut Num) {
+	defer { n.val+=1 }
+	println("Hi")
+	if i < 5 {
+		return
+	} else {
+		n.val+=1
+	}
+}
+
+fn set_num_opt(n mut Num) ?int {
+	defer { n.val = 1 }
+	return 99
+}
+
+struct Num {
+mut:
+	val int
+}
+
+fn test_defer_early_exit() {
+	mut sum := Num{0}
+	for i in 0..10 {
+		set_num(i, mut sum)
+	}
+	println("sum: $sum.val")
+	assert sum.val == 15
+}
+
+fn test_defer_option() {
+	mut ok := Num{0}
+	set_num_opt(mut ok)
+	assert ok.val == 1
+}