|
|
@ -24,6 +24,7 @@ import org.apache.dolphinscheduler.remote.command.Message; |
|
|
|
import org.apache.dolphinscheduler.remote.command.MessageType; |
|
|
|
import org.apache.dolphinscheduler.remote.command.MessageType; |
|
|
|
import org.apache.dolphinscheduler.remote.command.task.TaskDispatchMessage; |
|
|
|
import org.apache.dolphinscheduler.remote.command.task.TaskDispatchMessage; |
|
|
|
import org.apache.dolphinscheduler.server.worker.config.WorkerConfig; |
|
|
|
import org.apache.dolphinscheduler.server.worker.config.WorkerConfig; |
|
|
|
|
|
|
|
import org.apache.dolphinscheduler.server.worker.registry.WorkerRegistryClient; |
|
|
|
import org.apache.dolphinscheduler.server.worker.rpc.WorkerMessageSender; |
|
|
|
import org.apache.dolphinscheduler.server.worker.rpc.WorkerMessageSender; |
|
|
|
import org.apache.dolphinscheduler.server.worker.rpc.WorkerRpcClient; |
|
|
|
import org.apache.dolphinscheduler.server.worker.rpc.WorkerRpcClient; |
|
|
|
import org.apache.dolphinscheduler.server.worker.runner.WorkerManagerThread; |
|
|
|
import org.apache.dolphinscheduler.server.worker.runner.WorkerManagerThread; |
|
|
@ -64,6 +65,9 @@ public class TaskDispatchProcessorTest { |
|
|
|
@Mock |
|
|
|
@Mock |
|
|
|
private StorageOperate storageOperate; |
|
|
|
private StorageOperate storageOperate; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@Mock |
|
|
|
|
|
|
|
private WorkerRegistryClient workerRegistryClient; |
|
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
@Test |
|
|
|
public void process() { |
|
|
|
public void process() { |
|
|
|
Channel channel = Mockito.mock(Channel.class); |
|
|
|
Channel channel = Mockito.mock(Channel.class); |
|
|
|