Function TestLoadPluginSkipError

pkg/plugin/plugin_test.go:178–252

(t *testing.T)

}

func TestLoadPluginSkipError(t *testing.T) {
	ctx := context.Background()

	pluginName := "tplugin"
	pluginVersion := uint16(1)
	pluginSign := pluginName + "-" + strconv.Itoa(int(pluginVersion))

	cfg := Config{
		Plugins:      []string{pluginSign, pluginSign, "notExists-2"},
		PluginDir:    "",
		EnvVersion:   map[string]uint16{"go": 1112},
		SkipWhenFail: true,
	}

	// setup load test hook.
	SetTestHook(func(plugin *Plugin, dir string, pluginID ID) (manifest func() *Manifest, err error) {
		return func() *Manifest {
			m := &AuditManifest{
				Manifest: Manifest{
					Kind:    Audit,
					Name:    pluginName,
					Version: pluginVersion,
					OnInit: func(ctx context.Context, manifest *Manifest) error {
						return io.EOF
					},
					OnShutdown: func(ctx context.Context, manifest *Manifest) error {
						return io.EOF
					},
					Validate: func(ctx context.Context, manifest *Manifest) error {
						return io.EOF
					},
				},
				OnGeneralEvent: func(ctx context.Context, sctx *variable.SessionVars, event GeneralEvent, cmd string) {
				},
			}
			return ExportManifest(m)
		}, nil
	})
	defer func() {
		testHook = nil
	}()

	// trigger load.
	err := Load(ctx, cfg)
	require.NoError(t, err)

	err = Init(ctx, cfg)
	require.NoError(t, err)
	require.False(t, IsEnable(Audit))

	// load all.
	ps := GetAll()
	require.Len(t, ps, 1)

	// find plugin by type and name
	p := Get(Audit, "tplugin")
	require.NotNil(t, p)
	p = Get(Audit, "tplugin2")