Search in sources :

Example 1 with LeadChangeRecord

use of org.talend.components.marketo.runtime.client.rest.type.LeadChangeRecord in project components by Talend.

the class MarketoLeadClient method convertLeadChangesRecords.

public List<IndexedRecord> convertLeadChangesRecords(List<LeadChangeRecord> recordList, Schema schema, Map<String, String> mappings) {
    List<IndexedRecord> results = new ArrayList<>();
    Gson gson = new Gson();
    for (LeadChangeRecord input : recordList) {
        IndexedRecord record = new Record(schema);
        for (Field f : schema.getFields()) {
            String col = mappings.get(f.name());
            switch(col) {
                case FIELD_ID:
                    record.put(f.pos(), input.getId());
                    break;
                case FIELD_MARKETO_GUID:
                    record.put(f.pos(), input.getMarketoGUID());
                    break;
                case FIELD_LEAD_ID:
                    record.put(f.pos(), input.getLeadId());
                    break;
                case FIELD_ACTIVITY_DATE:
                    record.put(f.pos(), input.getActivityDate().getTime());
                    break;
                case FIELD_ACTIVITY_TYPE_ID:
                    record.put(f.pos(), input.getActivityTypeId());
                    break;
                case FIELD_ACTIVITY_TYPE_VALUE:
                    record.put(f.pos(), getActivityTypeNameById(input.getActivityTypeId()));
                    break;
                case FIELD_FIELDS:
                    record.put(f.pos(), gson.toJson(input.getFields()));
                    break;
                case FIELD_CAMPAIGN_ID:
                    record.put(f.pos(), input.getCampaignId());
                    break;
                default:
                    String attr = input.getMktoAttributes().get(col);
                    if (attr != null) {
                        record.put(f.pos(), attr);
                    }
            }
        }
        results.add(record);
    }
    return results;
}
Also used : Field(org.apache.avro.Schema.Field) IndexedRecord(org.apache.avro.generic.IndexedRecord) ArrayList(java.util.ArrayList) Gson(com.google.gson.Gson) Record(org.apache.avro.generic.GenericData.Record) IndexedRecord(org.apache.avro.generic.IndexedRecord) LeadChangeRecord(org.talend.components.marketo.runtime.client.rest.type.LeadChangeRecord) LeadActivityRecord(org.talend.components.marketo.runtime.client.rest.type.LeadActivityRecord) LeadChangeRecord(org.talend.components.marketo.runtime.client.rest.type.LeadChangeRecord)

Example 2 with LeadChangeRecord

use of org.talend.components.marketo.runtime.client.rest.type.LeadChangeRecord in project components by Talend.

the class MarketoLeadClientTest method testGetLeadChanges.

