61890b15cb
* wip * move print and JSON functions to modules * builtin functions are not replacable now * builtin functions are added for default nil symbol table * importables: builtin modules and source modules * refactoring runtime tests * fix tests * update documentation * cleanup * clean up cli * fix REPL prints
49 lines
1.4 KiB
Go
49 lines
1.4 KiB
Go
package script_test
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/d5/tengo/assert"
|
|
"github.com/d5/tengo/objects"
|
|
"github.com/d5/tengo/script"
|
|
)
|
|
|
|
func TestScriptSourceModule(t *testing.T) {
|
|
// script1 imports "mod1"
|
|
scr := script.New([]byte(`out := import("mod")`))
|
|
scr.SetImports(map[string]objects.Importable{
|
|
"mod": &objects.SourceModule{Src: []byte(`export 5`)},
|
|
})
|
|
c, err := scr.Run()
|
|
assert.Equal(t, int64(5), c.Get("out").Value())
|
|
|
|
// executing module function
|
|
scr = script.New([]byte(`fn := import("mod"); out := fn()`))
|
|
scr.SetImports(map[string]objects.Importable{
|
|
"mod": &objects.SourceModule{Src: []byte(`a := 3; export func() { return a + 5 }`)},
|
|
})
|
|
c, err = scr.Run()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, int64(8), c.Get("out").Value())
|
|
|
|
scr = script.New([]byte(`out := import("mod")`))
|
|
scr.SetImports(map[string]objects.Importable{
|
|
"text": &objects.BuiltinModule{
|
|
Attrs: map[string]objects.Object{
|
|
"title": &objects.UserFunction{Name: "title", Value: func(args ...objects.Object) (ret objects.Object, err error) {
|
|
s, _ := objects.ToString(args[0])
|
|
return &objects.String{Value: strings.Title(s)}, nil
|
|
}},
|
|
},
|
|
},
|
|
"mod": &objects.SourceModule{Src: []byte(`text := import("text"); export text.title("foo")`)},
|
|
})
|
|
c, err = scr.Run()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "Foo", c.Get("out").Value())
|
|
scr.SetImports(nil)
|
|
_, err = scr.Run()
|
|
assert.Error(t, err)
|
|
|
|
}
|