httprunner/plugin/examples/plugin_test.go

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