Added another FileParser test

parse/argument.go

@@ -6,7 +6,7 @@ import (
 )
 
 // TODO: this should be a tagged union.
-// The supported caller arguments are registers, globals, immediates (and labales?)
+// The supported caller arguments are registers, globals, immediates (and labels?)
 
 type ArgumentNode struct {
 	source.UnmanagedSourceView

parse/file_test.go

@@ -9,6 +9,7 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+// The purpose of the test is to verify the structure of "simple" source file.
 func TestSingleFunction(t *testing.T) {
 	src := `def $i32 @add $i32 %x $i32 %y {
 	%res = add %x %y
@@ -72,3 +73,29 @@ func TestSingleFunction(t *testing.T) {
 	assert.Nil(t, perr)
 	assert.Equal(t, expected, file)
 }
+
+func TestFileParserTwoFunctionsNoExtraSeparator(t *testing.T) {
+	src := `def @first {
+}
+def @second {
+}`
+
+	expected := `def @first {
+}
+
+def @second {
+}
+`
+
+	v := source.NewSourceView(src)
+	_, ctx := v.Detach()
+	tkns, err := lex.NewTokenizer().Tokenize(v)
+	assert.NoError(t, err)
+
+	tknsView := parse.NewTokenView(tkns)
+	file, perr := parse.FileParser{}.Parse(&tknsView)
+	assert.Nil(t, perr)
+
+	got := file.String(ctx)
+	assert.Equal(t, expected, got)
+}