public PutRecordsResult unmarshall(JsonUnmarshallerContext context) throws Exception { PutRecordsResult putRecordsResult = new PutRecordsResult();
public PutRecordsResult unmarshall(JsonUnmarshallerContext context) throws Exception { PutRecordsResult putRecordsResult = new PutRecordsResult(); AwsJsonReader reader = context.getReader(); reader.beginObject(); while (reader.hasNext()) { String name = reader.nextName(); if (name.equals("FailedRecordCount")) { putRecordsResult.setFailedRecordCount(IntegerJsonUnmarshaller.getInstance() .unmarshall(context)); } else if (name.equals("Records")) { putRecordsResult.setRecords(new ListUnmarshaller<PutRecordsResultEntry>( PutRecordsResultEntryJsonUnmarshaller.getInstance() ) .unmarshall(context)); } else if (name.equals("EncryptionType")) { putRecordsResult.setEncryptionType(StringJsonUnmarshaller.getInstance() .unmarshall(context)); } else { reader.skipValue(); } } reader.endObject(); return putRecordsResult; }
@Test(expected = AmazonClientException.class) public void testSendBatchException() { String streamName = "stream"; int count = 10; // create an ok result PutRecordsResult result = new PutRecordsResult(); List<PutRecordsResultEntry> entries = new ArrayList<PutRecordsResultEntry>(); result.setFailedRecordCount(0); for (int i = 0; i < count; i++) { PutRecordsResultEntry entry = new PutRecordsResultEntry(); entry.setSequenceNumber("record_id_" + i); entries.add(entry); } result.setRecords(entries); // create data List<byte[]> data = new ArrayList<byte[]>(); for (int i = 0; i < count; i++) { data.add(("record" + i).getBytes(StringUtils.UTF8)); } Mockito.when(client.putRecords(any(PutRecordsRequest.class))).thenThrow( new InvalidArgumentException("invalid argument")); sender.sendBatch(streamName, data); }
PutRecordsResult result = new PutRecordsResult(); List<PutRecordsResultEntry> entries = new ArrayList<PutRecordsResultEntry>(); result.setFailedRecordCount(5);
PutRecordsResult result = new PutRecordsResult(); List<PutRecordsResultEntry> entries = new ArrayList<PutRecordsResultEntry>(); result.setFailedRecordCount(0);
.willAnswer(invocation -> { AsyncHandler<PutRecordsRequest, PutRecordsResult> handler = invocation.getArgument(1); handler.onSuccess(new PutRecordsRequest(), new PutRecordsResult()); return mock(Future.class); })
public PutRecordsResult unmarshall(JsonUnmarshallerContext context) throws Exception { PutRecordsResult putRecordsResult = new PutRecordsResult();