src/test/java/org/embulk/input/marketo/rest/MarketoRestClientTest.java in embulk-input-marketo-0.6.15 vs src/test/java/org/embulk/input/marketo/rest/MarketoRestClientTest.java in embulk-input-marketo-0.6.16

- old
+ new

@@ -33,10 +33,11 @@ import java.io.IOException; import java.io.InputStream; import java.nio.ByteBuffer; import java.text.SimpleDateFormat; +import java.util.ArrayList; import java.util.Arrays; import java.util.Date; import java.util.HashMap; import java.util.Iterator; import java.util.List; @@ -147,15 +148,16 @@ { MarketoResponse<ObjectNode> marketoResponse = new MarketoResponse<>(); marketoResponse.setSuccess(true); Date startDate = new Date(1506865856000L); Date endDate = new Date(1507297856000L); + List<Integer> activityTypeIds = new ArrayList<>(); ObjectNode bulkExtractResult = OBJECT_MAPPER.createObjectNode(); bulkExtractResult.set("exportId", new TextNode("bulkExtractId")); marketoResponse.setResult(Arrays.asList(bulkExtractResult)); ArgumentCaptor<String> argumentCaptor = ArgumentCaptor.forClass(String.class); Mockito.doReturn(marketoResponse).when(marketoRestClient).doPost(Mockito.eq(END_POINT + MarketoRESTEndpoint.CREATE_ACTIVITY_EXTRACT.getEndpoint()), Mockito.isNull(Map.class), Mockito.isNull(ImmutableListMultimap.class), argumentCaptor.capture(), Mockito.any(MarketoResponseJetty92EntityReader.class)); - String bulkExtractId = marketoRestClient.createActivityExtract(startDate, endDate); + String bulkExtractId = marketoRestClient.createActivityExtract(activityTypeIds, startDate, endDate); Assert.assertEquals("bulkExtractId", bulkExtractId); String postContent = argumentCaptor.getValue(); ObjectNode marketoBulkExtractRequest = (ObjectNode) OBJECT_MAPPER.readTree(postContent); ObjectNode filter = (ObjectNode) marketoBulkExtractRequest.get("filter"); Assert.assertTrue(filter.has("createdAt"));