public PutRecordsResultEntry unmarshall(JsonUnmarshallerContext context) throws Exception { PutRecordsResultEntry putRecordsResultEntry = new PutRecordsResultEntry();
@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); }
result.setFailedRecordCount(5); for (int i = 0; i < count; i++) { PutRecordsResultEntry entry = new PutRecordsResultEntry();
result.setFailedRecordCount(0); for (int i = 0; i < count; i++) { PutRecordsResultEntry entry = new PutRecordsResultEntry(); entry.setSequenceNumber("record_id_" + i); entries.add(entry);
public PutRecordsResultEntry unmarshall(JsonUnmarshallerContext context) throws Exception { AwsJsonReader reader = context.getReader(); if (!reader.isContainer()) { reader.skipValue(); return null; } PutRecordsResultEntry putRecordsResultEntry = new PutRecordsResultEntry(); reader.beginObject(); while (reader.hasNext()) { String name = reader.nextName(); if (name.equals("SequenceNumber")) { putRecordsResultEntry.setSequenceNumber(StringJsonUnmarshaller.getInstance() .unmarshall(context)); } else if (name.equals("ShardId")) { putRecordsResultEntry.setShardId(StringJsonUnmarshaller.getInstance() .unmarshall(context)); } else if (name.equals("ErrorCode")) { putRecordsResultEntry.setErrorCode(StringJsonUnmarshaller.getInstance() .unmarshall(context)); } else if (name.equals("ErrorMessage")) { putRecordsResultEntry.setErrorMessage(StringJsonUnmarshaller.getInstance() .unmarshall(context)); } else { reader.skipValue(); } } reader.endObject(); return putRecordsResultEntry; }
public PutRecordsResultEntry unmarshall(JsonUnmarshallerContext context) throws Exception { PutRecordsResultEntry putRecordsResultEntry = new PutRecordsResultEntry();