com.github.sleroy.junit.mail.server.MailServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fakesmtp-junit-runner Show documentation
JUnit Rule to create a FakeSmtp server to write mailing integration tests
There is a newer version: 0.1.1
Show newest version
/*
 *
 */
package com.github.sleroy.junit.mail.server;

import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.List;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.github.sleroy.fakesmtp.core.ServerConfiguration;
import com.github.sleroy.fakesmtp.core.exception.BindPortException;
import com.github.sleroy.fakesmtp.core.exception.InvalidHostException;
import com.github.sleroy.fakesmtp.core.exception.InvalidPortException;
import com.github.sleroy.fakesmtp.core.exception.OutOfRangePortException;
import com.github.sleroy.fakesmtp.model.EmailModel;
import com.github.sleroy.fakesmtp.model.MailServerModel;

/**
 * The Class MailServer initializes the MailServer.
 */
public class MailServer implements AutoCloseable {

    /** The Constant LOGGER. */
    private static final Logger LOGGER = LoggerFactory.getLogger(MailServer.class);

    /** The smtp auth handler. */
    private SMTPAuthHandler smtpAuthHandler;

    /** The server configuration. */
    private final ServerConfiguration serverConfiguration;

    /** The mail saver. */
    private MailSaver mailSaver;

    /** The mail listener. */
    private MailListener mailListener;

    /** The smtp auth handler factory. */
    private SMTPAuthHandlerFactory smtpAuthHandlerFactory;

    /** The smtp server handler. */
    private SMTPServerHandler smtpServerHandler;

    /** The mail server model. */
    private MailServerModel mailServerModel;

    /** The started. */
    private boolean started = false;

    /**
     * Instantiates a new mail server.
     *
     * @param serverConfiguration
     *            the server configuration
     */
    public MailServer(final ServerConfiguration serverConfiguration) {
	super();
	this.serverConfiguration = serverConfiguration;

    }

    /*
     * (non-Javadoc)
     *
     * @see java.lang.AutoCloseable#close()
     */
    @Override
    public void close() throws Exception {
	LOGGER.info("Stopping the server");
	if (smtpServerHandler != null) {
	    smtpServerHandler.stopServer();
	}

    }

    /**
     * Gets the mails of the server.
     *
     * @return the mails
     */
    public List getMails() {

	return mailServerModel.getEmailModels();
    }

    /**
     * Gets the rejected mails from the server.
     *
     * @return the rejected mails
     */
    public List getRejectedMails() {

	return mailServerModel.getRejectedMails();
    }

    /**
     * Gets the relay domains.
     *
     * @return the relay domains
     */
    public List getRelayDomains() {
	return serverConfiguration.getRelayDomains();
    }

    /**
     * Checks if the mail server is running.
     *
     * @return true, if is running
     */
    public boolean isRunning() {

	return started;
    }

    /**
     * Start.
     *
     * @throws BindPortException
     *             the bind port exception
     * @throws OutOfRangePortException
     *             the out of range port exception
     * @throws InvalidPortException
     *             the invalid port exception
     * @throws InvalidHostException
     *             the invalid host exception
     */
    public void start() throws BindPortException, OutOfRangePortException, InvalidPortException, InvalidHostException {

	if (started) {
	    LOGGER.warn("The server has been already started");
	    // Do nothing. We can't stop the server. User has to quit the app
	    // (issue with SubethaSMTP)
	} else {
	    // Model
	    LOGGER.info("Configuration of the server. {}", serverConfiguration);

	    mailServerModel = new MailServerModel(); // Cycle

	    smtpAuthHandler = new SMTPAuthHandler(serverConfiguration.getAuthentication().getUserName(),
	            serverConfiguration.getAuthentication().getPassword());

	    smtpAuthHandlerFactory = new SMTPAuthHandlerFactory(smtpAuthHandler);

	    mailSaver = new MailSaver(serverConfiguration);

	    // Adding MailServerModel to obtain the notifications
	    mailSaver.addObserver(mailServerModel);

	    mailListener = new MailListener(mailSaver);

	    smtpServerHandler = new SMTPServerHandler(mailListener, smtpAuthHandlerFactory);

	    final String hostStr = serverConfiguration.getBindAddress();
	    started = false;
	    try {
		final int port = Integer.parseInt(serverConfiguration.getPort());
		InetAddress host = null;
		if (hostStr != null && !hostStr.isEmpty()) {
		    host = InetAddress.getByName(hostStr);
		}

		smtpServerHandler.startServer(port, host);
		started = true;
	    } catch (final NumberFormatException e) {
		throw new InvalidPortException(e);
	    } catch (final UnknownHostException e) {
		throw new InvalidHostException(e, hostStr);
	    }
	}
    }

}