43 lines
888 B
Go
43 lines
888 B
Go
// Copyright 2016-2017 Liam Stanley <me@liamstanley.io>. All rights reserved.
|
|
// Use of this source code is governed by the MIT license that can be
|
|
// found in the LICENSE file.
|
|
|
|
package girc
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestEventLimiter(t *testing.T) {
|
|
events := []*Event{}
|
|
sendFunc := func(event *Event) error {
|
|
events = append(events, event)
|
|
|
|
return nil
|
|
}
|
|
|
|
limiter := NewEventLimiter(1, 150*time.Millisecond, sendFunc)
|
|
|
|
var e1, e2 *Event
|
|
go func() {
|
|
if err := limiter.SendAll(e1, e2); err != nil {
|
|
t.Fatalf("SendAll gave: %v", err)
|
|
}
|
|
}()
|
|
|
|
// Checking it immediately should yield 1 time.
|
|
if len(events) > 1 {
|
|
t.Fatalf("limiter has %v events, wanted 0 or 1", len(events))
|
|
}
|
|
|
|
time.Sleep(500 * time.Millisecond)
|
|
|
|
// It should now show a length of two.
|
|
if len(events) != 2 {
|
|
t.Fatalf("limiter has %v events, wanted 2", len(events))
|
|
}
|
|
|
|
limiter.Stop()
|
|
}
|