|
@@ -6,17 +6,10 @@ import (
|
|
|
"io"
|
|
|
"net"
|
|
|
"strconv"
|
|
|
+ "testing"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
-
|
|
|
-type testState interface {
|
|
|
- Error(args ...interface{})
|
|
|
- Fatal(args ...interface{})
|
|
|
- Errorf(format string, args ...interface{})
|
|
|
- Fatalf(format string, args ...interface{})
|
|
|
-}
|
|
|
-
|
|
|
|
|
|
|
|
|
|
|
@@ -33,7 +26,7 @@ type mockBroker struct {
|
|
|
stopper chan bool
|
|
|
expectations chan encoder
|
|
|
listener net.Listener
|
|
|
- t testState
|
|
|
+ t *testing.T
|
|
|
latency time.Duration
|
|
|
}
|
|
|
|
|
@@ -130,16 +123,16 @@ func (b *mockBroker) serverError(err error, conn net.Conn) bool {
|
|
|
return false
|
|
|
}
|
|
|
|
|
|
-
|
|
|
+
|
|
|
|
|
|
-
|
|
|
-func newMockBroker(t testState, brokerID int32) *mockBroker {
|
|
|
+
|
|
|
+func newMockBroker(t *testing.T, brokerID int32) *mockBroker {
|
|
|
return newMockBrokerAddr(t, brokerID, "localhost:0")
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
-func newMockBrokerAddr(t testState, brokerID int32, addr string) *mockBroker {
|
|
|
+func newMockBrokerAddr(t *testing.T, brokerID int32, addr string) *mockBroker {
|
|
|
var err error
|
|
|
|
|
|
broker := &mockBroker{
|