compiler: add test for compile and save

pkg/compiler/compiler_test.go

@@ -7,21 +7,28 @@ import (
 	"testing"
 
 	"github.com/CityOfZion/neo-go/pkg/compiler"
+	"github.com/stretchr/testify/require"
 )
 
 const examplePath = "../../examples"
+const exampleCompilePath = "testdata/compile"
+const exampleSavePath = exampleCompilePath + "/save"
 
-func TestExamplesFolder(t *testing.T) {
-	infos, err := ioutil.ReadDir(examplePath)
-	if err != nil {
-		t.Fatal(err)
+type compilerTestCase struct {
+	name     string
+	function func()
 }
 
+func TestCompiler(t *testing.T) {
+	testCases := []compilerTestCase{
+		{
+			name: "TestCompile",
+			function: func() {
+				infos, err := ioutil.ReadDir(examplePath)
+				require.NoError(t, err)
 				for _, info := range infos {
 					infos, err := ioutil.ReadDir(path.Join(examplePath, info.Name()))
-		if err != nil {
-			t.Fatal(err)
-		}
+					require.NoError(t, err)
 					if len(infos) == 0 {
 						t.Fatal("detected smart contract folder with no contract in it")
 					}
@@ -32,6 +39,32 @@ func TestExamplesFolder(t *testing.T) {
 						t.Fatal(err)
 					}
 				}
+			},
+		},
+		{
+			name: "TestCompileAndSave",
+			function: func() {
+				infos, err := ioutil.ReadDir(exampleCompilePath)
+				require.NoError(t, err)
+				err = os.MkdirAll(exampleSavePath, os.ModePerm)
+				require.NoError(t, err)
+				outfile := exampleSavePath + "/test.avm"
+				if _, err := compiler.CompileAndSave(exampleCompilePath+"/"+infos[0].Name(), &compiler.Options{Outfile: outfile}); err != nil {
+					t.Fatal(err)
+				}
+				defer func() {
+					err := os.RemoveAll(exampleSavePath)
+					require.NoError(t, err)
+				}()
+			},
+		},
+	}
+
+	for _, tcase := range testCases {
+		t.Run(tcase.name, func(t *testing.T) {
+			tcase.function()
+		})
+	}
 }
 
 func filterFilename(infos []os.FileInfo) string {

pkg/compiler/testdata/compile/test.go

@@ -0,0 +1,7 @@
+package compile
+
+import "github.com/CityOfZion/neo-go/pkg/interop/runtime"
+
+func Main() {
+	runtime.Notify("Hello world!")
+}