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.dataformat.tarfile; import java.io.ByteArrayInputStream; import java.io.ByteArrayOutputStream; import java.io.File; import java.io.FileInputStream; import java.io.IOException; import java.util.concurrent.TimeUnit; import org.apache.camel.Exchange; import org.apache.camel.builder.NotifyBuilder; import org.apache.camel.builder.RouteBuilder; import org.apache.camel.component.mock.MockEndpoint; import org.apache.camel.test.junit4.CamelTestSupport; import org.apache.camel.util.IOHelper; import org.apache.camel.util.ObjectHelper; import org.apache.commons.compress.archivers.tar.TarArchiveEntry; import org.apache.commons.compress.archivers.tar.TarArchiveOutputStream; import org.junit.Test; import static org.apache.camel.Exchange.FILE_NAME; /** * Unit tests for {@link TarFileDataFormat}. */ public class TarFileDataFormatTest extends CamelTestSupport { private static final String TEXT = "The Masque of Queen Bersabe (excerpt) \n" + "by: Algernon Charles Swinburne \n\n" + "My lips kissed dumb the word of Ah \n" + "Sighed on strange lips grown sick thereby. \n" + "God wrought to me my royal bed; \n" + "The inner work thereof was red, \n" + "The outer work was ivory. \n" + "My mouth's heat was the heat of flame \n" + "For lust towards the kings that came \n" + "With horsemen riding royally."; private static final File TEST_DIR = new File("target/tar"); @Test public void testTarWithoutFileName() throws Exception { MockEndpoint mock = getMockEndpoint("mock:tar"); mock.expectedMessageCount(1); template.sendBody("direct:tar", TEXT); assertMockEndpointsSatisfied(); Exchange exchange = mock.getReceivedExchanges().get(0); assertEquals(exchange.getIn().getMessageId() + ".tar", exchange.getIn().getHeader(FILE_NAME)); assertTrue(ObjectHelper.equalByteArray(getTaredText(exchange.getIn().getMessageId()), (byte[]) exchange.getIn().getBody())); } @Test public void testTarWithFileName() throws Exception { getMockEndpoint("mock:tar").expectedBodiesReceived(getTaredText("poem.txt")); getMockEndpoint("mock:tar").expectedHeaderReceived(FILE_NAME, "poem.txt.tar"); template.sendBodyAndHeader("direct:tar", TEXT, FILE_NAME, "poem.txt"); assertMockEndpointsSatisfied(); } @Test public void testUntar() throws Exception { getMockEndpoint("mock:untar").expectedBodiesReceived(TEXT); getMockEndpoint("mock:untar").expectedHeaderReceived(FILE_NAME, "file"); template.sendBody("direct:untar", getTaredText("file")); assertMockEndpointsSatisfied(); } @Test public void testTarAndUntar() throws Exception { MockEndpoint mock = getMockEndpoint("mock:tarAndUntar"); mock.expectedMessageCount(1); template.sendBody("direct:tarAndUntar", TEXT); assertMockEndpointsSatisfied(); Exchange exchange = mock.getReceivedExchanges().get(0); assertEquals(exchange.getIn().getMessageId(), exchange.getIn().getHeader(FILE_NAME)); assertEquals(TEXT, new String((byte[]) exchange.getIn().getBody(), "UTF-8")); } @Test public void testTarToFileWithoutFileName() throws Exception { NotifyBuilder notify = new NotifyBuilder(context).whenDone(1).create(); String[] files = TEST_DIR.list(); assertTrue(files == null || files.length == 0); MockEndpoint mock = getMockEndpoint("mock:intercepted"); mock.expectedMessageCount(1); template.sendBody("direct:tarToFile", TEXT); assertMockEndpointsSatisfied(); // use builder to ensure the exchange is fully done before we check for file exists assertTrue(notify.matches(5, TimeUnit.SECONDS)); Exchange exchange = mock.getReceivedExchanges().get(0); File file = new File(TEST_DIR, exchange.getIn().getMessageId() + ".tar"); assertTrue(file.exists()); assertTrue(ObjectHelper.equalByteArray(getTaredText(exchange.getIn().getMessageId()), getBytes(file))); } @Test public void testTarToFileWithFileName() throws Exception { NotifyBuilder notify = new NotifyBuilder(context).whenDone(1).create(); MockEndpoint mock = getMockEndpoint("mock:tarToFile"); mock.expectedMessageCount(1); File file = new File(TEST_DIR, "poem.txt.tar"); assertFalse(file.exists()); template.sendBodyAndHeader("direct:tarToFile", TEXT, FILE_NAME, "poem.txt"); // just make sure the file is created mock.assertIsSatisfied(); // use builder to ensure the exchange is fully done before we check for file exists assertTrue(notify.matches(5, TimeUnit.SECONDS)); assertTrue(file.exists()); assertTrue(ObjectHelper.equalByteArray(getTaredText("poem.txt"), getBytes(file))); } @Test public void testDslTar() throws Exception { getMockEndpoint("mock:dslTar").expectedBodiesReceived(getTaredText("poem.txt")); getMockEndpoint("mock:dslTar").expectedHeaderReceived(FILE_NAME, "poem.txt.tar"); template.sendBodyAndHeader("direct:dslTar", TEXT, FILE_NAME, "poem.txt"); assertMockEndpointsSatisfied(); } @Test public void testDslUntar() throws Exception { getMockEndpoint("mock:dslUntar").expectedBodiesReceived(TEXT); getMockEndpoint("mock:dslUntar").expectedHeaderReceived(FILE_NAME, "test.txt"); template.sendBody("direct:dslUntar", getTaredText("test.txt")); assertMockEndpointsSatisfied(); } @Override public void setUp() throws Exception { deleteDirectory(TEST_DIR); super.setUp(); } @Override protected RouteBuilder createRouteBuilder() throws Exception { return new RouteBuilder() { @Override public void configure() throws Exception { interceptSendToEndpoint("file:*").to("mock:intercepted"); TarFileDataFormat tar = new TarFileDataFormat(); from("direct:tar").marshal(tar).to("mock:tar"); from("direct:untar").unmarshal(tar).to("mock:untar"); from("direct:tarAndUntar").marshal(tar).unmarshal(tar).to("mock:tarAndUntar"); from("direct:tarToFile").marshal(tar).to("file:" + TEST_DIR.getPath()).to("mock:tarToFile"); from("direct:dslTar").marshal(tar).to("mock:dslTar"); from("direct:dslUntar").unmarshal(tar).to("mock:dslUntar"); } }; } private static byte[] getTaredText(String entryName) throws IOException { ByteArrayInputStream bais = new ByteArrayInputStream(TEXT.getBytes("UTF-8")); ByteArrayOutputStream baos = new ByteArrayOutputStream(); TarArchiveOutputStream tos = new TarArchiveOutputStream(baos); try { TarArchiveEntry entry = new TarArchiveEntry(entryName); entry.setSize(bais.available()); tos.putArchiveEntry(entry); IOHelper.copy(bais, tos); } finally { tos.closeArchiveEntry(); IOHelper.close(bais, tos); } return baos.toByteArray(); } private static byte[] getBytes(File file) throws IOException { FileInputStream fis = new FileInputStream(file); ByteArrayOutputStream baos = new ByteArrayOutputStream(); try { IOHelper.copy(fis, baos); } finally { IOHelper.close(fis, baos); } return baos.toByteArray(); } }