func (m *MockNetworkTransport) Connect(arg0 context.Context, arg1 peer.AddrInfo) error {
    m.ctrl.T.Helper()
    ret := m.ctrl.Call(m, "Connect", arg0, arg1)
    ret0, _ := ret[0].(error)
    return ret0