107 lines
2.0 KiB
Go
107 lines
2.0 KiB
Go
package shutils_test
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
|
|
"go.arsenm.dev/lure/distro"
|
|
"go.arsenm.dev/lure/internal/shutils"
|
|
"go.arsenm.dev/lure/internal/shutils/decoder"
|
|
"mvdan.cc/sh/v3/interp"
|
|
"mvdan.cc/sh/v3/syntax"
|
|
)
|
|
|
|
const testScript = `
|
|
name='test'
|
|
version='0.0.1'
|
|
release=1
|
|
epoch=2
|
|
desc="Test package"
|
|
homepage='https://lure.arsenm.dev'
|
|
maintainer='Arsen Musayelyan <arsen@arsenm.dev>'
|
|
architectures=('arm64' 'amd64')
|
|
license=('GPL-3.0-or-later')
|
|
provides=('test')
|
|
conflicts=('test')
|
|
replaces=('test-old')
|
|
replaces_test_os=('test-legacy')
|
|
|
|
deps=('sudo')
|
|
|
|
build_deps=('golang')
|
|
build_deps_arch=('go')
|
|
|
|
test() {
|
|
test-cmd "Hello, World"
|
|
test-fb
|
|
}
|
|
|
|
package() {
|
|
install-binary test
|
|
}
|
|
`
|
|
|
|
var osRelease = &distro.OSRelease{
|
|
ID: "test_os",
|
|
Like: []string{"arch"},
|
|
}
|
|
|
|
func TestExecFuncs(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
fl, err := syntax.NewParser().Parse(strings.NewReader(testScript), "lure.sh")
|
|
if err != nil {
|
|
t.Fatalf("Expected no error, got %s", err)
|
|
}
|
|
|
|
runner, err := interp.New()
|
|
if err != nil {
|
|
t.Fatalf("Expected no error, got %s", err)
|
|
}
|
|
|
|
err = runner.Run(ctx, fl)
|
|
if err != nil {
|
|
t.Fatalf("Expected no error, got %s", err)
|
|
}
|
|
|
|
dec := decoder.New(osRelease, runner)
|
|
fn, ok := dec.GetFunc("test")
|
|
if !ok {
|
|
t.Fatalf("Expected test() function to exist")
|
|
}
|
|
|
|
eh := shutils.ExecFuncs{
|
|
"test-cmd": func(hc interp.HandlerContext, name string, args []string) error {
|
|
if name != "test-cmd" {
|
|
t.Errorf("Expected name to be 'test-cmd', got '%s'", name)
|
|
}
|
|
|
|
if len(args) < 1 {
|
|
t.Fatalf("Expected at least one argument, got %d", len(args))
|
|
}
|
|
|
|
if args[0] != "Hello, World" {
|
|
t.Errorf("Expected first argument to be 'Hello, World', got '%s'", args[0])
|
|
}
|
|
|
|
return nil
|
|
},
|
|
}
|
|
|
|
fbInvoked := false
|
|
fbHandler := func(context.Context, []string) error {
|
|
fbInvoked = true
|
|
return nil
|
|
}
|
|
|
|
err = fn(ctx, interp.ExecHandler(eh.ExecHandler(fbHandler)))
|
|
if err != nil {
|
|
t.Errorf("Expected no error, got %s", err)
|
|
}
|
|
|
|
if !fbInvoked {
|
|
t.Errorf("Expected fallback handler to be invoked")
|
|
}
|
|
}
|