@Override public JobDataFragment truncate(int maxRows) { return delegate.truncate(maxRows); }
@Override public JobDataFragment truncate(int maxRows) { return new JobDataFragmentWrapper(0, delegate.truncate(maxRows)); }
final JobDataFragment data = job.getData().truncate(1);
@Override public JobUI answer(InvocationOnMock invocation) throws Throwable { String query = invocation.getArgumentAt(0, SqlQuery.class).getSql(); Job job = mock(Job.class); JobData jobData = mock(JobData.class); when(job.getData()).thenReturn(jobData); if ("SELECT * FROM dataset".equals(query)) { when(jobData.getJobResultsTable()).thenReturn("jobResults.previewJob"); } else if (query.contains("jobResults.previewJob")) { if (expFilter != null) { assertTrue(query, query.contains(expFilter)); } else { assertFalse(query, query.contains("WHERE")); } JobDataFragment fragment = mock(JobDataFragment.class); when(jobData.truncate(1)).thenReturn(fragment); when(fragment.getSchema()).thenReturn( BatchSchema.newBuilder() .addField(new Field("dremio_selection_count", true, new ArrowType.Int(64, true), null)) .build() ); when(fragment.extractValue("dremio_selection_count", 0)).thenReturn(expCount); } return new JobUI(job); } }
private RelNode getPlan(final String queryString) { final AtomicReference<RelNode> physical = new AtomicReference<>(null); final SqlQuery query = new SqlQuery(queryString, DEFAULT_USERNAME); final Job job = getJobsService().submitJob(JobRequest.newBuilder() .setSqlQuery(query) .setQueryType(QueryType.ACCELERATOR_CREATE) .setDatasetPath(DatasetPath.NONE.toNamespaceKey()) .setDatasetVersion(DatasetVersion.NONE) .build(), new NoOpJobStatusListener() { @Override public void planRelTransform(final PlannerPhase phase, final RelNode before, final RelNode after, final long millisTaken) { if (phase == PlannerPhase.PHYSICAL) { physical.set(after); } super.planRelTransform(phase, before, after, millisTaken); } }); job.getData().truncate(1); return physical.get(); } }
.build(), NoOpJobStatusListener.INSTANCE); job1_0.getData().loadIfNecessary(); JobDataFragment result = job1_0.getData().truncate(10); JobDataFragment storedResult = jobsService.getJob(job1_0.getJobId()).getData().truncate(10); for (Field column: result.getSchema()) { assertTrue(storedResult.getSchema().getFields().contains(column));
@Test public void testQueryOnFile() throws Exception { final JobsService jobService = l(JobsService.class); Job job = jobService.submitJob(JobRequest.newBuilder() .setSqlQuery(new SqlQuery("SELECT * FROM dachdfs_test.dir1.json.\"users.json\"", SampleDataPopulator.DEFAULT_USER_NAME)) .build(), NoOpJobStatusListener.INSTANCE); JobDataFragment jobData = job.getData().truncate(500); assertEquals(3, jobData.getReturnedRowCount()); assertEquals(2, jobData.getSchema().getFieldCount()); } }