refactor: plugin functions can be defined in any format

plugin/examples/debugtalk.go

@@ -0,0 +1,65 @@
+package main
+
+import (
+	"fmt"
+
+	plugin "github.com/httprunner/hrp/plugin"
+)
+
+func SumTwoInt(a, b int) int {
+	return a + b
+}
+
+func SumInts(args ...int) int {
+	var sum int
+	for _, arg := range args {
+		sum += arg
+	}
+	return sum
+}
+
+func Sum(args ...interface{}) (interface{}, error) {
+	var sum float64
+	for _, arg := range args {
+		switch v := arg.(type) {
+		case int:
+			sum += float64(v)
+		case float64:
+			sum += v
+		default:
+			return nil, fmt.Errorf("unexpected type: %T", arg)
+		}
+	}
+	return sum, nil
+}
+
+func SumTwoString(a, b string) string {
+	return a + b
+}
+
+func SumStrings(s ...string) string {
+	var sum string
+	for _, arg := range s {
+		sum += arg
+	}
+	return sum
+}
+
+func Concatenate(args ...interface{}) (interface{}, error) {
+	var result string
+	for _, arg := range args {
+		result += fmt.Sprintf("%v", arg)
+	}
+	return result, nil
+}
+
+// register functions and build to plugin binary
+func main() {
+	plugin.Register("sum_ints", SumInts)
+	plugin.Register("sum_two_int", SumTwoInt)
+	plugin.Register("sum", Sum)
+	plugin.Register("sum_two_string", SumTwoString)
+	plugin.Register("sum_strings", SumStrings)
+	plugin.Register("concatenate", Concatenate)
+	plugin.Serve()
+}

plugin/examples/plugin_test.go

@@ -0,0 +1,87 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"github.com/httprunner/hrp/plugin/host"
+)
+
+func TestMain(m *testing.M) {
+	fmt.Println("[TestMain] build go plugin")
+	cmd := exec.Command("go", "build", "-o=debugtalk.bin", "./debugtalk.go")
+	if err := cmd.Run(); err != nil {
+		panic(err)
+	}
+	os.Exit(m.Run())
+}
+
+func TestInitHashicorpPlugin(t *testing.T) {
+	f, err := host.Init("./debugtalk.bin")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer host.Quit()
+
+	v1, err := f.GetNames()
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !assert.Contains(t, v1, "sum_ints") {
+		t.Fatal(err)
+	}
+	if !assert.Contains(t, v1, "concatenate") {
+		t.Fatal(err)
+	}
+
+	var v2 interface{}
+	v2, err = f.Call("sum_ints", 1, 2, 3, 4)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !assert.Equal(t, 10, v2) {
+		t.Fail()
+	}
+	v2, err = f.Call("sum_two_int", 1, 2)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !assert.Equal(t, 3, v2) {
+		t.Fail()
+	}
+	v2, err = f.Call("sum", 1, 2, 3.4, 5)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !assert.Equal(t, 11.4, v2) {
+		t.Fail()
+	}
+
+	var v3 interface{}
+	v3, err = f.Call("sum_two_string", "a", "b")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !assert.Equal(t, "ab", v3) {
+		t.Fail()
+	}
+	v3, err = f.Call("sum_strings", "a", "b", "c")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !assert.Equal(t, "abc", v3) {
+		t.Fail()
+	}
+
+	v3, err = f.Call("concatenate", "a", 2, "c", 3.4)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !assert.Equal(t, "a2c3.4", v3) {
+		t.Fail()
+	}
+}