(t *testing.T)
| 45 | } |
| 46 | |
| 47 | func newMockService(t *testing.T) *Service { |
| 48 | ctr := gomock.NewController(t) |
| 49 | |
| 50 | clusterMgrCli := NewMockClusterMgrAPI(ctr) |
| 51 | blobDeleteMgr := NewMockTaskRunner(ctr) |
| 52 | shardRepairMgr := NewMockTaskRunner(ctr) |
| 53 | diskDropMgr := NewMockMigrater(ctr) |
| 54 | diskRepairMgr := NewMockMigrater(ctr) |
| 55 | manualMgr := NewMockMigrater(ctr) |
| 56 | balanceMgr := NewMockMigrater(ctr) |
| 57 | inspectorMgr := NewMockVolumeInspector(ctr) |
| 58 | clusterTopology := NewMockClusterTopology(ctr) |
| 59 | |
| 60 | // return disk repair task |
| 61 | diskRepairMgr.EXPECT().AcquireTask(any, any).Return(proto.MigrateTask{TaskType: proto.TaskTypeDiskRepair}, nil) |
| 62 | |
| 63 | // reclaim repair task |
| 64 | diskRepairMgr.EXPECT().ReclaimTask(any, any, any, any, any, any).Return(nil) |
| 65 | clusterMgrCli.EXPECT().AllocVolumeUnit(any, any).Return(&client.AllocVunitInfo{}, nil) |
| 66 | // reclaim balance task |
| 67 | balanceMgr.EXPECT().ReclaimTask(any, any, any, any, any, any).Return(nil) |
| 68 | clusterMgrCli.EXPECT().AllocVolumeUnit(any, any).Return(&client.AllocVunitInfo{}, nil) |
| 69 | // reclaim disk drop task |
| 70 | diskDropMgr.EXPECT().ReclaimTask(any, any, any, any, any, any).Return(nil) |
| 71 | clusterMgrCli.EXPECT().AllocVolumeUnit(any, any).Return(&client.AllocVunitInfo{}, nil) |
| 72 | // reclaim manual migrate task |
| 73 | manualMgr.EXPECT().ReclaimTask(any, any, any, any, any, any).Return(nil) |
| 74 | clusterMgrCli.EXPECT().AllocVolumeUnit(any, any).Return(&client.AllocVunitInfo{}, nil) |
| 75 | |
| 76 | // cancel repair task |
| 77 | diskRepairMgr.EXPECT().CancelTask(any, any).Return(nil) |
| 78 | // cancel balance task |
| 79 | balanceMgr.EXPECT().CancelTask(any, any).Return(nil) |
| 80 | // cancel disk drop task |
| 81 | diskDropMgr.EXPECT().CancelTask(any, any).Return(nil) |
| 82 | // cancel manual migrate task |
| 83 | manualMgr.EXPECT().CancelTask(any, any).Return(nil) |
| 84 | |
| 85 | // complete repair task |
| 86 | diskRepairMgr.EXPECT().CompleteTask(any, any).Return(nil) |
| 87 | // complete balance task |
| 88 | balanceMgr.EXPECT().CompleteTask(any, any).Return(nil) |
| 89 | // complete disk drop task |
| 90 | diskDropMgr.EXPECT().CompleteTask(any, any).Return(nil) |
| 91 | // complete manual migrate task |
| 92 | manualMgr.EXPECT().CompleteTask(any, any).Return(nil) |
| 93 | |
| 94 | // renewal repair task |
| 95 | diskRepairMgr.EXPECT().RenewalTask(any, any, any).Times(3).Return(nil) |
| 96 | // renewal balance task |
| 97 | balanceMgr.EXPECT().RenewalTask(any, any, any).Times(3).Return(nil) |
| 98 | // renewal disk drop task |
| 99 | diskDropMgr.EXPECT().RenewalTask(any, any, any).Times(3).Return(nil) |
| 100 | // renewal manual migrate task |
| 101 | manualMgr.EXPECT().RenewalTask(any, any, any).Times(3).Return(nil) |
| 102 | |
| 103 | // report repair task |
| 104 | diskRepairMgr.EXPECT().ReportWorkerTaskStats(any).Return() |
no test coverage detected