FallbackTest.java

/*
 * $Id: EmailAddressTest.java 1105 2009-05-10 04:16:03Z lhoriman $
 * $URL: https://subetha.googlecode.com/svn/branches/resin/rtest/src/org/subethamail/rtest/EmailAddressTest.java $
 */

package org.subethamail.rtest;

import static org.junit.Assert.assertEquals;

import javax.mail.Message.RecipientType;
import javax.mail.Transport;
import javax.mail.internet.MimeMessage;

import org.junit.After;
import org.junit.Assume;
import org.junit.Before;
import org.subethamail.rtest.util.AdminMixin;
import org.subethamail.rtest.util.MailingListMixin;
import org.subethamail.rtest.util.PersonInfoMixin;
import org.subethamail.rtest.util.SubEthaTestCase;
import org.subethamail.wiser.Wiser;

/**
 * Testing the fallthrough mechanism.
 * 
 * @author Jeff Schnitzer
 */
public class FallbackTest extends SubEthaTestCase
{
	/** */
	public static final int FALLBACK_PORT = 2526;

	/** */
	Wiser fallback;
	AdminMixin admin;
	MailingListMixin ml;
	PersonInfoMixin pers;
	boolean runteardown=false;
	
	/** */
	@Before
	public void setUp() throws Exception
	{
		super.setUp();
		
		this.fallback = new Wiser(FALLBACK_PORT);
		this.fallback.start();
		
		this.admin = new AdminMixin();
		this.ml = new MailingListMixin(this.admin, null);
		this.pers = new PersonInfoMixin();
		
		this.admin.getEegor().setFallbackHost("localhost:" + FALLBACK_PORT);
	}
	
	/** */
	@After
	public void tearDown() throws Exception
	{
	    if (!runteardown) return;
	    
		this.fallback.stop();
		this.admin.getEegor().setFallbackHost(null);
		
		super.tearDown();
	}
	
	/** */
    @org.junit.Test
	public void testListOnly() throws Exception
	{
        Assume.assumeTrue(ResinTestSetup.exists());
		MimeMessage msg = this.createMimeMessage(this.pers.getAddress(), this.ml.getAddress());
		Transport.send(msg);
		Thread.sleep(1000);
		
		assertEquals(0, this.fallback.getMessages().size());
		assertEquals(0, this.smtp.countSubject(TEST_SUBJECT));
		assertEquals(1, this.admin.getArchiver().countMailByList(this.ml.getId()));
	}
	
	/** */
    @org.junit.Test
	public void testTwoLists() throws Exception
	{
        Assume.assumeTrue(ResinTestSetup.exists());
		MailingListMixin ml2 = new MailingListMixin(this.admin, null);
		
		MimeMessage msg = this.createMimeMessage(this.pers.getAddress(), this.ml.getAddress());
		msg.addRecipient(RecipientType.TO, ml2.getAddress());
		Transport.send(msg);
		Thread.sleep(1000);
		
		assertEquals(0, this.fallback.getMessages().size());
		assertEquals(0, this.smtp.countSubject(TEST_SUBJECT));
		assertEquals(1, this.admin.getArchiver().countMailByList(this.ml.getId()));
		assertEquals(1, this.admin.getArchiver().countMailByList(ml2.getId()));
	}
	
	/** */
    @org.junit.Test
	public void testFallthroughOnly() throws Exception
	{
        Assume.assumeTrue(ResinTestSetup.exists());
		PersonInfoMixin pers2 = new PersonInfoMixin();
		
		MimeMessage msg = this.createMimeMessage(this.pers.getAddress(), pers2.getAddress());
		Transport.send(msg);
		Thread.sleep(1000);
		
		assertEquals(1, this.fallback.getMessages().size());
		assertEquals(0, this.smtp.countSubject(TEST_SUBJECT));
		assertEquals(0, this.admin.getArchiver().countMailByList(this.ml.getId()));
	}
	
	/** */
    @org.junit.Test
	public void testTwoFallthroughs() throws Exception
	{
        Assume.assumeTrue(ResinTestSetup.exists());
		PersonInfoMixin pers2 = new PersonInfoMixin();
		PersonInfoMixin pers3 = new PersonInfoMixin();
		
		MimeMessage msg = this.createMimeMessage(this.pers.getAddress(), pers2.getAddress());
		msg.addRecipient(RecipientType.TO, pers3.getAddress());
		Transport.send(msg);
		Thread.sleep(1000);
		
		assertEquals(2, this.fallback.getMessages().size());
		assertEquals(0, this.smtp.countSubject(TEST_SUBJECT));
		assertEquals(0, this.admin.getArchiver().countMailByList(this.ml.getId()));
	}
	
	/** */
    @org.junit.Test
	public void testFallthroughPlusList() throws Exception
	{
        Assume.assumeTrue(ResinTestSetup.exists());
		PersonInfoMixin pers2 = new PersonInfoMixin();
		
		MimeMessage msg = this.createMimeMessage(this.pers.getAddress(), this.ml.getAddress());
		msg.addRecipient(RecipientType.TO, pers2.getAddress());
		Transport.send(msg);
		Thread.sleep(1000);
		
		assertEquals(1, this.fallback.getMessages().size());
		assertEquals(0, this.smtp.countSubject(TEST_SUBJECT));
		assertEquals(1, this.admin.getArchiver().countMailByList(this.ml.getId()));
	}	
}