protected CountDownLatch latch = new CountDownLatch(1);
protected String entityName = MultiSteps.class.getName();
protected String queryText = "select o from " + entityName + " o where o.step = 1";
public void testProducerInsertsIntoDatabaseThenConsumerFiresMessageExchange() throws Exception {
transactionStrategy.execute(new JpaCallback() {
public Object doInJpa(EntityManager entityManager) throws PersistenceException {
// lets delete any exiting records before the test
entityManager.createQuery("delete from " + entityName).executeUpdate();
// now lets create a dummy entry
MultiSteps dummy = new MultiSteps("cheese");
dummy.setStep(4);
entityManager.persist(dummy);
return null;
}
});
List results = jpaTemplate.find(queryText);
assertEquals("Should have no results: " + results, 0, results.size());
// lets produce some objects
template.send(endpoint, new Processor() {
public void process(Exchange exchange) {
exchange.getIn().setBody(new MultiSteps("foo@bar.com"));
}
});
// now lets assert that there is a result
results = jpaTemplate.find(queryText);
assertEquals("Should have results: " + results, 1, results.size());
MultiSteps mail = (MultiSteps)results.get(0);
assertEquals("address property", "foo@bar.com", mail.getAddress());
// now lets create a consumer to consume it
consumer = endpoint.createConsumer(new Processor() {
public void process(Exchange e) {
LOG.info("Received exchange: " + e.getIn());
receivedExchange = e;
latch.countDown();
}
});
consumer.start();
boolean received = latch.await(50, TimeUnit.SECONDS);
assertTrue("Did not receive the message!", received);
assertNotNull(receivedExchange);
MultiSteps result = receivedExchange.getIn().getBody(MultiSteps.class);
assertNotNull("Received a POJO", result);
assertEquals("address property", "foo@bar.com", result.getAddress());
// lets now test that the database is updated
// TODO we need to sleep as we will be invoked from inside the
// transaction!
Thread.sleep(1000);
transactionStrategy.execute(new JpaCallback() {
public Object doInJpa(EntityManager entityManager) throws PersistenceException {
// now lets assert that there are still 2 entities left
List<MultiSteps> rows = entityManager.createQuery("select x from MultiSteps x").getResultList();
assertEquals("Number of entities: " + rows, 2, rows.size());