Author: cmueller
Date: Thu Aug 5 12:34:25 2010
New Revision: 982577
URL: http://svn.apache.org/viewvc?rev=982577&view=rev
Log:
CAMEL-2980: camel-jpa doesn't use EntityManager.merge(entity) in the right way
Added:
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/AbstractJpaMethodTest.java
(with props)
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUseMergeTest.java
(with props)
Modified:
camel/trunk/components/camel-jpa/src/main/java/org/apache/camel/component/jpa/JpaProducer.java
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUsePersistTest.java
Modified:
camel/trunk/components/camel-jpa/src/main/java/org/apache/camel/component/jpa/JpaProducer.java
URL:
http://svn.apache.org/viewvc/camel/trunk/components/camel-jpa/src/main/java/org/apache/camel/component/jpa/JpaProducer.java?rev=982577&r1=982576&r2=982577&view=diff
==============================================================================
---
camel/trunk/components/camel-jpa/src/main/java/org/apache/camel/component/jpa/JpaProducer.java
(original)
+++
camel/trunk/components/camel-jpa/src/main/java/org/apache/camel/component/jpa/JpaProducer.java
Thu Aug 5 12:34:25 2010
@@ -16,7 +16,10 @@
*/
package org.apache.camel.component.jpa;
+import java.util.ArrayList;
+import java.util.Collection;
import java.util.Iterator;
+import java.util.List;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceException;
@@ -24,7 +27,6 @@ import javax.persistence.PersistenceExce
import org.apache.camel.Exchange;
import org.apache.camel.Expression;
import org.apache.camel.impl.DefaultProducer;
-import org.apache.camel.util.ObjectHelper;
import org.springframework.orm.jpa.JpaCallback;
/**
@@ -42,26 +44,60 @@ public class JpaProducer extends Default
this.template = endpoint.createTransactionStrategy();
}
- public void process(Exchange exchange) {
+ public void process(final Exchange exchange) {
exchange.getIn().setHeader(JpaConstants.JPA_TEMPLATE,
endpoint.getTemplate());
final Object values = expression.evaluate(exchange, Object.class);
if (values != null) {
template.execute(new JpaCallback() {
+ @SuppressWarnings("unchecked")
public Object doInJpa(EntityManager entityManager) throws
PersistenceException {
- Iterator iter = ObjectHelper.createIterator(values);
- while (iter.hasNext()) {
- Object value = iter.next();
- if (endpoint.isUsePersist()) {
- entityManager.persist(value);
- } else {
- entityManager.merge(value);
+ if (values.getClass().isArray()) {
+ Object[] array = (Object[]) values;
+ for (int index = 0; index < array.length; index++) {
+ Object managedEntity = save(array[index],
entityManager);
+ if (!endpoint.isUsePersist()) {
+ array[index] = managedEntity;
+ }
+ }
+ } else if (values instanceof Collection) {
+ Collection collection = (Collection) values;
+ List managedEntities = new ArrayList();
+ for (Iterator iter = collection.iterator();
iter.hasNext();) {
+ Object managedEntity = save(iter.next(),
entityManager);
+ managedEntities.add(managedEntity);
+ }
+ if (!endpoint.isUsePersist()) {
+ collection.clear();
+ collection.addAll(managedEntities);
+ }
+ } else {
+ Object managedEntity = save(values, entityManager);
+ if (!endpoint.isUsePersist()) {
+ exchange.getIn().setBody(managedEntity);
}
}
+
if (endpoint.isFlushOnSend()) {
entityManager.flush();
}
return null;
}
+
+ /**
+ * save the given entity end return the managed entity
+ *
+ * @param entity
+ * @param entityManager
+ * @return the managed entity
+ */
+ private Object save(final Object entity, EntityManager
entityManager) {
+ if (endpoint.isUsePersist()) {
+ entityManager.persist(entity);
+ return entity;
+ } else {
+ return entityManager.merge(entity);
+ }
+ }
});
}
exchange.getIn().removeHeader(JpaConstants.JPA_TEMPLATE);
Added:
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/AbstractJpaMethodTest.java
URL:
http://svn.apache.org/viewvc/camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/AbstractJpaMethodTest.java?rev=982577&view=auto
==============================================================================
---
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/AbstractJpaMethodTest.java
(added)
+++
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/AbstractJpaMethodTest.java
Thu Aug 5 12:34:25 2010
@@ -0,0 +1,213 @@
+/**
+ * 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.ArrayList;
+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.Exchange;
+import org.apache.camel.Processor;
+import org.apache.camel.ProducerTemplate;
+import org.apache.camel.examples.Address;
+import org.apache.camel.examples.Customer;
+import org.apache.camel.impl.DefaultCamelContext;
+import org.apache.camel.impl.DefaultExchange;
+import org.junit.After;
+import org.junit.Assert;
+import org.junit.Test;
+import org.springframework.orm.jpa.JpaCallback;
+import org.springframework.orm.jpa.JpaTemplate;
+
+import static org.apache.camel.util.ServiceHelper.startServices;
+import static org.apache.camel.util.ServiceHelper.stopServices;
+
+/**
+ * @version $Revision: $
+ */
+public abstract class AbstractJpaMethodTest extends Assert {
+
+ protected CamelContext camelContext = new DefaultCamelContext();
+ protected ProducerTemplate template;
+ protected JpaEndpoint endpoint;
+ protected TransactionStrategy transactionStrategy;
+ protected JpaTemplate jpaTemplate;
+ protected Consumer consumer;
+ protected Exchange receivedExchange;
+
+ abstract boolean usePersist();
+
+ @After
+ public void tearDown() throws Exception {
+ stopServices(consumer, template, camelContext);
+ }
+
+ @SuppressWarnings("unchecked")
+ @Test
+ public void produceNewEntity() throws Exception {
+ setUp("jpa://" + Customer.class.getName() + "?usePersist=" +
(usePersist() ? "true" : "false"));
+
+ Customer customer = createDefaultCustomer();
+ Exchange exchange = new DefaultExchange(camelContext);
+ exchange.getIn().setBody(customer);
+ Exchange returnedExchange = template.send(endpoint, exchange);
+
+ Customer receivedCustomer =
returnedExchange.getIn().getBody(Customer.class);
+ assertEquals(customer.getName(), receivedCustomer.getName());
+ assertNotNull(receivedCustomer.getId());
+ assertEquals(customer.getAddress().getAddressLine1(),
receivedCustomer.getAddress().getAddressLine1());
+ assertEquals(customer.getAddress().getAddressLine2(),
receivedCustomer.getAddress().getAddressLine2());
+ assertNotNull(receivedCustomer.getAddress().getId());
+
+ List results = jpaTemplate.find("select o from " +
Customer.class.getName() + " o");
+ assertEquals(1, results.size());
+ Customer persistedCustomer = (Customer) results.get(0);
+ assertEquals(receivedCustomer.getName(), persistedCustomer.getName());
+ assertEquals(receivedCustomer.getId(), persistedCustomer.getId());
+ assertEquals(receivedCustomer.getAddress().getAddressLine1(),
persistedCustomer.getAddress().getAddressLine1());
+ assertEquals(receivedCustomer.getAddress().getAddressLine2(),
persistedCustomer.getAddress().getAddressLine2());
+ assertEquals(receivedCustomer.getAddress().getId(),
persistedCustomer.getAddress().getId());
+ }
+
+ @SuppressWarnings("unchecked")
+ @Test
+ public void produceNewEntitiesFromList() throws Exception {
+ setUp("jpa://" + List.class.getName() + "?usePersist=" + (usePersist()
? "true" : "false"));
+
+ List<Customer> customers = new ArrayList<Customer>();
+ customers.add(createDefaultCustomer());
+ customers.add(createDefaultCustomer());
+ Exchange exchange = new DefaultExchange(camelContext);
+ exchange.getIn().setBody(customers);
+ Exchange returnedExchange = template.send(endpoint, exchange);
+
+ List returnedCustomers = returnedExchange.getIn().getBody(List.class);
+ assertEquals(2, returnedCustomers.size());
+
+ assertEntitiesInDatabase(2, Customer.class.getName());
+ assertEntitiesInDatabase(2, Address.class.getName());
+ }
+
+ @Test
+ public void produceNewEntitiesFromArray() throws Exception {
+ setUp("jpa://" + Customer[].class.getName() + "?usePersist=" +
(usePersist() ? "true" : "false"));
+
+ Customer[] customers = new Customer[] {createDefaultCustomer(),
createDefaultCustomer()};
+ Exchange exchange = new DefaultExchange(camelContext);
+ exchange.getIn().setBody(customers);
+ Exchange returnedExchange = template.send(endpoint, exchange);
+
+ Customer[] returnedCustomers =
returnedExchange.getIn().getBody(Customer[].class);
+ assertEquals(2, returnedCustomers.length);
+
+ assertEntitiesInDatabase(2, Customer.class.getName());
+ assertEntitiesInDatabase(2, Address.class.getName());
+ }
+
+ @Test
+ public void consumeEntity() throws Exception {
+ setUp("jpa://" + Customer.class.getName() + "?usePersist=" +
(usePersist() ? "true" : "false"));
+
+ final Customer customer = createDefaultCustomer();
+ save(customer);
+
+ final CountDownLatch latch = new CountDownLatch(1);
+
+ consumer = endpoint.createConsumer(new Processor() {
+ public void process(Exchange e) {
+ receivedExchange = e;
+ assertNotNull(e.getIn().getHeader(JpaConstants.JPA_TEMPLATE,
JpaTemplate.class));
+ latch.countDown();
+ }
+ });
+ consumer.start();
+
+ boolean received = latch.await(50, TimeUnit.SECONDS);
+
+ assertTrue(received);
+ assertNotNull(receivedExchange);
+ Customer receivedCustomer =
receivedExchange.getIn().getBody(Customer.class);
+ assertEquals(customer.getName(), receivedCustomer.getName());
+ assertEquals(customer.getId(), receivedCustomer.getId());
+ assertEquals(customer.getAddress().getAddressLine1(),
receivedCustomer.getAddress().getAddressLine1());
+ assertEquals(customer.getAddress().getAddressLine2(),
receivedCustomer.getAddress().getAddressLine2());
+ assertEquals(customer.getAddress().getId(),
receivedCustomer.getAddress().getId());
+
+ // give a bit tiem for consumer to delete after done
+ Thread.sleep(1000);
+
+ assertEntitiesInDatabase(0, Customer.class.getName());
+ assertEntitiesInDatabase(0, Address.class.getName());
+ }
+
+ @SuppressWarnings("unchecked")
+ protected void setUp(String endpointUri) throws Exception {
+ template = camelContext.createProducerTemplate();
+ startServices(template, camelContext);
+
+ endpoint = camelContext.getEndpoint(endpointUri, JpaEndpoint.class);
+
+ transactionStrategy = endpoint.createTransactionStrategy();
+ jpaTemplate = endpoint.getTemplate();
+
+ transactionStrategy.execute(new JpaCallback() {
+ public Object doInJpa(EntityManager entityManager) throws
PersistenceException {
+ entityManager.createQuery("delete from " +
Customer.class.getName()).executeUpdate();
+ return null;
+ }
+ });
+
+ assertEntitiesInDatabase(0, Customer.class.getName());
+ assertEntitiesInDatabase(0, Address.class.getName());
+ }
+
+ @SuppressWarnings("unchecked")
+ protected void save(final Customer customer) {
+ transactionStrategy.execute(new JpaCallback() {
+ public Object doInJpa(EntityManager entityManager) throws
PersistenceException {
+ entityManager.persist(customer);
+ entityManager.flush();
+ return null;
+ }
+ });
+
+ assertEntitiesInDatabase(1, Customer.class.getName());
+ assertEntitiesInDatabase(1, Address.class.getName());
+ }
+
+ @SuppressWarnings("unchecked")
+ protected void assertEntitiesInDatabase(int count, String entity) {
+ List results = jpaTemplate.find("select o from " + entity + " o");
+ assertEquals(count, results.size());
+ }
+
+ protected Customer createDefaultCustomer() {
+ Customer customer = new Customer();
+ customer.setName("Christian Mueller");
+ Address address = new Address();
+ address.setAddressLine1("Hahnstr. 1");
+ address.setAddressLine2("60313 Frankfurt am Main");
+ customer.setAddress(address);
+ return customer;
+ }
+}
\ No newline at end of file
Propchange:
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/AbstractJpaMethodTest.java
------------------------------------------------------------------------------
svn:eol-style = native
Propchange:
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/AbstractJpaMethodTest.java
------------------------------------------------------------------------------
svn:mime-type = text/plain
Added:
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUseMergeTest.java
URL:
http://svn.apache.org/viewvc/camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUseMergeTest.java?rev=982577&view=auto
==============================================================================
---
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUseMergeTest.java
(added)
+++
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUseMergeTest.java
Thu Aug 5 12:34:25 2010
@@ -0,0 +1,81 @@
+/**
+ * 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 javax.persistence.EntityManager;
+import javax.persistence.PersistenceException;
+
+import org.apache.camel.Exchange;
+import org.apache.camel.examples.Address;
+import org.apache.camel.examples.Customer;
+import org.apache.camel.impl.DefaultExchange;
+import org.junit.Test;
+import org.springframework.orm.jpa.JpaCallback;
+
+/**
+ * @version $Revision: $
+ */
+public class JpaUseMergeTest extends AbstractJpaMethodTest {
+
+ public boolean usePersist() {
+ return false;
+ }
+
+ @SuppressWarnings("unchecked")
+ @Test
+ public void produceExistingEntity() throws Exception {
+ setUp("jpa://" + Customer.class.getName() + "?usePersist=false");
+
+ final Customer customer = createDefaultCustomer();
+ transactionStrategy.execute(new JpaCallback() {
+ public Object doInJpa(EntityManager entityManager) throws
PersistenceException {
+ entityManager.persist(customer);
+ entityManager.flush();
+ return null;
+ }
+ });
+
+ assertEntitiesInDatabase(1, Customer.class.getName());
+ assertEntitiesInDatabase(1, Address.class.getName());
+
+ customer.setName("Max Mustermann");
+ customer.getAddress().setAddressLine1("Musterstr. 1");
+ customer.getAddress().setAddressLine2("11111 Enterhausen");
+
+ Exchange exchange = new DefaultExchange(camelContext);
+ exchange.getIn().setBody(customer);
+ Exchange returnedExchange = template.send(endpoint, exchange);
+
+ Customer receivedCustomer =
returnedExchange.getIn().getBody(Customer.class);
+ assertEquals(customer.getName(), receivedCustomer.getName());
+ assertNotNull(receivedCustomer.getId());
+ assertEquals(customer.getAddress().getAddressLine1(),
receivedCustomer.getAddress().getAddressLine1());
+ assertEquals(customer.getAddress().getAddressLine2(),
receivedCustomer.getAddress().getAddressLine2());
+ assertNotNull(receivedCustomer.getAddress().getId());
+
+ List results = jpaTemplate.find("select o from " +
Customer.class.getName() + " o");
+ assertEquals(1, results.size());
+ Customer persistedCustomer = (Customer) results.get(0);
+ assertEquals(receivedCustomer.getName(), persistedCustomer.getName());
+ assertEquals(receivedCustomer.getId(), persistedCustomer.getId());
+ assertEquals(receivedCustomer.getAddress().getAddressLine1(),
persistedCustomer.getAddress().getAddressLine1());
+ assertEquals(receivedCustomer.getAddress().getAddressLine2(),
persistedCustomer.getAddress().getAddressLine2());
+ assertEquals(receivedCustomer.getAddress().getId(),
persistedCustomer.getAddress().getId());
+ }
+}
\ No newline at end of file
Propchange:
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUseMergeTest.java
------------------------------------------------------------------------------
svn:eol-style = native
Propchange:
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUseMergeTest.java
------------------------------------------------------------------------------
svn:mime-type = text/plain
Modified:
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUsePersistTest.java
URL:
http://svn.apache.org/viewvc/camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUsePersistTest.java?rev=982577&r1=982576&r2=982577&view=diff
==============================================================================
---
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUsePersistTest.java
(original)
+++
camel/trunk/components/camel-jpa/src/test/java/org/apache/camel/component/jpa/JpaUsePersistTest.java
Thu Aug 5 12:34:25 2010
@@ -16,110 +16,19 @@
*/
package org.apache.camel.component.jpa;
-import java.util.ArrayList;
-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.Exchange;
-import org.apache.camel.Processor;
-import org.apache.camel.ProducerTemplate;
import org.apache.camel.examples.Address;
import org.apache.camel.examples.Customer;
-import org.apache.camel.impl.DefaultCamelContext;
import org.apache.camel.impl.DefaultExchange;
-import org.junit.After;
-import org.junit.Assert;
import org.junit.Test;
-import org.springframework.orm.jpa.JpaCallback;
-import org.springframework.orm.jpa.JpaTemplate;
-
-import static org.apache.camel.util.ServiceHelper.startServices;
-import static org.apache.camel.util.ServiceHelper.stopServices;
/**
* @version $Revision: 931444 $
*/
-public class JpaUsePersistTest extends Assert {
-
- protected CamelContext camelContext = new DefaultCamelContext();
- protected ProducerTemplate template;
- protected JpaEndpoint endpoint;
- protected TransactionStrategy transactionStrategy;
- protected JpaTemplate jpaTemplate;
- protected Consumer consumer;
- protected Exchange receivedExchange;
+public class JpaUsePersistTest extends AbstractJpaMethodTest {
- @After
- public void tearDown() throws Exception {
- stopServices(consumer, template, camelContext);
- }
-
- @SuppressWarnings("unchecked")
- @Test
- public void produceNewEntity() throws Exception {
- setUp("jpa://" + Customer.class.getName() + "?usePersist=true");
-
- Customer customer = createDefaultCustomer();
- Exchange exchange = new DefaultExchange(camelContext);
- exchange.getIn().setBody(customer);
- Exchange returnedExchange = template.send(endpoint, exchange);
-
- Customer receivedCustomer =
returnedExchange.getIn().getBody(Customer.class);
- assertEquals(customer.getName(), receivedCustomer.getName());
- assertNotNull(receivedCustomer.getId());
- assertEquals(customer.getAddress().getAddressLine1(),
receivedCustomer.getAddress().getAddressLine1());
- assertEquals(customer.getAddress().getAddressLine2(),
receivedCustomer.getAddress().getAddressLine2());
- assertNotNull(receivedCustomer.getAddress().getId());
-
- List results = jpaTemplate.find("select o from " +
Customer.class.getName() + " o");
- assertEquals(1, results.size());
- Customer persistedCustomer = (Customer) results.get(0);
- assertEquals(receivedCustomer.getName(), persistedCustomer.getName());
- assertEquals(receivedCustomer.getId(), persistedCustomer.getId());
- assertEquals(receivedCustomer.getAddress().getAddressLine1(),
persistedCustomer.getAddress().getAddressLine1());
- assertEquals(receivedCustomer.getAddress().getAddressLine2(),
persistedCustomer.getAddress().getAddressLine2());
- assertEquals(receivedCustomer.getAddress().getId(),
persistedCustomer.getAddress().getId());
- }
-
- @SuppressWarnings("unchecked")
- @Test
- public void produceNewEntitiesFromList() throws Exception {
- setUp("jpa://" + List.class.getName() + "?usePersist=true");
-
- List<Customer> customers = new ArrayList<Customer>();
- customers.add(createDefaultCustomer());
- customers.add(createDefaultCustomer());
- Exchange exchange = new DefaultExchange(camelContext);
- exchange.getIn().setBody(customers);
- Exchange returnedExchange = template.send(endpoint, exchange);
-
- List returnedCustomers = returnedExchange.getIn().getBody(List.class);
- assertEquals(2, returnedCustomers.size());
-
- assertEntitiesInDatabase(2, Customer.class.getName());
- assertEntitiesInDatabase(2, Address.class.getName());
- }
-
- @Test
- public void produceNewEntitiesFromArray() throws Exception {
- setUp("jpa://" + Customer[].class.getName() + "?usePersist=true");
-
- Customer[] customers = new Customer[] {createDefaultCustomer(),
createDefaultCustomer()};
- Exchange exchange = new DefaultExchange(camelContext);
- exchange.getIn().setBody(customers);
- Exchange returnedExchange = template.send(endpoint, exchange);
-
- Customer[] returnedCustomers =
returnedExchange.getIn().getBody(Customer[].class);
- assertEquals(2, returnedCustomers.length);
-
- assertEntitiesInDatabase(2, Customer.class.getName());
- assertEntitiesInDatabase(2, Address.class.getName());
+ public boolean usePersist() {
+ return true;
}
@Test
@@ -142,91 +51,4 @@ public class JpaUsePersistTest extends A
assertEntitiesInDatabase(1, Customer.class.getName());
assertEntitiesInDatabase(1, Address.class.getName());
}
-
- @Test
- public void consumeEntity() throws Exception {
- setUp("jpa://" + Customer.class.getName() + "?usePersist=true");
-
- final Customer customer = createDefaultCustomer();
- save(customer);
-
- final CountDownLatch latch = new CountDownLatch(1);
-
- consumer = endpoint.createConsumer(new Processor() {
- public void process(Exchange e) {
- receivedExchange = e;
- assertNotNull(e.getIn().getHeader(JpaConstants.JPA_TEMPLATE,
JpaTemplate.class));
- latch.countDown();
- }
- });
- consumer.start();
-
- boolean received = latch.await(50, TimeUnit.SECONDS);
-
- assertTrue(received);
- assertNotNull(receivedExchange);
- Customer receivedCustomer =
receivedExchange.getIn().getBody(Customer.class);
- assertEquals(customer.getName(), receivedCustomer.getName());
- assertEquals(customer.getId(), receivedCustomer.getId());
- assertEquals(customer.getAddress().getAddressLine1(),
receivedCustomer.getAddress().getAddressLine1());
- assertEquals(customer.getAddress().getAddressLine2(),
receivedCustomer.getAddress().getAddressLine2());
- assertEquals(customer.getAddress().getId(),
receivedCustomer.getAddress().getId());
-
- // give a bit tiem for consumer to delete after done
- Thread.sleep(1000);
-
- assertEntitiesInDatabase(0, Customer.class.getName());
- assertEntitiesInDatabase(0, Address.class.getName());
- }
-
- @SuppressWarnings("unchecked")
- private void setUp(String endpointUri) throws Exception {
- template = camelContext.createProducerTemplate();
- startServices(template, camelContext);
-
- endpoint = camelContext.getEndpoint(endpointUri, JpaEndpoint.class);
-
- transactionStrategy = endpoint.createTransactionStrategy();
- jpaTemplate = endpoint.getTemplate();
-
- transactionStrategy.execute(new JpaCallback() {
- public Object doInJpa(EntityManager entityManager) throws
PersistenceException {
- entityManager.createQuery("delete from " +
Customer.class.getName()).executeUpdate();
- return null;
- }
- });
-
- assertEntitiesInDatabase(0, Customer.class.getName());
- assertEntitiesInDatabase(0, Address.class.getName());
- }
-
- @SuppressWarnings("unchecked")
- private void save(final Customer customer) {
- transactionStrategy.execute(new JpaCallback() {
- public Object doInJpa(EntityManager entityManager) throws
PersistenceException {
- entityManager.persist(customer);
- entityManager.flush();
- return null;
- }
- });
-
- assertEntitiesInDatabase(1, Customer.class.getName());
- assertEntitiesInDatabase(1, Address.class.getName());
- }
-
- @SuppressWarnings("unchecked")
- private void assertEntitiesInDatabase(int count, String entity) {
- List results = jpaTemplate.find("select o from " + entity + " o");
- assertEquals(count, results.size());
- }
-
- private Customer createDefaultCustomer() {
- Customer customer = new Customer();
- customer.setName("Christian Mueller");
- Address address = new Address();
- address.setAddressLine1("Hahnstr. 1");
- address.setAddressLine2("60313 Frankfurt am Main");
- customer.setAddress(address);
- return customer;
- }
}
\ No newline at end of file
