mirror of
https://github.com/buger/goreplay.git
synced 2024-04-21 12:32:02 +00:00
Urban Ishimwe
6d812ceb7f
// PluginReader is an interface for input plugins
type PluginReader interface {
PluginRead() (msg *Message, err error)
}
// PluginWriter is an interface for output plugins
type PluginWriter interface {
PluginWrite(msg *Message) (n int, err error)
}
44 lines
1012 B
Go
44 lines
1012 B
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestPluginsRegistration(t *testing.T) {
|
|
Settings.InputDummy = MultiOption{"[]"}
|
|
Settings.OutputDummy = MultiOption{"[]"}
|
|
Settings.OutputHTTP = MultiOption{"www.example.com|10"}
|
|
Settings.InputFile = MultiOption{"/dev/null"}
|
|
|
|
plugins := NewPlugins()
|
|
|
|
if len(plugins.Inputs) != 3 {
|
|
t.Errorf("Should be 3 inputs got %d", len(plugins.Inputs))
|
|
}
|
|
|
|
if _, ok := plugins.Inputs[0].(*DummyInput); !ok {
|
|
t.Errorf("First input should be DummyInput")
|
|
}
|
|
|
|
if _, ok := plugins.Inputs[1].(*FileInput); !ok {
|
|
t.Errorf("Second input should be FileInput")
|
|
}
|
|
|
|
if len(plugins.Outputs) != 2 {
|
|
t.Errorf("Should be 2 output %d", len(plugins.Outputs))
|
|
}
|
|
|
|
if _, ok := plugins.Outputs[0].(*DummyOutput); !ok {
|
|
t.Errorf("First output should be DummyOutput")
|
|
}
|
|
|
|
if l, ok := plugins.Outputs[1].(*Limiter); ok {
|
|
if _, ok := l.plugin.(*HTTPOutput); !ok {
|
|
t.Errorf("HTTPOutput should be wrapped in limiter")
|
|
}
|
|
} else {
|
|
t.Errorf("Second output should be Limiter")
|
|
}
|
|
|
|
}
|