/*
* 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.tika.server;
import static java.nio.charset.StandardCharsets.UTF_8;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardCopyOption;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.compress.archivers.ArchiveEntry;
import org.apache.commons.compress.archivers.ArchiveInputStream;
import org.apache.commons.compress.archivers.zip.ZipArchiveEntry;
import org.apache.commons.compress.archivers.zip.ZipFile;
import org.apache.commons.io.IOUtils;
import org.apache.cxf.binding.BindingFactoryManager;
import org.apache.cxf.endpoint.Server;
import org.apache.cxf.jaxrs.JAXRSBindingFactory;
import org.apache.cxf.jaxrs.JAXRSServerFactoryBean;
import org.apache.tika.config.TikaConfig;
import org.apache.tika.parser.utils.CommonsDigester;
import org.apache.tika.server.resource.TikaResource;
import org.apache.tika.server.resource.UnpackerResource;
import org.junit.After;
import org.junit.Before;
public abstract class CXFTestBase {
private final static int DIGESTER_READ_LIMIT = 20*1024*1024;
protected static final String endPoint =
"http://localhost:" + TikaServerCli.DEFAULT_PORT;
protected Server server;
private TikaConfig tika;
public static void assertContains(String needle, String haystack) {
assertTrue(needle + " not found in:\n" + haystack, haystack.contains(needle));
}
public static void assertNotFound(String needle, String haystack) {
assertFalse(needle + " unexpectedly found in:\n" + haystack, haystack.contains(needle));
}
protected static InputStream copy(InputStream in, int remaining) throws IOException {
ByteArrayOutputStream out = new ByteArrayOutputStream();
while (remaining > 0) {
byte[] bytes = new byte[remaining];
int n = in.read(bytes);
if (n <= 0) {
break;
}
out.write(bytes, 0, n);
remaining -= n;
}
return new ByteArrayInputStream(out.toByteArray());
}
@Before
public void setUp() {
this.tika = TikaConfig.getDefaultConfig();
TikaResource.init(tika,
new CommonsDigester(DIGESTER_READ_LIMIT, CommonsDigester.DigestAlgorithm.MD5),
new DefaultInputStreamFactory());
JAXRSServerFactoryBean sf = new JAXRSServerFactoryBean();
setUpResources(sf);
setUpProviders(sf);
sf.setAddress(endPoint + "/");
BindingFactoryManager manager = sf.getBus().getExtension(
BindingFactoryManager.class
);
JAXRSBindingFactory factory = new JAXRSBindingFactory();
factory.setBus(sf.getBus());
manager.registerBindingFactory(
JAXRSBindingFactory.JAXRS_BINDING_ID,
factory
);
server = sf.create();
}
/**
* Have the test do {@link JAXRSServerFactoryBean#setResourceClasses(Class...)}
* and {@link JAXRSServerFactoryBean#setResourceProvider(Class, org.apache.cxf.jaxrs.lifecycle.ResourceProvider)}
*/
protected abstract void setUpResources(JAXRSServerFactoryBean sf);
/**
* Have the test do {@link JAXRSServerFactoryBean#setProviders(java.util.List)}, if needed
*/
protected abstract void setUpProviders(JAXRSServerFactoryBean sf);
@After
public void tearDown() throws Exception {
server.stop();
server.destroy();
}
protected String getStringFromInputStream(InputStream in) throws Exception {
return IOUtils.toString(in, UTF_8);
}
protected Map<String, String> readZipArchive(InputStream inputStream) throws IOException {
Map<String, String> data = new HashMap<String, String>();
Path tempFile = writeTemporaryArchiveFile(inputStream, "zip");
ZipFile zip = new ZipFile(tempFile.toFile());
Enumeration<ZipArchiveEntry> entries = zip.getEntries();
while (entries.hasMoreElements()) {
ZipArchiveEntry entry = entries.nextElement();
ByteArrayOutputStream bos = new ByteArrayOutputStream();
IOUtils.copy(zip.getInputStream(entry), bos);
data.put(entry.getName(), DigestUtils.md5Hex(bos.toByteArray()));
}
zip.close();
Files.delete(tempFile);
return data;
}
protected String readArchiveText(InputStream inputStream) throws IOException {
Path tempFile = writeTemporaryArchiveFile(inputStream, "zip");
ZipFile zip = new ZipFile(tempFile.toFile());
zip.getEntry(UnpackerResource.TEXT_FILENAME);
ByteArrayOutputStream bos = new ByteArrayOutputStream();
IOUtils.copy(zip.getInputStream(zip.getEntry(UnpackerResource.TEXT_FILENAME)), bos);
zip.close();
Files.delete(tempFile);
return bos.toString(UTF_8.name());
}
protected Map<String, String> readArchiveFromStream(ArchiveInputStream zip) throws IOException {
Map<String, String> data = new HashMap<String, String>();
while (true) {
ArchiveEntry entry = zip.getNextEntry();
if (entry == null) {
break;
}
ByteArrayOutputStream bos = new ByteArrayOutputStream();
IOUtils.copy(zip, bos);
data.put(entry.getName(), DigestUtils.md5Hex(bos.toByteArray()));
}
return data;
}
private Path writeTemporaryArchiveFile(InputStream inputStream, String archiveType) throws IOException {
Path tmp = Files.createTempFile("apache-tika-server-test-tmp-", "."+archiveType);
Files.copy(inputStream, tmp, StandardCopyOption.REPLACE_EXISTING);
return tmp;
}
}