package script_test

import (
	"errors"
	"testing"

	"github.com/d5/tengo/assert"
	"github.com/d5/tengo/script"
)

func TestScript_SetUserModuleLoader(t *testing.T) {
	// script1 imports "mod1"
	scr := script.New([]byte(`out := import("mod")`))
	scr.SetUserModuleLoader(func(name string) ([]byte, error) {
		return []byte(`export 5`), nil
	})
	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.SetUserModuleLoader(func(name string) ([]byte, error) {
		return []byte(`a := 3; export func() { return a + 5 }`), nil
	})
	c, err = scr.Run()
	assert.NoError(t, err)
	assert.Equal(t, int64(8), c.Get("out").Value())

	// disabled builtin function
	scr = script.New([]byte(`out := import("mod")`))
	scr.SetUserModuleLoader(func(name string) ([]byte, error) {
		return []byte(`export len([1, 2, 3])`), nil
	})
	c, err = scr.Run()
	assert.NoError(t, err)
	assert.Equal(t, int64(3), c.Get("out").Value())
	scr.DisableBuiltinFunction("len")
	_, err = scr.Run()
	assert.Error(t, err)

	// disabled stdlib
	scr = script.New([]byte(`out := import("mod")`))
	scr.SetUserModuleLoader(func(name string) ([]byte, error) {
		if name == "mod" {
			return []byte(`text := import("text"); export text.title("foo")`), nil
		}
		return nil, errors.New("module not found")
	})
	c, err = scr.Run()
	assert.NoError(t, err)
	assert.Equal(t, "Foo", c.Get("out").Value())
	scr.DisableStdModule("text")
	_, err = scr.Run()
	assert.Error(t, err)

}