load_test.go

package plugins_test

import (
	"os"
	"os/exec"
	"path/filepath"
	"plugin"
	"runtime"
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/alpacahq/marketstore/v4/plugins"
)

func setup(t *testing.T) (tearDown func(), testPluginLib, oldGoPath, absTestPluginLib string) {
	t.Helper()

	dirName := t.TempDir()

	if osType := runtime.GOOS; osType != "linux" {
		t.Skip("Only linux runs plugins")
	}

	binDirName := filepath.Join(dirName, "bin")
	assert.Nil(t, os.MkdirAll(binDirName, 0o777))
	testFileName := "plugin.go"
	testFilePath := filepath.Join(dirName, testFileName)
	testPluginLib = "plugin.so"
	soFilePath := filepath.Join(binDirName, testPluginLib)
	file, err := os.Create(testFilePath)
	if err != nil {
		t.Fatal("Could not create test plugin source file")
	}
	code := `package main
func main() {}
`
	_, err = file.WriteString(code)
	assert.Nil(t, err)
	assert.Nil(t, file.Close())
	cmd := exec.Command("go",
		"build",
		"-buildmode=plugin",
		"-o",
		soFilePath,
		testFilePath)

	if err := cmd.Run(); err != nil {
		t.Log(err)
		t.Skip("Unable to build test plugin ** is go version > 1.9 in your path?")
	}

	goPath := os.Getenv("GOPATH")
	newGoPath := dirName + ":" + goPath
	oldGoPath = goPath
	absTestPluginLib = soFilePath
	os.Setenv("GOPATH", newGoPath)

	return func() {
		if oldGoPath != "" {
			os.Setenv("GOPATH", oldGoPath)
		}
	}, testPluginLib, oldGoPath, absTestPluginLib
}

func TestLoadFromGOPATH(t *testing.T) {
	tearDown, testPluginLib, _, absTestPluginLib := setup(t)
	defer tearDown()

	var pi *plugin.Plugin
	var err error
	pi, err = plugins.Load(testPluginLib)
	assert.NotNil(t, pi)
	assert.Nil(t, err)

	pi, err = plugins.Load("nonexistent")
	assert.Nil(t, pi)
	assert.NotNil(t, err)

	// abs path case
	pi, err = plugins.Load(absTestPluginLib)
	assert.NotNil(t, pi)
	assert.Nil(t, err)
}