refactor: plugin functions can be defined in any format

This commit is contained in:
debugtalk
2022-01-14 16:37:36 +08:00
parent 5a943ff721
commit b8d6192a34
12 changed files with 350 additions and 259 deletions

View File

@@ -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()
}
}