FederatedBigQueryOutputCommitter extends ForwardingBigQueryFileOutputCommitter { @Override public void abortJob(JobContext context, State state) throws IOException { super.abortJob(context, state); cleanup(context); } FederatedBigQueryOutputCommitter(TaskAttemptContext context, OutputCommitter delegate); @Override void commitJob(JobContext context); @Override void abortJob(JobContext context, State state);  }
@Test public void testAbortJob() throws IOException { generateSampleFiles(); committer.abortJob(mockTaskAttemptContext, State.KILLED); assertThat(!ghfs.exists(outputPath)).isTrue(); assertThat(!ghfs.exists(outputSampleFilePath)).isTrue(); verify(mockCommitter).abortJob(eq(mockTaskAttemptContext), eq(State.KILLED)); }