@Test
public void testGetLeadChanges() throws Exception {
    iprops.inputOperation.setValue(InputOperation.getLeadChanges);
    iprops.sinceDateTime.setValue("2017-01-30 10:00:00 +0100");
    iprops.afterInputOperation();
    List<Field> moreFields = new ArrayList<>();
    moreFields.add(new Field("attrName", AvroUtils._string(), "", null));
    moreFields.add(new Field("campaignId", AvroUtils._int(), "", null));
    iprops.schemaInput.schema.setValue(MarketoUtils.newSchema(iprops.schemaInput.schema.getValue(), "test", moreFields));
    iprops.beforeMappingInput();
    // 
    doThrow(new MarketoException("REST", "error")).when(client).executeGetRequest(LeadChangesResult.class);
    mktoRR = client.getLeadChanges(iprops, null);
    assertFalse(mktoRR.isSuccess());
    assertFalse(mktoRR.getErrorsString().isEmpty());
    // 
    doReturn(new LeadChangesResult()).when(client).executeGetRequest(LeadChangesResult.class);
    mktoRR = client.getLeadChanges(iprops, null);
    assertFalse(mktoRR.isSuccess());
    // 
    LeadChangesResult lcr = new LeadChangesResult();
    lcr.setSuccess(true);
    List<LeadChangeRecord> lcrs = new ArrayList<>();
    LeadChangeRecord lc = new LeadChangeRecord();
    lc.setId(123456);
    lc.setMarketoGUID("ABC-123-DEF");
    lc.setLeadId(12345);
    lc.setActivityTypeId(1);
    lc.setActivityTypeValue("Visit Webpage");
    lc.setActivityDate(new Date());
    lc.setCampaignId(456);
    List<Map<String, String>> fields = new ArrayList<>();
    Map<String, String> field = new HashMap<>();
    field.put("field1", "value1");
    fields.add(field);
    lc.setFields(fields);
    List<Map<String, Object>> attributes = new ArrayList<>();
    Map<String, Object> attribute = new HashMap<>();
    attribute.put("name", "attrName");
    attribute.put("value", "attrValue");
    attributes.add(attribute);
    lc.setAttributes(attributes);
    lcrs.add(lc);
    lcr.setResult(lcrs);
    // 
    doReturn(lcr).when(client).executeGetRequest(LeadChangesResult.class);
    mktoRR = client.getLeadChanges(iprops, null);
    assertTrue(mktoRR.isSuccess());
    List<IndexedRecord> records = mktoRR.getRecords();
    assertNotNull(records);
    IndexedRecord record = records.get(0);
    assertNotNull(record);
    Schema refSchema = iprops.schemaInput.schema.getValue();
    assertEquals(refSchema, record.getSchema());
    assertEquals(123456, record.get(refSchema.getField("id").pos()));
    assertEquals("ABC-123-DEF", record.get(refSchema.getField("marketoGUID").pos()));
    assertEquals(12345, record.get(refSchema.getField("leadId").pos()));
    assertEquals(1, record.get(refSchema.getField("activityTypeId").pos()));
    assertEquals("Visit Webpage", record.get(refSchema.getField("activityTypeValue").pos()));
    assertEquals(456, record.get(refSchema.getField("campaignId").pos()));
    assertEquals("[{\"field1\":\"value1\"}]", record.get(refSchema.getField("fields").pos()));
    assertTrue(record.get(refSchema.getField("activityDate").pos()) instanceof Long);
    assertEquals("attrValue", record.get(refSchema.getField("attrName").pos()));
}
Also used : IndexedRecord(org.apache.avro.generic.IndexedRecord) HashMap(java.util.HashMap) MarketoException(org.talend.components.marketo.runtime.client.type.MarketoException) Schema(org.apache.avro.Schema) ArrayList(java.util.ArrayList) Matchers.anyString(org.mockito.Matchers.anyString) Date(java.util.Date) LeadChangesResult(org.talend.components.marketo.runtime.client.rest.response.LeadChangesResult) LeadChangeRecord(org.talend.components.marketo.runtime.client.rest.type.LeadChangeRecord) Field(org.apache.avro.Schema.Field) JsonObject(com.google.gson.JsonObject) Map(java.util.Map) HashMap(java.util.HashMap) Test(org.junit.Test)

Aggregations

ArrayList (java.util.ArrayList)2 Field (org.apache.avro.Schema.Field)2 IndexedRecord (org.apache.avro.generic.IndexedRecord)2 LeadChangeRecord (org.talend.components.marketo.runtime.client.rest.type.LeadChangeRecord)2 Gson (com.google.gson.Gson)1 JsonObject (com.google.gson.JsonObject)1 Date (java.util.Date)1 HashMap (java.util.HashMap)1 Map (java.util.Map)1 Schema (org.apache.avro.Schema)1 Record (org.apache.avro.generic.GenericData.Record)1 Test (org.junit.Test)1 Matchers.anyString (org.mockito.Matchers.anyString)1 LeadChangesResult (org.talend.components.marketo.runtime.client.rest.response.LeadChangesResult)1 LeadActivityRecord (org.talend.components.marketo.runtime.client.rest.type.LeadActivityRecord)1 MarketoException (org.talend.components.marketo.runtime.client.type.MarketoException)1