Examples with LookupFieldsParameter org.motechproject.tasks.domain.mds.task.LookupFieldsParameter used on opensource projects

Example 6 with LookupFieldsParameter

use of org.motechproject.tasks.domain.mds.task.LookupFieldsParameter in project motech by motech.

the class TaskDataProviderServiceImplTest method shouldRegisterProviderFromString.

@Test
public void shouldRegisterProviderFromString() throws IOException {
    String fieldParameterAsJson = "{ displayName: 'displayName', fieldKey: 'fieldKey' }";
    String lookupFieldsAsJson = "['lookupField']";
    String objectAsJson = String.format("{ displayName: 'displayName', type: 'type', lookupFields: %s, fields: [%s]}", lookupFieldsAsJson, fieldParameterAsJson);
    String providerAsJson = String.format("{ name: '%s', objects: [%s] }", PROVIDER_NAME, objectAsJson);
    Type type = new TypeToken<TaskDataProvider>() {
    }.getType();
    StringWriter writer = new StringWriter();
    List<LookupFieldsParameter> lookupFields = asList(new LookupFieldsParameter("lookupField", asList("lookupField")));
    List<FieldParameter> fields = asList(new FieldParameter("displayName", "fieldKey"));
    List<TaskDataProviderObject> objects = new ArrayList<>();
    objects.add(new TaskDataProviderObject("displayName", "type", lookupFields, fields));
    TaskDataProvider provider = new TaskDataProvider(PROVIDER_NAME, objects);
    when(motechJsonReader.readFromStream(any(InputStream.class), eq(type))).thenReturn(provider);
    taskDataProviderService.registerProvider(providerAsJson);
    ArgumentCaptor<InputStream> captor = ArgumentCaptor.forClass(InputStream.class);
    verify(motechJsonReader).readFromStream(captor.capture(), eq(type));
    verify(dataProviderDataService).create(provider);
    InputStream value = captor.getValue();
    IOUtils.copy(value, writer);
    assertTrue(value instanceof ByteArrayInputStream);
    assertEquals(providerAsJson, writer.toString());
}

Example 7 with LookupFieldsParameter

use of org.motechproject.tasks.domain.mds.task.LookupFieldsParameter in project motech by motech.

the class TaskDataProviderServiceImplTest method shouldSendEventWhenProviderWasUpdated.

@Test
public void shouldSendEventWhenProviderWasUpdated() {
    Type type = new TypeToken<TaskDataProvider>() {
    }.getType();
    List<TaskDataProviderObject> objects = new ArrayList<>();
    List<LookupFieldsParameter> lookupFields = asList(new LookupFieldsParameter("lookupField", asList("lookupField")));
    List<FieldParameter> fields = asList(new FieldParameter("displayName", "fieldKey"));
    objects.add(new TaskDataProviderObject("displayName", "type", lookupFields, fields));
    TaskDataProvider provider = new TaskDataProvider(PROVIDER_NAME, objects);
    List<TaskDataProviderObject> updatedObjects = new ArrayList<>(objects);
    updatedObjects.add(new TaskDataProviderObject("displayName2", "type2", lookupFields, fields));
    TaskDataProvider updatedProvider = new TaskDataProvider(PROVIDER_NAME, updatedObjects);
    when(motechJsonReader.readFromStream(inputStream, type)).thenReturn(updatedProvider);
    when(dataProviderDataService.findByName(PROVIDER_NAME)).thenReturn(provider);
    ArgumentCaptor<MotechEvent> captor = ArgumentCaptor.forClass(MotechEvent.class);
    taskDataProviderService.registerProvider(inputStream);
    ArgumentCaptor<TaskDataProvider> taskDataProviderArgumentCaptor = ArgumentCaptor.forClass(TaskDataProvider.class);
    verify(dataProviderDataService).update(taskDataProviderArgumentCaptor.capture());
    assertEquals(PROVIDER_NAME, taskDataProviderArgumentCaptor.getValue().getName());
    verify(dataProviderDataService).update(provider);
    verify(eventRelay).sendEventMessage(captor.capture());
    MotechEvent event = captor.getValue();
    assertEquals(DATA_PROVIDER_UPDATE_SUBJECT, event.getSubject());
    assertEquals(PROVIDER_NAME, event.getParameters().get(DATA_PROVIDER_NAME));
}

Example 8 with LookupFieldsParameter

