Revert "remove unused columns"

trocco-io · Oct 3, 2024 · 917d705 · 917d705
1 parent 9facb58
commit 917d705
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/build.gradle b/build.gradle
@@ -16,7 +16,7 @@ group = "com.treasuredata.embulk.plugins"
 description = "Loads records from Marketo."
 version = {
     def baseVersion = "0.6.26"
-    def troccoVersion = "0.3.0"
+    def troccoVersion = "0.1.1"
     def tag = "${baseVersion}-trocco-${troccoVersion}"
     def vd = versionDetails()
     if (vd.lastTag != "${tag}") {

diff --git a/src/main/java/org/embulk/input/marketo/delegate/ProgramInputPlugin.java b/src/main/java/org/embulk/input/marketo/delegate/ProgramInputPlugin.java
@@ -211,11 +211,15 @@ public ServiceResponseMapper<? extends ValueLocator> buildServiceResponseMapper(
                 .add("description", Types.STRING)
                 .add("createdAt", Types.TIMESTAMP, MarketoUtils.MARKETO_DATE_TIME_FORMAT)
                 .add("updatedAt", Types.TIMESTAMP, MarketoUtils.MARKETO_DATE_TIME_FORMAT)
+                .add("startDate", Types.TIMESTAMP, MarketoUtils.MARKETO_DATE_TIME_FORMAT)
+                .add("endDate", Types.TIMESTAMP, MarketoUtils.MARKETO_DATE_TIME_FORMAT)
                 .add("url", Types.STRING)
                 .add("type", Types.STRING)
                 .add("channel", Types.STRING)
                 .add("folder", Types.JSON)
                 .add("status", Types.STRING)
+                .add("costs", Types.JSON)
+                .add("tags", Types.JSON)
                 .add("workspace", Types.STRING);
         return builder.build();
     }

diff --git a/src/test/java/org/embulk/input/marketo/delegate/ProgramInputPluginTest.java b/src/test/java/org/embulk/input/marketo/delegate/ProgramInputPluginTest.java
@@ -263,8 +263,8 @@ private void testRun(ConfigSource config, Predicate<MarketoRestClient> expectedC
         expectedCall.test(mockRestClient);
 
         Schema embulkSchema = mapper.getEmbulkSchema();
-        // 13 columns
-        assertEquals(embulkSchema.size(), 13);
+        // 17 columns
+        assertEquals(embulkSchema.size(), 17);
         // verify 3 times the method setLong for column id has been called
         ArgumentCaptor<Long> longArgumentCaptor = ArgumentCaptor.forClass(Long.class);
         Mockito.verify(mockPageBuilder, Mockito.times(3)).setLong(Mockito.eq(embulkSchema.lookupColumn("id")), longArgumentCaptor.capture());