func testLoadBalancerOKSend(t *testing.T, events []modetest.EventInfo) {
var collected [][]outputs.Data
mode, _ := NewSync(
modetest.SyncClients(1, &modetest.MockClient{
Connected: false,
CBPublish: modetest.PublishCollect(&collected),
}),
2,
1*time.Millisecond,
1*time.Millisecond,
10*time.Millisecond,
)
modetest.TestMode(t, mode, testNoOpts, events, modetest.Signals(true), &collected)
}
func testLoadBalancerFlakyConnectionOkSend(t *testing.T, events []modetest.EventInfo) {
var collected [][]outputs.Data
mode, _ := NewSync(
modetest.SyncClients(2, &modetest.MockClient{
Connected: true,
CBPublish: modetest.PublishCollectAfterFailStart(1, &collected),
}),
3,
1*time.Millisecond,
1*time.Millisecond,
10*time.Millisecond,
)
modetest.TestMode(t, mode, testNoOpts, events, modetest.Signals(true), &collected)
}
func TestLoadBalancerStartStop(t *testing.T) {
enableLogging([]string{"*"})
mode, _ := NewSync(
modetest.SyncClients(1, &modetest.MockClient{
Connected: false,
}),
1,
1*time.Millisecond,
1*time.Millisecond,
10*time.Millisecond,
)
modetest.TestMode(t, mode, testNoOpts, nil, nil, nil)
}
func testLoadBalancerFlakyGuaranteed(t *testing.T, events []modetest.EventInfo) {
var collected [][]common.MapStr
mode, _ := NewSync(
modetest.SyncClients(2, &modetest.MockClient{
Connected: true,
CBPublish: modetest.PublishCollectAfterFailStart(25, &collected),
}),
3,
1*time.Millisecond,
1*time.Millisecond,
10*time.Millisecond,
)
modetest.TestMode(t, mode, testGuaranteed, events, modetest.Signals(true), &collected)
}
func testLoadBalancerTempFlakyGuaranteed(t *testing.T, events []modetest.EventInfo) {
var collected [][]outputs.Data
err := mode.ErrTempBulkFailure
mode, _ := NewSync(
modetest.SyncClients(2, &modetest.MockClient{
Connected: true,
CBPublish: modetest.PublishCollectAfterFailStartWith(25, err, &collected),
}),
3,
1*time.Millisecond,
1*time.Millisecond,
10*time.Millisecond,
)
modetest.TestMode(t, mode, testGuaranteed, events, modetest.Signals(true), &collected)
}
func testLoadBalancerTempFlakyInfAttempts(t *testing.T, events []modetest.EventInfo) {
var collected [][]common.MapStr
err := mode.ErrTempBulkFailure
mode, _ := NewSync(
modetest.SyncClients(2, &modetest.MockClient{
Connected: true,
CBPublish: modetest.PublishCollectAfterFailStartWith(25, err, &collected),
}),
0,
1*time.Millisecond,
1*time.Millisecond,
10*time.Millisecond,
)
modetest.TestMode(t, mode, testNoOpts, events, modetest.Signals(true), &collected)
}
func testLoadBalancerFailSendWithoutActiveConnections(
t *testing.T,
events []modetest.EventInfo,
) {
errFail := errors.New("fail connect")
mode, _ := NewSync(
modetest.SyncClients(2, &modetest.MockClient{
Connected: false,
CBConnect: modetest.ConnectFail(errFail),
}),
2,
1*time.Millisecond,
1*time.Millisecond,
10*time.Millisecond,
)
modetest.TestMode(t, mode, testNoOpts, events, modetest.Signals(false), nil)
}