diff options
Diffstat (limited to 'plugin/special/test.go')
| -rw-r--r-- | plugin/special/test.go | 20 |
1 files changed, 16 insertions, 4 deletions
diff --git a/plugin/special/test.go b/plugin/special/test.go index cf83ed83..6425ca93 100644 --- a/plugin/special/test.go +++ b/plugin/special/test.go @@ -18,19 +18,31 @@ var testCIDRs = []string{ func init() { lib.RegisterInputConfigCreator(typeTest, func(action lib.Action, data json.RawMessage) (lib.InputConverter, error) { - return newTest(action, data) + return NewTestFromBytes(action, data) }) lib.RegisterInputConverter(typeTest, &test{ Description: descTest, }) } -func newTest(action lib.Action, data json.RawMessage) (lib.InputConverter, error) { - return &test{ +func NewTest(action lib.Action, opts ...lib.InputOption) lib.InputConverter { + t := &test{ Type: typeTest, Action: action, Description: descTest, - }, nil + } + + for _, opt := range opts { + if opt != nil { + opt(t) + } + } + + return t +} + +func NewTestFromBytes(action lib.Action, data []byte) (lib.InputConverter, error) { + return NewTest(action), nil } type test struct { |
