Skip to content

Commit

Permalink
Merge 5a9f2c1 into 2e49ce8
Browse files Browse the repository at this point in the history
  • Loading branch information
yirutang authored Mar 2, 2021
2 parents 2e49ce8 + 5a9f2c1 commit 628de81
Showing 1 changed file with 24 additions and 12 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -471,19 +471,23 @@ public void testBatchIsFlushedWithError() throws Exception {
.build());

ApiFuture<AppendRowsResponse> appendFuture1 = sendTestMessage(writer, new String[] {"A"});
ApiFuture<AppendRowsResponse> appendFuture2 = sendTestMessage(writer, new String[] {"B"});
ApiFuture<AppendRowsResponse> appendFuture3 = sendTestMessage(writer, new String[] {"C"});
try {
appendFuture2.get();
} catch (ExecutionException ex) {
assertEquals(DataLossException.class, ex.getCause().getClass());
}
assertFalse(appendFuture3.isDone());
writer.shutdown();
try {
appendFuture3.get();
} catch (ExecutionException ex) {
assertEquals(AbortedException.class, ex.getCause().getClass());
ApiFuture<AppendRowsResponse> appendFuture2 = sendTestMessage(writer, new String[] {"B"});
ApiFuture<AppendRowsResponse> appendFuture3 = sendTestMessage(writer, new String[] {"C"});
try {
appendFuture2.get();
} catch (ExecutionException ex) {
assertEquals(DataLossException.class, ex.getCause().getClass());
}
assertFalse(appendFuture3.isDone());
writer.shutdown();
try {
appendFuture3.get();
} catch (ExecutionException ex) {
assertEquals(AbortedException.class, ex.getCause().getClass());
}
} catch (IllegalStateException ex) {
assertEquals("Stream already failed.", ex.getMessage());
}
}
}
Expand Down Expand Up @@ -739,6 +743,14 @@ public Throwable call() {
}
}
return null;
} catch (ExecutionException ex) {
// Some wiredness in test presubmit runs, it seems this thread is always started after
// the main thread.
if (ex.getCause().getClass() == AbortedException.class) {
return null;
} else {
return ex;
}
} catch (Exception e) {
return e;
}
Expand Down

0 comments on commit 628de81

Please sign in to comment.