|
@@ -28,6 +28,16 @@ type mockedConnectionInfo struct {
|
|
|
establishedAtReturnsOnCall map[int]struct {
|
|
|
result1 time.Time
|
|
|
}
|
|
|
+ IsLocalStub func() bool
|
|
|
+ isLocalMutex sync.RWMutex
|
|
|
+ isLocalArgsForCall []struct {
|
|
|
+ }
|
|
|
+ isLocalReturns struct {
|
|
|
+ result1 bool
|
|
|
+ }
|
|
|
+ isLocalReturnsOnCall map[int]struct {
|
|
|
+ result1 bool
|
|
|
+ }
|
|
|
PriorityStub func() int
|
|
|
priorityMutex sync.RWMutex
|
|
|
priorityArgsForCall []struct {
|
|
@@ -188,6 +198,59 @@ func (fake *mockedConnectionInfo) EstablishedAtReturnsOnCall(i int, result1 time
|
|
|
}{result1}
|
|
|
}
|
|
|
|
|
|
+func (fake *mockedConnectionInfo) IsLocal() bool {
|
|
|
+ fake.isLocalMutex.Lock()
|
|
|
+ ret, specificReturn := fake.isLocalReturnsOnCall[len(fake.isLocalArgsForCall)]
|
|
|
+ fake.isLocalArgsForCall = append(fake.isLocalArgsForCall, struct {
|
|
|
+ }{})
|
|
|
+ stub := fake.IsLocalStub
|
|
|
+ fakeReturns := fake.isLocalReturns
|
|
|
+ fake.recordInvocation("IsLocal", []interface{}{})
|
|
|
+ fake.isLocalMutex.Unlock()
|
|
|
+ if stub != nil {
|
|
|
+ return stub()
|
|
|
+ }
|
|
|
+ if specificReturn {
|
|
|
+ return ret.result1
|
|
|
+ }
|
|
|
+ return fakeReturns.result1
|
|
|
+}
|
|
|
+
|
|
|
+func (fake *mockedConnectionInfo) IsLocalCallCount() int {
|
|
|
+ fake.isLocalMutex.RLock()
|
|
|
+ defer fake.isLocalMutex.RUnlock()
|
|
|
+ return len(fake.isLocalArgsForCall)
|
|
|
+}
|
|
|
+
|
|
|
+func (fake *mockedConnectionInfo) IsLocalCalls(stub func() bool) {
|
|
|
+ fake.isLocalMutex.Lock()
|
|
|
+ defer fake.isLocalMutex.Unlock()
|
|
|
+ fake.IsLocalStub = stub
|
|
|
+}
|
|
|
+
|
|
|
+func (fake *mockedConnectionInfo) IsLocalReturns(result1 bool) {
|
|
|
+ fake.isLocalMutex.Lock()
|
|
|
+ defer fake.isLocalMutex.Unlock()
|
|
|
+ fake.IsLocalStub = nil
|
|
|
+ fake.isLocalReturns = struct {
|
|
|
+ result1 bool
|
|
|
+ }{result1}
|
|
|
+}
|
|
|
+
|
|
|
+func (fake *mockedConnectionInfo) IsLocalReturnsOnCall(i int, result1 bool) {
|
|
|
+ fake.isLocalMutex.Lock()
|
|
|
+ defer fake.isLocalMutex.Unlock()
|
|
|
+ fake.IsLocalStub = nil
|
|
|
+ if fake.isLocalReturnsOnCall == nil {
|
|
|
+ fake.isLocalReturnsOnCall = make(map[int]struct {
|
|
|
+ result1 bool
|
|
|
+ })
|
|
|
+ }
|
|
|
+ fake.isLocalReturnsOnCall[i] = struct {
|
|
|
+ result1 bool
|
|
|
+ }{result1}
|
|
|
+}
|
|
|
+
|
|
|
func (fake *mockedConnectionInfo) Priority() int {
|
|
|
fake.priorityMutex.Lock()
|
|
|
ret, specificReturn := fake.priorityReturnsOnCall[len(fake.priorityArgsForCall)]
|
|
@@ -460,6 +523,8 @@ func (fake *mockedConnectionInfo) Invocations() map[string][][]interface{} {
|
|
|
defer fake.cryptoMutex.RUnlock()
|
|
|
fake.establishedAtMutex.RLock()
|
|
|
defer fake.establishedAtMutex.RUnlock()
|
|
|
+ fake.isLocalMutex.RLock()
|
|
|
+ defer fake.isLocalMutex.RUnlock()
|
|
|
fake.priorityMutex.RLock()
|
|
|
defer fake.priorityMutex.RUnlock()
|
|
|
fake.remoteAddrMutex.RLock()
|