org.subethamail.smtp.MessageContext Java Examples

public SmtpConsumer(
        final MessageContext context,
        final ProcessSessionFactory sessionFactory,
        final int port,
        final String host,
        final ComponentLog log,
        final int maxMessageSize
) {
    this.context = context;
    this.sessionFactory = sessionFactory;
    this.port = port;
    if (host == null || host.trim().isEmpty()) {
        this.host = context.getSMTPServer().getHostName();
    } else {
        this.host = host;
    }
    this.log = log;
    this.maxMessageSize = maxMessageSize;
}
 

public SmtpConsumer(
        final MessageContext context,
        final ProcessSessionFactory sessionFactory,
        final int port,
        final String host,
        final ComponentLog log,
        final int maxMessageSize
) {
    this.context = context;
    this.sessionFactory = sessionFactory;
    this.port = port;
    if (host == null || host.trim().isEmpty()) {
        this.host = context.getSMTPServer().getHostName();
    } else {
        this.host = host;
    }
    this.log = log;
    this.maxMessageSize = maxMessageSize;
}
 

@Test
public void testCompletedMailTransaction() throws Exception {

	new Expectations() {
		{
			messageHandlerFactory.create((MessageContext) any);
			result = messageHandler;

			messageHandler.from(anyString);
			messageHandler.recipient(anyString);
			messageHandler.data((InputStream) any);
			messageHandler.done();
		}
	};

	SmartClient client = new SmartClient("localhost", smtpServer.getPort(),
			"localhost");
	client.from("[email protected]");
	client.to("[email protected]");
	client.dataStart();
	client.dataWrite(TextUtils.getAsciiBytes("body"), 4);
	client.dataEnd();
	client.quit();
	smtpServer.stop(); // wait for the server to catch up
}
 

@Test
public void testAbortedMailTransaction() throws Exception {

	new Expectations() {
		{
			messageHandlerFactory.create((MessageContext) any);
			result = messageHandler;

			messageHandler.from(anyString);
			messageHandler.done();
		}
	};

	SmartClient client = new SmartClient("localhost", smtpServer.getPort(),
			"localhost");
	client.from("[email protected]");
	client.quit();
	smtpServer.stop(); // wait for the server to catch up
}
 

@Override
public SMTPHandler create(MessageContext ctx) {

    SMTPHandler smtpHandler = new SMTPHandler(ctx);
    beanFactory.autowireBean(smtpHandler);
    return smtpHandler;
}
 

@Override
public MessageHandler create(MessageContext ctx) {
    MailTransactionImpl mailTransaction = new MailTransactionImpl(ctx);
    FilterInstances filterInstances =
            filters.createInstanceChain(mailTransaction);
    FilterChainMessageHandler filterChainMessageHandler =
            new FilterChainMessageHandler(filterInstances, mailTransaction);
    return new ErrorHandlerMessageHandler(filterChainMessageHandler);
}
 

@Test
public void testDisconnectImmediately() throws Exception {

	new Expectations() {
		{
			messageHandlerFactory.create((MessageContext) any);
			times = 0;
		}
	};

	SmartClient client = new SmartClient("localhost", smtpServer.getPort(),
			"localhost");
	client.quit();
	smtpServer.stop(); // wait for the server to catch up
}
 

public Handler(MessageContext ctx) {
    this.ctx = ctx;
}
 

public MessageHandler create(MessageContext ctx) {
    return new Handler(ctx);
}
 

MessageHandlerImpl(MessageContext context) {
    this.context = context;
}
 

