| /** |
| * 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. |
| */ |
| /** |
| * 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.dataformat.csv; |
| |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.regex.Pattern; |
| |
| import org.apache.camel.Exchange; |
| import org.apache.camel.Message; |
| import org.apache.camel.builder.RouteBuilder; |
| import org.apache.camel.component.mock.MockEndpoint; |
| import org.apache.camel.test.junit4.CamelTestSupport; |
| import org.apache.commons.csv.writer.CSVConfig; |
| import org.apache.commons.csv.writer.CSVField; |
| import org.junit.Test; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| /** |
| * @version |
| */ |
| public class CsvRouteTest extends CamelTestSupport { |
| private static final transient Logger LOG = LoggerFactory.getLogger(CsvRouteTest.class); |
| |
| @Test |
| public void testSendMessage() throws Exception { |
| MockEndpoint resultEndpoint = resolveMandatoryEndpoint("mock:result", MockEndpoint.class); |
| resultEndpoint.expectedMessageCount(1); |
| |
| // START SNIPPET: marshalInput |
| Map<String, Object> body = new HashMap<String, Object>(); |
| body.put("foo", "abc"); |
| body.put("bar", 123); |
| // END SNIPPET: marshalInput |
| template.sendBody("direct:start", body); |
| |
| resultEndpoint.assertIsSatisfied(); |
| List<Exchange> list = resultEndpoint.getReceivedExchanges(); |
| for (Exchange exchange : list) { |
| Message in = exchange.getIn(); |
| String text = in.getBody(String.class); |
| |
| log.debug("Received " + text); |
| assertNotNull("Should be able to convert received body to a string", text); |
| |
| // order is not guaranteed with a Map (which was passed in before) |
| // so we need to check for both combinations |
| assertTrue("Text body has wrong value.", "abc,123".equals(text.trim()) |
| || "123,abc".equals(text.trim())); |
| } |
| } |
| |
| @Test |
| public void testMultipleMessages() throws Exception { |
| MockEndpoint resultEndpoint = resolveMandatoryEndpoint("mock:resultMulti", |
| MockEndpoint.class); |
| resultEndpoint.expectedMessageCount(2); |
| Map<String, Object> body1 = new HashMap<String, Object>(); |
| body1.put("foo", "abc"); |
| body1.put("bar", 123); |
| |
| Map<String, Object> body2 = new HashMap<String, Object>(); |
| body2.put("foo", "def"); |
| body2.put("bar", 456); |
| body2.put("baz", 789); |
| |
| template.sendBody("direct:startMulti", body1); |
| template.sendBody("direct:startMulti", body2); |
| |
| resultEndpoint.assertIsSatisfied(); |
| List<Exchange> list = resultEndpoint.getReceivedExchanges(); |
| Message in1 = list.get(0).getIn(); |
| String text1 = in1.getBody(String.class); |
| |
| log.debug("Received " + text1); |
| assertTrue("First CSV body has wrong value", |
| Pattern.matches("(abc,123)|(123,abc)", text1.trim())); |
| |
| Message in2 = list.get(1).getIn(); |
| String text2 = in2.getBody(String.class); |
| |
| log.debug("Received " + text2); |
| |
| // fields should keep the same order from one call to the other |
| if (text1.trim().equals("abc,123")) { |
| assertEquals("Second CSV body has wrong value", |
| "def,456,789", text2.trim()); |
| } else { |
| assertEquals("Second CSV body has wrong value", |
| "456,def,789", text2.trim()); |
| } |
| } |
| |
| @Test |
| public void testPresetConfig() throws Exception { |
| MockEndpoint resultEndpoint = resolveMandatoryEndpoint("mock:resultMultiCustom", |
| MockEndpoint.class); |
| resultEndpoint.expectedMessageCount(2); |
| Map<String, Object> body1 = new HashMap<String, Object>(); |
| body1.put("foo", "abc"); |
| body1.put("bar", 123); |
| |
| Map<String, Object> body2 = new HashMap<String, Object>(); |
| body2.put("foo", "def"); |
| body2.put("bar", 456); |
| body2.put("baz", 789); |
| body2.put("buz", "000"); |
| |
| template.sendBody("direct:startMultiCustom", body1); |
| template.sendBody("direct:startMultiCustom", body2); |
| |
| List<Exchange> list = resultEndpoint.getReceivedExchanges(); |
| Message in1 = list.get(0).getIn(); |
| String text1 = in1.getBody(String.class); |
| |
| log.debug("Received " + text1); |
| assertEquals("First CSV body has wrong value", |
| "abc;;123", text1.trim()); |
| |
| Message in2 = list.get(1).getIn(); |
| String text2 = in2.getBody(String.class); |
| |
| log.debug("Received " + text2); |
| assertEquals("Second CSV body has wrong value", |
| "def;789;456", text2.trim()); |
| |
| } |
| |
| @SuppressWarnings("unchecked") |
| @Test |
| public void testUnMarshal() throws Exception { |
| MockEndpoint endpoint = getMockEndpoint("mock:daltons"); |
| endpoint.expectedMessageCount(1); |
| endpoint.assertIsSatisfied(); |
| Exchange exchange = endpoint.getExchanges().get(0); |
| // START SNIPPET : unmarshalResult |
| List<List<String>> data = (List<List<String>>) exchange.getIn().getBody(); |
| for (List<String> line : data) { |
| LOG.debug(String.format("%s has an IQ of %s and is currently %s", |
| line.get(0), line.get(1), line.get(2))); |
| } |
| // END SNIPPET : unmarshalResult |
| } |
| |
| protected RouteBuilder createRouteBuilder() { |
| return new RouteBuilder() { |
| public void configure() { |
| // START SNIPPET: marshalRoute |
| from("direct:start"). |
| marshal().csv(). |
| to("mock:result"); |
| // END SNIPPET: marshalRoute |
| |
| from("direct:startMulti"). |
| marshal().csv(). |
| to("mock:resultMulti"); |
| |
| CsvDataFormat customCsv = new CsvDataFormat(); |
| CSVConfig custom = new CSVConfig(); |
| custom.setDelimiter(';'); |
| custom.addField(new CSVField("foo")); |
| custom.addField(new CSVField("baz")); |
| custom.addField(new CSVField("bar")); |
| customCsv.setConfig(custom); |
| customCsv.setAutogenColumns(false); |
| |
| from("direct:startMultiCustom"). |
| marshal(customCsv). |
| to("mock:resultMultiCustom"); |
| |
| // START SNIPPET: unmarshalRoute |
| from("file:src/test/resources/?fileName=daltons.csv&noop=true"). |
| unmarshal().csv(). |
| to("mock:daltons"); |
| // END SNIPPET: unmarshalRoute |
| } |
| }; |
| } |
| } |