static void initRateLimiter(RateLimiter rateLimiter) { Map<TaskName, TaskModel> tasks = IntStream.range(0, NUMBER_OF_TASKS) .mapToObj(i -> new TaskName("task-" + i)) .collect(Collectors.toMap(Function.identity(), x -> mock(TaskModel.class))); ContainerModel containerModel = mock(ContainerModel.class); when(containerModel.getTasks()).thenReturn(tasks); JobModel jobModel = mock(JobModel.class); Map<String, ContainerModel> containerModelMap = new HashMap<>(); containerModelMap.put("container-1", containerModel); when(jobModel.getContainers()).thenReturn(containerModelMap); Context context = mock(Context.class); TaskContextImpl taskContext = mock(TaskContextImpl.class); when(context.getTaskContext()).thenReturn(taskContext); when(taskContext.getJobModel()).thenReturn(jobModel); when(context.getTaskContext().getTaskModel()).thenReturn(mock(TaskModel.class)); rateLimiter.init(context); } }
RateLimiter rateLimiter = deserializeObject(tableConfig, RemoteTableDescriptor.RATE_LIMITER); if (rateLimiter != null) { rateLimiter.init(this.context);
RateLimiter rateLimiter = deserializeObject(RATE_LIMITER); if (rateLimiter != null) { rateLimiter.init(this.context);
RateLimiter rateLimiter = deserializeObject(RATE_LIMITER); if (rateLimiter != null) { rateLimiter.init(this.context);
RateLimiter rateLimiter = deserializeObject(RATE_LIMITER); if (rateLimiter != null) { rateLimiter.init(this.context);
RateLimiter rateLimiter = deserializeObject(RATE_LIMITER); if (rateLimiter != null) { rateLimiter.init(this.context);