Java tutorial
/** * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You under the Apache License, Version 2.0 * (the "License"); you may not use this file except in compliance with * the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.apache.camel.component.jpa; import java.util.List; import java.util.concurrent.CountDownLatch; import java.util.concurrent.TimeUnit; import javax.persistence.EntityManager; import javax.persistence.PersistenceException; import org.apache.camel.CamelContext; import org.apache.camel.Consumer; import org.apache.camel.Endpoint; import org.apache.camel.Exchange; import org.apache.camel.Processor; import org.apache.camel.ProducerTemplate; import org.apache.camel.examples.MultiSteps; import org.apache.camel.impl.DefaultCamelContext; import org.apache.camel.util.ServiceHelper; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.junit.After; import org.junit.Assert; import org.junit.Before; import org.junit.Test; import org.springframework.orm.jpa.JpaCallback; import org.springframework.orm.jpa.JpaTemplate; /** * @version $Revision$ */ public class JpaWithNamedQueryTest extends Assert { protected static final transient Log LOG = LogFactory.getLog(JpaWithNamedQueryTest.class); protected CamelContext camelContext = new DefaultCamelContext(); protected ProducerTemplate template; protected JpaEndpoint endpoint; protected TransactionStrategy transactionStrategy; protected JpaTemplate jpaTemplate; protected Consumer consumer; protected Exchange receivedExchange; protected CountDownLatch latch = new CountDownLatch(1); protected String entityName = MultiSteps.class.getName(); protected String queryText = "select o from " + entityName + " o where o.step = 1"; @Test 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); assertReceivedResult(receivedExchange); // lets now test that the database is updated // we need to sleep as we will be invoked from inside the transaction! Thread.sleep(1000); transactionStrategy.execute(new JpaCallback() { @SuppressWarnings("unchecked") 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()); int counter = 1; for (MultiSteps row : rows) { LOG.info("entity: " + counter++ + " = " + row); if (row.getAddress().equals("foo@bar.com")) { LOG.info("Found updated row: " + row); assertEquals("Updated row step for: " + row, getUpdatedStepValue(), row.getStep()); } else { // dummy row assertEquals("dummy row step for: " + row, 4, row.getStep()); } } return null; } }); JpaConsumer jpaConsumer = (JpaConsumer) consumer; assertURIQueryOption(jpaConsumer); } protected void assertReceivedResult(Exchange exchange) { assertNotNull(exchange); MultiSteps result = exchange.getIn().getBody(MultiSteps.class); assertNotNull("Received a POJO", result); assertEquals("address property", "foo@bar.com", result.getAddress()); } protected int getUpdatedStepValue() { return 2; } protected void assertURIQueryOption(JpaConsumer jpaConsumer) { assertEquals("step1", jpaConsumer.getNamedQuery()); } @Before public void setUp() throws Exception { template = camelContext.createProducerTemplate(); ServiceHelper.startServices(template, camelContext); Endpoint value = camelContext.getEndpoint(getEndpointUri()); assertNotNull("Could not find endpoint!", value); assertTrue("Should be a JPA endpoint but was: " + value, value instanceof JpaEndpoint); endpoint = (JpaEndpoint) value; transactionStrategy = endpoint.createTransactionStrategy(); jpaTemplate = endpoint.getTemplate(); } protected String getEndpointUri() { return "jpa://" + MultiSteps.class.getName() + "?consumer.namedQuery=step1"; } @After public void tearDown() throws Exception { ServiceHelper.stopServices(consumer, template, camelContext); } }