private SMTPServer prepareServer(final ProcessContext context, final ProcessSessionFactory sessionFactory) {
    final int port = context.getProperty(SMTP_PORT).asInteger();
    final String host = context.getProperty(SMTP_HOSTNAME).getValue();
    final ComponentLog log = getLogger();
    final int maxMessageSize = context.getProperty(SMTP_MAXIMUM_MSG_SIZE).asDataSize(DataUnit.B).intValue();
    //create message handler factory
    final MessageHandlerFactory messageHandlerFactory = (final MessageContext mc) -> {
        return new SmtpConsumer(mc, sessionFactory, port, host, log, maxMessageSize);
    };
    //create smtp server
    final SSLContextService sslContextService = context.getProperty(SSL_CONTEXT_SERVICE).asControllerService(SSLContextService.class);
    final SMTPServer smtpServer = sslContextService == null ? new SMTPServer(messageHandlerFactory) : new SMTPServer(messageHandlerFactory) {
        @Override
        public SSLSocket createSSLSocket(Socket socket) throws IOException {
            InetSocketAddress remoteAddress = (InetSocketAddress) socket.getRemoteSocketAddress();
            String clientAuth = context.getProperty(CLIENT_AUTH).getValue();
            SSLContext sslContext = sslContextService.createSSLContext(SslContextFactory.ClientAuth.valueOf(clientAuth));
            SSLSocketFactory socketFactory = sslContext.getSocketFactory();
            SSLSocket sslSocket = (SSLSocket) (socketFactory.createSocket(socket, remoteAddress.getHostName(), socket.getPort(), true));
            sslSocket.setUseClientMode(false);

            if (SslContextFactory.ClientAuth.REQUIRED.toString().equals(clientAuth)) {
                this.setRequireTLS(true);
                sslSocket.setNeedClientAuth(true);
            }
            return sslSocket;
        }
    };
    if (sslContextService != null) {
        smtpServer.setEnableTLS(true);
    } else {
        smtpServer.setHideTLS(true);
    }
    smtpServer.setSoftwareName("Apache NiFi SMTP");
    smtpServer.setPort(port);
    smtpServer.setMaxConnections(context.getProperty(SMTP_MAXIMUM_CONNECTIONS).asInteger());
    smtpServer.setMaxMessageSize(maxMessageSize);
    smtpServer.setConnectionTimeout(context.getProperty(SMTP_TIMEOUT).asTimePeriod(TimeUnit.MILLISECONDS).intValue());
    if (context.getProperty(SMTP_HOSTNAME).isSet()) {
        smtpServer.setHostName(context.getProperty(SMTP_HOSTNAME).getValue());
    }
    return smtpServer;
}
 

public MockMessageHandler(MessageContext ctx) {
	this._ctx = ctx;
	if (null == _messages) {
		_messages = new ArrayList<MockMailMessage>();
	}
}
 

public MessageHandler create(MessageContext ctx) {
	if (this._messageHandler == null) {
		this._messageHandler = new MockMessageHandler(ctx);
	}
    return _messageHandler;
}
 

@Override
public MessageContext getMessageContext() {
    return messageContext;
}
 

public MailTransactionImpl(MessageContext messageContext) {
    super();
    this.messageContext = messageContext;
}
 

@Test
public void testTwoMailsInOneSession() throws Exception {

	new Expectations() {
		{
			messageHandlerFactory.create((MessageContext) any);
			result = messageHandler;

			onInstance(messageHandler).from(anyString);
			onInstance(messageHandler).recipient(anyString);
			onInstance(messageHandler).data((InputStream) any);
			onInstance(messageHandler).done();

			messageHandlerFactory.create((MessageContext) any);
			result = messageHandler2;

			onInstance(messageHandler2).from(anyString);
			onInstance(messageHandler2).recipient(anyString);
			onInstance(messageHandler2).data((InputStream) any);
			onInstance(messageHandler2).done();
		}
	};

	SmartClient client = new SmartClient("localhost", smtpServer.getPort(),
			"localhost");

	client.from("[email protected]");
	client.to("[email protected]");
	client.dataStart();
	client.dataWrite(TextUtils.getAsciiBytes("body1"), 5);
	client.dataEnd();

	client.from("[email protected]");
	client.to("[email protected]");
	client.dataStart();
	client.dataWrite(TextUtils.getAsciiBytes("body2"), 5);
	client.dataEnd();

	client.quit();

	smtpServer.stop(); // wait for the server to catch up
}
 

/** */
public Handler(MessageContext ctx)
{
	this.ctx = ctx;
}
 

