summaryrefslogtreecommitdiff
path: root/plugin/special/test.go
diff options
context:
space:
mode:
Diffstat (limited to 'plugin/special/test.go')
-rw-r--r--plugin/special/test.go20
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 {