Initial commit

internal/shutils/handlers/exec_test.go

@@ -0,0 +1,124 @@
+/*
+ * LURE - Linux User REpository
+ * Copyright (C) 2023 Elara Musayelyan
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package handlers_test
+
+import (
+	"context"
+	"strings"
+	"testing"
+
+	"lure.sh/lure/internal/shutils/handlers"
+	"lure.sh/lure/internal/shutils/decoder"
+	"lure.sh/lure/pkg/distro"
+	"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.sh'
+	maintainer='Elara Musayelyan <elara@elara.ws>'
+	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")
+	}
+}