Merge pull request #568 from nspcc-dev/feature/compiler_clean_and_tests

compiler: clean and tests
This commit is contained in:
Roman Khimov 2020-01-14 18:44:07 +03:00 committed by GitHub
commit 9eb880a095
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 58 additions and 34 deletions

View file

@ -3,10 +3,7 @@ package compiler
import (
"bytes"
"fmt"
"go/ast"
"go/build"
"go/parser"
"go/types"
"io"
"io/ioutil"
"os"
@ -61,11 +58,6 @@ func Compile(r io.Reader) ([]byte, error) {
return buf, nil
}
type archive struct {
f *ast.File
typeInfo *types.Info
}
// CompileAndSave will compile and save the file to disk.
func CompileAndSave(src string, o *Options) ([]byte, error) {
if !strings.HasSuffix(src, ".go") {
@ -90,11 +82,3 @@ func CompileAndSave(src string, o *Options) ([]byte, error) {
out := fmt.Sprintf("%s.%s", o.Outfile, o.Ext)
return b, ioutil.WriteFile(out, b, os.ModePerm)
}
func gopath() string {
gopath := os.Getenv("GOPATH")
if len(gopath) == 0 {
gopath = build.Default.GOPATH
}
return gopath
}

View file

@ -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) {
type compilerTestCase struct {
name string
function func()
}
func TestCompiler(t *testing.T) {
testCases := []compilerTestCase{
{
name: "TestCompile",
function: func() {
infos, err := ioutil.ReadDir(examplePath)
if err != nil {
t.Fatal(err)
}
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 {

7
pkg/compiler/testdata/compile/test.go vendored Normal file
View file

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