protected final Step newEndpointStep(String connectorId, String actionId, Consumer<Connection.Builder> connectionConsumer, Consumer<Step.Builder> stepConsumer) throws IOException { Connector connector = resourceManager.mandatoryLoadConnector(connectorId); ConnectorAction action = resourceManager.mandatoryLookupAction(connector, actionId); Connection.Builder connectionBuilder = new Connection.Builder().connector(connector); connectionConsumer.accept(connectionBuilder); Step.Builder stepBuilder = new Step.Builder().stepKind(StepKind.endpoint).action(action).connection(connectionBuilder.build()); stepConsumer.accept(stepBuilder); return stepBuilder.build(); }
public Step toCurrentSteps(Step step) { Step.Builder from = new Step.Builder().createFrom(step); step.getConnection().ifPresent(c->{ from.connection(toCurrentConnection(c)); }); return from.build(); }
@Test public void shouldAugmentWithConnectorUsage() { final Connector connector1 = newConnector("1"); final Connector connector2 = newConnector("2"); final Connector connector3 = newConnector("3"); final Step step1a = new Step.Builder().action(newActionBy(connector1)).build(); final Step step1b = new Step.Builder().action(newActionBy(connector1)).build(); final Step step2 = new Step.Builder().action(newActionBy(connector2)).build(); final Integration deployment1 = newIntegration(Arrays.asList(step1a, step1b)); final Integration deployment2 = newIntegration(Collections.singletonList(step2)); final Integration deployment3 = newIntegration(Collections.singletonList(step2)); when(dataManager.fetchAll(Integration.class)) .thenReturn(new ListResult.Builder<Integration>().addItem(deployment1, deployment2, deployment3).build()); final List<Connector> augmented = handler.augmentedWithUsage(Arrays.asList(connector1, connector2, connector3)); assertThat(augmented).contains(usedConnector(connector1, 1), usedConnector(connector2, 2), usedConnector(connector3, 0)); }
protected final Step newSimpleEndpointStep(String scheme, Consumer<ConnectorDescriptor.Builder> consumer) { ConnectorDescriptor.Builder builder = new ConnectorDescriptor.Builder().componentScheme(scheme); consumer.accept(builder); return new Step.Builder() .stepKind(StepKind.endpoint) .action(new ConnectorAction.Builder() .descriptor(builder.build()) .build()) .build(); }