fix: Interval in SyncProbesResponse (#2466)
Signed-off-by: Gaius <gaius.qi@gmail.com>
This commit is contained in:
parent
cb5f336060
commit
5342fb97fc
|
|
@ -5,6 +5,7 @@
|
|||
package mocks
|
||||
|
||||
import (
|
||||
fs "io/fs"
|
||||
reflect "reflect"
|
||||
|
||||
gomock "github.com/golang/mock/gomock"
|
||||
|
|
@ -47,6 +48,20 @@ func (mr *MockDfpathMockRecorder) CacheDir() *gomock.Call {
|
|||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CacheDir", reflect.TypeOf((*MockDfpath)(nil).CacheDir))
|
||||
}
|
||||
|
||||
// CacheDirMode mocks base method.
|
||||
func (m *MockDfpath) CacheDirMode() fs.FileMode {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "CacheDirMode")
|
||||
ret0, _ := ret[0].(fs.FileMode)
|
||||
return ret0
|
||||
}
|
||||
|
||||
// CacheDirMode indicates an expected call of CacheDirMode.
|
||||
func (mr *MockDfpathMockRecorder) CacheDirMode() *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CacheDirMode", reflect.TypeOf((*MockDfpath)(nil).CacheDirMode))
|
||||
}
|
||||
|
||||
// DaemonLockPath mocks base method.
|
||||
func (m *MockDfpath) DaemonLockPath() string {
|
||||
m.ctrl.T.Helper()
|
||||
|
|
@ -89,6 +104,20 @@ func (mr *MockDfpathMockRecorder) DataDir() *gomock.Call {
|
|||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DataDir", reflect.TypeOf((*MockDfpath)(nil).DataDir))
|
||||
}
|
||||
|
||||
// DataDirMode mocks base method.
|
||||
func (m *MockDfpath) DataDirMode() fs.FileMode {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "DataDirMode")
|
||||
ret0, _ := ret[0].(fs.FileMode)
|
||||
return ret0
|
||||
}
|
||||
|
||||
// DataDirMode indicates an expected call of DataDirMode.
|
||||
func (mr *MockDfpathMockRecorder) DataDirMode() *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DataDirMode", reflect.TypeOf((*MockDfpath)(nil).DataDirMode))
|
||||
}
|
||||
|
||||
// DfgetLockPath mocks base method.
|
||||
func (m *MockDfpath) DfgetLockPath() string {
|
||||
m.ctrl.T.Helper()
|
||||
|
|
@ -144,3 +173,17 @@ func (mr *MockDfpathMockRecorder) WorkHome() *gomock.Call {
|
|||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WorkHome", reflect.TypeOf((*MockDfpath)(nil).WorkHome))
|
||||
}
|
||||
|
||||
// WorkHomeMode mocks base method.
|
||||
func (m *MockDfpath) WorkHomeMode() fs.FileMode {
|
||||
m.ctrl.T.Helper()
|
||||
ret := m.ctrl.Call(m, "WorkHomeMode")
|
||||
ret0, _ := ret[0].(fs.FileMode)
|
||||
return ret0
|
||||
}
|
||||
|
||||
// WorkHomeMode indicates an expected call of WorkHomeMode.
|
||||
func (mr *MockDfpathMockRecorder) WorkHomeMode() *gomock.Call {
|
||||
mr.mock.ctrl.T.Helper()
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WorkHomeMode", reflect.TypeOf((*MockDfpath)(nil).WorkHomeMode))
|
||||
}
|
||||
|
|
|
|||
|
|
@ -28,7 +28,6 @@ import (
|
|||
"go.opentelemetry.io/otel/trace"
|
||||
"google.golang.org/grpc/codes"
|
||||
"google.golang.org/grpc/status"
|
||||
"google.golang.org/protobuf/types/known/durationpb"
|
||||
|
||||
commonv1 "d7y.io/api/pkg/apis/common/v1"
|
||||
commonv2 "d7y.io/api/pkg/apis/common/v2"
|
||||
|
|
@ -711,7 +710,6 @@ func (v *V1) SyncProbes(stream schedulerv1.Scheduler_SyncProbesServer) error {
|
|||
logger.Infof("probe started: %#v", probedHosts)
|
||||
if err := stream.Send(&schedulerv1.SyncProbesResponse{
|
||||
Hosts: probedHosts,
|
||||
ProbeInterval: durationpb.New(v.config.NetworkTopology.Probe.Interval),
|
||||
}); err != nil {
|
||||
logger.Error(err)
|
||||
return err
|
||||
|
|
|
|||
|
|
@ -2715,7 +2715,6 @@ func TestServiceV1_SyncProbes(t *testing.T) {
|
|||
Idc: mockRawHost.Network.IDC,
|
||||
},
|
||||
},
|
||||
ProbeInterval: durationpb.New(mockNetworkTopologyConfig.Probe.Interval),
|
||||
})).Return(nil).Times(1),
|
||||
ms.Recv().Return(nil, io.EOF).Times(1),
|
||||
)
|
||||
|
|
@ -2947,7 +2946,6 @@ func TestServiceV1_SyncProbes(t *testing.T) {
|
|||
Idc: mockRawHost.Network.IDC,
|
||||
},
|
||||
},
|
||||
ProbeInterval: durationpb.New(mockNetworkTopologyConfig.Probe.Interval),
|
||||
})).Return(errors.New("send synchronize probes response error")).Times(1),
|
||||
)
|
||||
},
|
||||
|
|
|
|||
|
|
@ -751,7 +751,6 @@ func (v *V2) SyncProbes(stream schedulerv2.Scheduler_SyncProbesServer) error {
|
|||
logger.Infof("probe started: %#v", probedHosts)
|
||||
if err := stream.Send(&schedulerv2.SyncProbesResponse{
|
||||
Hosts: probedHosts,
|
||||
ProbeInterval: durationpb.New(v.config.NetworkTopology.Probe.Interval),
|
||||
}); err != nil {
|
||||
logger.Error(err)
|
||||
return err
|
||||
|
|
|
|||
|
|
@ -996,7 +996,6 @@ func TestServiceV2_SyncProbes(t *testing.T) {
|
|||
Build: mockV2Probe.Host.Build,
|
||||
},
|
||||
},
|
||||
ProbeInterval: durationpb.New(mockNetworkTopologyConfig.Probe.Interval),
|
||||
})).Return(nil).Times(1),
|
||||
ms.Recv().Return(nil, io.EOF).Times(1),
|
||||
)
|
||||
|
|
@ -1300,7 +1299,6 @@ func TestServiceV2_SyncProbes(t *testing.T) {
|
|||
Build: mockV2Probe.Host.Build,
|
||||
},
|
||||
},
|
||||
ProbeInterval: durationpb.New(mockNetworkTopologyConfig.Probe.Interval),
|
||||
})).Return(errors.New("send synchronize probes response error")).Times(1),
|
||||
)
|
||||
},
|
||||
|
|
|
|||
Loading…
Reference in New Issue