use of org.motechproject.tasks.domain.mds.task.LookupFieldsParameter in project motech by motech.

the class TaskServiceImplTest method shouldValidateTasksAfterChannelUpdateForInvalidTaskDataProviders.

@Test
public void shouldValidateTasksAfterChannelUpdateForInvalidTaskDataProviders() {
    TaskConfig config = new TaskConfig().add(new DataSource("TestProvider", 1234L, 1L, "Test", "id", "specifiedName", asList(new Lookup("id", "trigger.value")), true));
    Task task = new Task("name", trigger, asList(action), config, true, false);
    task.setId(12345l);
    List<LookupFieldsParameter> lookupFields = new ArrayList<>();
    lookupFields.add(new LookupFieldsParameter("property", asList("property")));
    TaskDataProvider provider = new TaskDataProvider("TestProvider", asList(new TaskDataProviderObject("test", "Test", lookupFields, null)));
    provider.setId(1234L);
    Channel triggerChannel = new Channel("test", "test-trigger", "0.15", "", asList(new TriggerEvent("send", "SEND", "", asList(new EventParameter("test", "value")), "")), null);
    Channel actionChannel = new Channel("test", "test-action", "0.14", "", null, asList(new ActionEventBuilder().setDisplayName("receive").setSubject("RECEIVE").setDescription("").setActionParameters(null).build()));
    when(tasksDataService.findById(12345l)).thenReturn(task);
    when(tasksDataService.retrieveAll()).thenReturn(asList(task));
    when(providerService.getProvider(provider.getName())).thenReturn(provider);
    when(channelService.getChannel(trigger.getModuleName())).thenReturn(triggerChannel);
    when(channelService.getChannel(action.getModuleName())).thenReturn(actionChannel);
    taskService.validateTasksAfterTaskDataProviderUpdate(getProviderUpdateEvent(provider.getName()));
    Task actualTask = verifyUpdateAndCaptureTask();
    assertFalse(actualTask.isEnabled());
    List<Object> errors = new ArrayList<>(actualTask.getValidationErrors());
    assertEquals(1, errors.size());
    assertThat(errors, hasItem(hasProperty("message", equalTo("task.validation.error.providerObjectLookupNotExist"))));
}

Example 9 with LookupFieldsParameter

use of org.motechproject.tasks.domain.mds.task.LookupFieldsParameter in project motech by motech.

the class TaskServiceImplTest method shouldSaveTask.

@Test
public void shouldSaveTask() {
    Map<String, String> map = new HashMap<>();
    map.put("phone", "12345");
    TaskConfig config = new TaskConfig().add(new DataSource("TestProvider", 1234L, 1L, "Test", "id", "specifiedName", asList(new Lookup("id", "trigger.value")), true));
    action.setValues(map);
    Task task = new Task("name", trigger, asList(action), config, true, false);
    task.setNumberOfRetries(5);
    task.setRetryTaskOnFailure(true);
    Channel triggerChannel = new Channel("test", "test-trigger", "0.15", "", asList(new TriggerEvent("send", "SEND", "", asList(new EventParameter("test", "value")), "")), null);
    ActionEvent actionEvent = new ActionEventBuilder().setDisplayName("receive").setSubject("RECEIVE").setDescription("").setActionParameters(null).build();
    actionEvent.addParameter(new ActionParameterBuilder().setDisplayName("Phone").setKey("phone").build(), true);
    Channel actionChannel = new Channel("test", "test-action", "0.14", "", null, asList(actionEvent));
    TaskDataProvider provider = new TaskDataProvider("TestProvider", asList(new TaskDataProviderObject("test", "Test", asList(new LookupFieldsParameter("id", asList("id"))), null)));
    provider.setId(1234L);
    when(channelService.getChannel(trigger.getModuleName())).thenReturn(triggerChannel);
    when(channelService.getChannel(action.getModuleName())).thenReturn(actionChannel);
    when(providerService.getProvider("TestProvider")).thenReturn(provider);
    when(triggerEventService.triggerExists(task.getTrigger())).thenReturn(true);
    taskService.save(task);
    verify(triggerHandler).registerHandlerFor(task.getTrigger().getEffectiveListenerSubject());
    // Because task has set number of retries to 5, it should register retries handler for this task
    verify(triggerHandler).registerHandlerFor(task.getTrigger().getEffectiveListenerRetrySubject(), true);
    verifyCreateAndCaptureTask();
}