@Before public void setUp() { MockitoAnnotations.initMocks(this); // Boilerplate AccelerationManager accelerationManager = mock(AccelerationManager.class); AccelerationDetailsPopulator populator = mock(AccelerationDetailsPopulator.class); when(populator.computeAcceleration()).thenReturn(ByteString.EMPTY_BYTE_ARRAY); when(accelerationManager.newPopulator()).thenReturn(populator); when(context.getAccelerationManager()).thenReturn(accelerationManager); when(context.getQueryUserName()).thenReturn("myuser"); when(context.getSession()).thenReturn(UserSession.Builder.newBuilder().build()); when(context.getNonDefaultOptions()).thenReturn(new OptionList()); when(catalog.getMetadataStatsCollector()).thenReturn(new MetadataStatsCollector()); }
public QueryManager( final QueryId queryId, final QueryContext context, final CoordToExecTunnelCreator tunnelCreator, final CompletionListener completionListener, final Pointer<QueryId> prepareId, final AttemptObservers observers, final boolean verboseProfiles, final boolean includeDatasetProfiles, final Catalog catalog) { this.queryId = queryId; this.tunnelCreator = tunnelCreator; this.completionListener = completionListener; this.context = context; this.prepareId = prepareId; this.catalog = catalog; this.nonDefaultOptions = context.getNonDefaultOptions(); resourceAllocationResultObserver = new ResourceAllocationResultObserver(); observers.add(resourceAllocationResultObserver); capturer = new PlanCaptureAttemptObserver(verboseProfiles, includeDatasetProfiles, context.getFunctionRegistry(), context.getAccelerationManager().newPopulator()); observers.add(capturer); observers.add(new TimeMarker()); }
final SimpleParallelizer parallelizer = new SimpleParallelizer(queryContext, observer); final OptionList fragmentOptions = queryContext.getNonDefaultOptions();
final OptionList fragmentOptions = queryContext.getNonDefaultOptions();