func (m *MockConn) NewStream(arg0 context.Context) (network.Stream, error) {
    m.ctrl.T.Helper()
    ret := m.ctrl.Call(m, "NewStream", arg0)
    ret0, _ := ret[0].(network.Stream)
    ret1, _ := ret[1].(error)