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"));