mail-archive/server/src/test/java/org/apache/sling/mailarchiveserver/impl/MessageStoreImplRepositoryTest.java

/*
 * 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.sling.mailarchiveserver.impl;

import static org.apache.sling.mailarchiveserver.impl.MessageStoreImplRepositoryTestUtil.assertValueMap;
import static org.apache.sling.mailarchiveserver.impl.MessageStoreImplRepositoryTestUtil.getResourcePath;
import static org.apache.sling.mailarchiveserver.impl.MessageStoreImplRepositoryTestUtil.readTextFile;
import static org.apache.sling.mailarchiveserver.impl.MessageStoreImplRepositoryTestUtil.specialPathFromFilePath;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import org.apache.james.mime4j.MimeException;
import org.apache.james.mime4j.dom.Message;
import org.apache.james.mime4j.dom.MessageBuilder;
import org.apache.james.mime4j.message.DefaultMessageBuilder;
import org.apache.sling.api.resource.ModifiableValueMap;
import org.apache.sling.api.resource.Resource;
import org.apache.sling.api.resource.ResourceResolver;
import org.apache.sling.discovery.impl.setup.MockedResourceResolver;
import org.apache.sling.mailarchiveserver.api.MboxParser;
import org.apache.sling.mailarchiveserver.util.MailArchiveServerConstants;
import org.apache.sling.mailarchiveserver.util.TU;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class MessageStoreImplRepositoryTest {
	private ResourceResolver resolver;
	private Resource testRoot;
	private MessageStoreImpl store;

	static final String TEST_RT_KEY = "sling_resourceType";
	private static final String BODY_SUFFIX = "_body";
	private static final String HEADERS_SUFFIX = "_headers";

	private static final String SINGLEPART_FILE = "singlepart.txt";
	private static final String MULTIPART_FILE = "multipart.txt";
	private static final String WRONGBODY_FILE = "wrongbody.txt";
	private static final String MBOX_FILE = "three_messages.mbox";

	/**
	 * Some code is taken from http://svn.apache.org/repos/asf/sling/trunk/launchpad/test-services/src/main/java/org/apache/sling/launchpad/testservices/serversidetests/WriteableResourcesTest.java
	 */
	@Before
	public void setup() throws Exception {
		resolver = new MockedResourceResolver();
		assertNotNull("Expecting non-null ResourceResolver", resolver);
		final Resource root = resolver.getResource("/");
		assertNotNull("Expecting non-null root Resource", root);
		final String path = getClass().getSimpleName() + "_" + System.currentTimeMillis();
		testRoot = resolver.create(root, path, null);
		resolver.commit();

		store = new MessageStoreImpl() {
		    protected ResourceResolver getResourceResolver() {
		        return resolver;
		    }
		};
		store.threadKeyGen = new ThreadKeyGeneratorImpl();
		store.archivePath = testRoot.getPath() + "/";
		store.resourceTypeKey = TEST_RT_KEY;
	}

	@After
	public void cleanup() throws Exception {
		resolver.close();
		resolver = null;
		testRoot = null;
		store = null;
	}


	@Test
	public void testSaveMessage() throws FileNotFoundException, MimeException, IOException  {
		assertSaveMessage(SINGLEPART_FILE);
		assertSaveMessage(MULTIPART_FILE);
		assertSaveMessage(WRONGBODY_FILE);
	}

	@Test
	public void testStructure() throws IOException {
		MboxParser parser = new Mime4jMboxParserImpl();
		final File file = new File(TU.TEST_FOLDER, MBOX_FILE);
		store.saveAll(parser.parse(new FileInputStream(file)));
		assertStructure();
	}

	private void assertSaveMessage(String messageFile) throws MimeException, IOException, FileNotFoundException {
		MessageBuilder builder = new DefaultMessageBuilder();
		Message msg = builder.parseMessage(new FileInputStream(new File(TU.TEST_FOLDER, messageFile)));

		store.save(msg);

		final Resource r = resolver.getResource(getResourcePath(msg, store));
		assertNotNull("Expecting non-null Resource", r);
		final ModifiableValueMap m = r.adaptTo(ModifiableValueMap.class);

		File bodyFile = new File(TU.TEST_FOLDER, specialPathFromFilePath(messageFile, BODY_SUFFIX));
		if (bodyFile.exists()) {
			String expectedBody = readTextFile(bodyFile);
			assertValueMap(m, "Body", expectedBody);
		}

		File headersFile = new File(TU.TEST_FOLDER, specialPathFromFilePath(messageFile, HEADERS_SUFFIX));
		if (headersFile.exists()) {
			MessageStoreImplRepositoryTestUtil.assertHeaders(headersFile, m);
		}

		assertTrue(headersFile.exists() || bodyFile.exists()); // test at least something 
	}

	private void assertStructure() {
		List<String> types = new ArrayList<String>();
		types.add(MailArchiveServerConstants.DOMAIN_RT);
		types.add(MailArchiveServerConstants.LIST_RT);
		types.add(null);
		types.add(null);
		types.add(MailArchiveServerConstants.THREAD_RT);
		types.add(MailArchiveServerConstants.MESSAGE_RT);

		MessageStoreImplRepositoryTestUtil.assertLayer(testRoot, types, 0);
	}

}