public MessageHandler create(MessageContext ctx)
{
	return new Handler(ctx);
}
 

/** */
public Handler(MessageContext ctx)
{
	this.ctx = ctx;
}
 

public MessageHandler create(MessageContext ctx)
{
	return new Handler(ctx);
}
 

public MessageContext getMessageContext()
{
    return messageContext;
}
 

public Handler(MessageContext messageContext)
{
    this.messageContext = messageContext;
}
 

public MessageHandler create(MessageContext messageContext)
{
    return new Handler(messageContext);
}
 

private SMTPServer prepareServer(final ProcessContext context, final ProcessSessionFactory sessionFactory) {
    final int port = context.getProperty(SMTP_PORT).asInteger();
    final String host = context.getProperty(SMTP_HOSTNAME).getValue();
    final ComponentLog log = getLogger();
    final int maxMessageSize = context.getProperty(SMTP_MAXIMUM_MSG_SIZE).asDataSize(DataUnit.B).intValue();
    //create message handler factory
    final MessageHandlerFactory messageHandlerFactory = (final MessageContext mc) -> {
        return new SmtpConsumer(mc, sessionFactory, port, host, log, maxMessageSize);
    };
    //create smtp server
    final SSLContextService sslContextService = context.getProperty(SSL_CONTEXT_SERVICE).asControllerService(SSLContextService.class);
    final SMTPServer smtpServer = sslContextService == null ? new SMTPServer(messageHandlerFactory) : new SMTPServer(messageHandlerFactory) {
        @Override
        public SSLSocket createSSLSocket(Socket socket) throws IOException {
            InetSocketAddress remoteAddress = (InetSocketAddress) socket.getRemoteSocketAddress();
            String clientAuth = context.getProperty(CLIENT_AUTH).getValue();
            SSLContext sslContext = sslContextService.createSSLContext(SSLContextService.ClientAuth.valueOf(clientAuth));
            SSLSocketFactory socketFactory = sslContext.getSocketFactory();
            SSLSocket sslSocket = (SSLSocket) (socketFactory.createSocket(socket, remoteAddress.getHostName(), socket.getPort(), true));
            sslSocket.setUseClientMode(false);

            if (SSLContextService.ClientAuth.REQUIRED.toString().equals(clientAuth)) {
                this.setRequireTLS(true);
                sslSocket.setNeedClientAuth(true);
            }
            return sslSocket;
        }
    };
    if (sslContextService != null) {
        smtpServer.setEnableTLS(true);
    } else {
        smtpServer.setHideTLS(true);
    }
    smtpServer.setSoftwareName("Apache NiFi SMTP");
    smtpServer.setPort(port);
    smtpServer.setMaxConnections(context.getProperty(SMTP_MAXIMUM_CONNECTIONS).asInteger());
    smtpServer.setMaxMessageSize(maxMessageSize);
    smtpServer.setConnectionTimeout(context.getProperty(SMTP_TIMEOUT).asTimePeriod(TimeUnit.MILLISECONDS).intValue());
    if (context.getProperty(SMTP_HOSTNAME).isSet()) {
        smtpServer.setHostName(context.getProperty(SMTP_HOSTNAME).getValue());
    }
    return smtpServer;
}
 

@Test
public void shouldCreateHandler() throws Exception{

    MessageContext ctxMock = mock(MessageContext.class);
    SMTPHandler handlerMock = mock(SMTPHandler.class);
    whenNew(SMTPHandler.class).withArguments(ctxMock).thenReturn(handlerMock);

    SMTPHandler actual = smtpHandlerFactory.create(ctxMock);

    assertThat(actual).isEqualTo(handlerMock);
    verify(beanFactoryMock).autowireBean(handlerMock);



}
 

/**
 * @x.category GETSET
 */
MessageContext getMessageContext();
 

public SMTPHandler(MessageContext ctx) {

        InetSocketAddress hostAddr = (InetSocketAddress) ctx.getRemoteAddress();
        this.senderHost = hostAddr.getAddress().getHostAddress();

    }