Smtp.java

/*
 * $Id$
 * $URL$
 */

package org.subethamail.rtest.util;

import java.io.IOException;
import java.util.Iterator;
import java.util.logging.Level;

import javax.mail.MessagingException;

import lombok.extern.java.Log;

import org.junit.Assume;
import org.subethamail.core.post.i.Constant;
import org.subethamail.core.post.i.MailType;
import org.subethamail.rtest.ResinTestSetup;
import org.subethamail.wiser.Wiser;
import org.subethamail.wiser.WiserMessage;

/**
 * A useful wrapper for Wiser that provides some Subetha-specific
 * methods.
 *
 * @author Jeff Schnitzer
 */
@Log
public class Smtp extends Wiser
{
	/** The port we use for wiser */
	public static final int PORT = 2525;

	public Smtp()
	{
		super();
		this.setPort(PORT);
	}

	/** Take care of futzing the server side smtp config */
	@Override
	public void start()
	{
		super.start();

		String host = System.getProperty("org.subethamail.smtp.host");
		if (host == null)
			host = "localhost";

		try
		{
			AdminMixin god = new AdminMixin();
			log.log(Level.FINE,"Calling enableTestMode()");
			god.getEegor().enableTestMode(host + ":" + PORT);
			log.log(Level.FINE,"Called enableTestMode()");
		}
		catch (Exception ex) { throw new RuntimeException(ex); }
	}

	/** Take care of futzing the server side smtp config */
	@Override
	public void stop()
	{
		try
		{
			AdminMixin god = new AdminMixin();
			god.getEegor().disableTestMode();
		}
		catch (Exception ex) { throw new RuntimeException(ex); }

		super.stop();
	}

	/** */
	public int size()
	{
		return super.getMessages().size();
	}

	/** */
	public WiserMessage get(int index)
	{
		return super.getMessages().get(index);
	}

	/**
	 * Gets the Nth instance of the specified mail type
	 * @throws MessagingException
	 */
	public WiserMessage get(MailType type, int index) throws MessagingException
	{
		int count = 0;

		Iterator<WiserMessage> it = super.getMessages().iterator();
		while (it.hasNext())
		{
			WiserMessage msg = it.next();

			if (msg.getMimeMessage().getSubject().startsWith(type.name()))
			{
				if (count == index)
					return msg;
				else
					count++;
			}
		}

		return null;
	}

	/**
	 * @return the number of messages of the specified type
	 * @throws MessagingException
	 */
	public int count(MailType type) throws MessagingException
	{
		int count = 0;

		Iterator<WiserMessage> it = super.getMessages().iterator();
		while (it.hasNext())
		{
			WiserMessage msg = it.next();

			if (msg.getMimeMessage().getSubject().startsWith(type.name()))
				count++;
		}

		return count;
	}

	/**
	 * @return the number of messages containing the specified subject
	 * @throws MessagingException
	 */
	public int countSubject(String subject) throws MessagingException
	{
		int count = 0;

		Iterator<WiserMessage> it = super.getMessages().iterator();
		while (it.hasNext())
		{
			WiserMessage msg = it.next();

			if (msg.getMimeMessage().getSubject().contains(subject))
				count++;
		}

		return count;
	}

	/**
	 * Gets the Nth instance of the mail with the subject
	 * @throws MessagingException
	 */
	public WiserMessage getSubject(String subject, int index) throws MessagingException
	{
		int count = 0;

		Iterator<WiserMessage> it = super.getMessages().iterator();
		while (it.hasNext())
		{
			WiserMessage msg = it.next();

			if (msg.getMimeMessage().getSubject().contains(subject))
			{
				if (count == index)
					return msg;
				else
					count++;
			}
		}

		return null;
	}
	
	/**
	 * @return the last message received
	 */
	public WiserMessage getLastMessage() throws MessagingException
	{
		return super.getMessages().get(super.getMessages().size() - 1);
	}

	/**
	 * Gets an embedded token from a message.
	 * @throws MessagingException
	 * @throws IOException
	 *
	 * @throws IllegalArgumentException if msg has no token
	 */
	public static String extractToken(WiserMessage msg) throws IOException, MessagingException
	{
		String body = (String) msg.getMimeMessage().getContent();

		int start = body.indexOf(Constant.DEBUG_TOKEN_BEGIN);
		if (start < 0)
			throw new IllegalStateException("Missing token from email: " + body);

		start += Constant.DEBUG_TOKEN_BEGIN.length();

		int end = body.indexOf(Constant.DEBUG_TOKEN_END, start);
		if (end < 0)
			throw new IllegalStateException("Missing token from email: " + body);

		return body.substring(start, end);
	}

	/**
	 * If debug is enabled, print subjects
	 */
	public void debugPrintSubjects() throws MessagingException
	{
	    if (log.isLoggable(Level.FINE))
		{
	        log.log(Level.FINE,"Messages we have received:");
			for (WiserMessage msg: this.getMessages())
			    log.log(Level.FINE,"   Subject: " + msg.getMimeMessage().getSubject());
		}

	}
}