use of com.redhat.service.smartevents.manager.models.Processor in project sandbox by 5733d9e2be6485d52ffa08870cabdee0.
the class ConnectorWorkerTest method handleWorkProvisioningWithKnownResourceMultiplePasses.
@Transactional
@ParameterizedTest
@MethodSource("provideArgsForCreateTest")
void handleWorkProvisioningWithKnownResourceMultiplePasses(ManagedResourceStatus resourceStatus, ConnectorState connectorState, boolean useSourceConnectorEntity, RhoasTopicAccessType expectedTopicAccessType, ManagedResourceStatus expectedResourceStatus) {
Bridge bridge = Fixtures.createBridge();
Processor processor = Fixtures.createProcessor(bridge, ManagedResourceStatus.READY);
ConnectorEntity connectorEntity = useSourceConnectorEntity ? Fixtures.createSourceConnector(processor, resourceStatus) : Fixtures.createSinkConnector(processor, resourceStatus);
// The publishedAt date is set by the Worker so reset that set by the Fixture
connectorEntity.setPublishedAt(null);
bridgeDAO.persist(bridge);
processorDAO.persist(processor);
connectorsDAO.persist(connectorEntity);
Work work = workManager.schedule(connectorEntity);
Connector connector = new Connector();
connector.setId(TEST_CONNECTOR_EXTERNAL_ID);
connector.setStatus(new ConnectorStatusStatus().state(connectorState));
when(connectorsApi.getConnector(TEST_CONNECTOR_EXTERNAL_ID)).thenReturn(null, connector);
when(connectorsApi.createConnector(connectorEntity)).thenReturn(connector);
ConnectorEntity refreshed = worker.handleWork(work);
verify(rhoasService).createTopicAndGrantAccessFor(connectorEntity.getTopicName(), expectedTopicAccessType);
verify(connectorsApi).createConnector(connectorEntity);
assertThat(refreshed.getStatus()).isEqualTo(ManagedResourceStatus.PREPARING);
assertThat(refreshed.getDependencyStatus()).isEqualTo(ManagedResourceStatus.PROVISIONING);
// This emulates a subsequent invocation by WorkManager
refreshed = worker.handleWork(work);
verify(rhoasService, times(2)).createTopicAndGrantAccessFor(connectorEntity.getTopicName(), expectedTopicAccessType);
verify(connectorsApi, atMostOnce()).createConnector(connectorEntity);
assertThat(refreshed.getStatus()).isEqualTo(expectedResourceStatus);
assertThat(refreshed.getDependencyStatus()).isEqualTo(expectedResourceStatus);
if (expectedResourceStatus == ManagedResourceStatus.READY) {
assertThat(refreshed.getPublishedAt()).isNotNull();
} else {
assertThat(refreshed.getPublishedAt()).isNull();
}
assertThat(workManager.exists(work)).isTrue();
}
use of com.redhat.service.smartevents.manager.models.Processor in project sandbox by 5733d9e2be6485d52ffa08870cabdee0.
the class ProcessorWorkerTest method handleWorkProvisioningWithKnownResourceWithConnector.
@Transactional
@ParameterizedTest
@MethodSource("srcHandleWorkProvisioningWithKnownResourceWithConnector")
void handleWorkProvisioningWithKnownResourceWithConnector(ManagedResourceStatus status, ManagedResourceStatus statusWhenComplete, ManagedResourceStatus dependencyStatusWhenComplete, boolean isWorkComplete) {
Bridge bridge = Fixtures.createBridge();
Processor processor = Fixtures.createProcessor(bridge, ManagedResourceStatus.READY);
processor.setStatus(status);
ConnectorEntity connectorEntity = Fixtures.createSinkConnector(processor, ManagedResourceStatus.ACCEPTED);
connectorEntity.setStatus(ManagedResourceStatus.ACCEPTED);
bridgeDAO.persist(bridge);
processorDAO.persist(processor);
connectorsDAO.persist(connectorEntity);
Work work = workManager.schedule(processor);
doAnswer((i) -> {
// Emulate ConnectorWorker completing work
connectorEntity.setStatus(dependencyStatusWhenComplete);
return connectorEntity;
}).when(connectorWorker).handleWork(any(Work.class));
Processor refreshed = worker.handleWork(work);
assertThat(refreshed.getStatus()).isEqualTo(statusWhenComplete);
assertThat(refreshed.getDependencyStatus()).isEqualTo(dependencyStatusWhenComplete);
ArgumentCaptor<Work> workArgumentCaptor = ArgumentCaptor.forClass(Work.class);
verify(connectorWorker).handleWork(workArgumentCaptor.capture());
Work connectorWork = workArgumentCaptor.getValue();
assertThat(connectorWork).isNotNull();
assertThat(connectorWork.getId()).isEqualTo(work.getId());
assertThat(connectorWork.getManagedResourceId()).isEqualTo(connectorEntity.getId());
assertThat(connectorWork.getSubmittedAt()).isEqualTo(work.getSubmittedAt());
assertThat(workManager.exists(work)).isNotEqualTo(isWorkComplete);
}
use of com.redhat.service.smartevents.manager.models.Processor in project sandbox by 5733d9e2be6485d52ffa08870cabdee0.
the class ProcessorDAOTest method findByBridgeIdAndCustomerId.
@Test
public void findByBridgeIdAndCustomerId() {
Bridge b = createBridge();
Processor p = createProcessor(b, "foo");
Processor p1 = createProcessor(b, "bar");
ListResult<Processor> listResult = processorDAO.findByBridgeIdAndCustomerId(b.getId(), TestConstants.DEFAULT_CUSTOMER_ID, new QueryProcessorResourceInfo(0, 100));
assertThat(listResult.getPage()).isZero();
assertThat(listResult.getSize()).isEqualTo(2L);
assertThat(listResult.getTotal()).isEqualTo(2L);
listResult.getItems().forEach((px) -> assertThat(px.getId()).isIn(p.getId(), p1.getId()));
}
use of com.redhat.service.smartevents.manager.models.Processor in project sandbox by 5733d9e2be6485d52ffa08870cabdee0.
the class ProcessorDAOTest method testGetProcessorsFilterByNameWildcard.
@Test
public void testGetProcessorsFilterByNameWildcard() {
Bridge b = createBridge();
Processor p1 = createProcessor(b, "foo1");
Processor p2 = createProcessor(b, "foo2");
ListResult<Processor> results = processorDAO.findByBridgeIdAndCustomerId(b.getId(), b.getCustomerId(), new QueryProcessorResourceInfo(0, 100, filter().by("foo").build()));
assertThat(results.getPage()).isZero();
assertThat(results.getSize()).isEqualTo(2L);
assertThat(results.getTotal()).isEqualTo(2L);
// Newest instances come first
assertThat(results.getItems().get(1).getId()).isEqualTo(p1.getId());
assertThat(results.getItems().get(0).getId()).isEqualTo(p2.getId());
}
use of com.redhat.service.smartevents.manager.models.Processor in project sandbox by 5733d9e2be6485d52ffa08870cabdee0.
the class ProcessorDAOTest method testGetProcessorsFilterByType.
@Test
void testGetProcessorsFilterByType() {
Bridge b = createBridge();
Processor p = createProcessor(b, "foo", SOURCE);
createProcessor(b, "bar");
ListResult<Processor> results = processorDAO.findByBridgeIdAndCustomerId(b.getId(), b.getCustomerId(), new QueryProcessorResourceInfo(0, 100, filter().by(p.getType()).build()));
assertThat(results.getPage()).isZero();
assertThat(results.getSize()).isEqualTo(1L);
assertThat(results.getTotal()).isEqualTo(1L);
assertThat(results.getItems().get(0).getId()).isEqualTo(p.getId());
}
Aggregations