LoginSystem/src/test/java/fr/xephi/authme/mail/SendMailSslTest.java

package fr.xephi.authme.mail;

import ch.jalu.injector.testing.BeforeInjecting;
import ch.jalu.injector.testing.DelayedInjectionRunner;
import ch.jalu.injector.testing.InjectDelayed;
import fr.xephi.authme.TestHelper;
import fr.xephi.authme.output.LogLevel;
import fr.xephi.authme.settings.Settings;
import fr.xephi.authme.settings.properties.EmailSettings;
import fr.xephi.authme.settings.properties.PluginSettings;
import org.apache.commons.mail.EmailException;
import org.apache.commons.mail.HtmlEmail;
import org.junit.BeforeClass;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;
import org.junit.runner.RunWith;
import org.mockito.Mock;

import java.io.IOException;
import java.util.Properties;

import static org.hamcrest.Matchers.equalTo;
import static org.hamcrest.Matchers.hasSize;
import static org.hamcrest.Matchers.not;
import static org.hamcrest.Matchers.nullValue;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.mockito.BDDMockito.given;

/**
 * Test for {@link SendMailSsl}.
 */
@RunWith(DelayedInjectionRunner.class)
public class SendMailSslTest {

    @InjectDelayed
    private SendMailSsl sendMailSsl;

    @Mock
    private Settings settings;

    @Rule
    public TemporaryFolder temporaryFolder = new TemporaryFolder();

    @BeforeClass
    public static void initLogger() {
        TestHelper.setupLogger();
    }

    @BeforeInjecting
    public void initFields() throws IOException {
        given(settings.getProperty(EmailSettings.MAIL_ACCOUNT)).willReturn("mail@example.org");
        given(settings.getProperty(EmailSettings.MAIL_PASSWORD)).willReturn("pass1234");
        given(settings.getProperty(PluginSettings.LOG_LEVEL)).willReturn(LogLevel.INFO);
    }

    @Test
    public void shouldHaveAllInformation() {
        // given / when / then
        assertThat(sendMailSsl.hasAllInformation(), equalTo(true));
    }

    @Test
    public void shouldCreateEmailObject() throws EmailException {
        // given
        given(settings.getProperty(EmailSettings.SMTP_PORT)).willReturn(465);
        String smtpHost = "mail.example.com";
        given(settings.getProperty(EmailSettings.SMTP_HOST)).willReturn(smtpHost);
        String senderAccount = "sender@example.org";
        given(settings.getProperty(EmailSettings.MAIL_ACCOUNT)).willReturn(senderAccount);
        String senderName = "Server administration";
        given(settings.getProperty(EmailSettings.MAIL_SENDER_NAME)).willReturn(senderName);
        given(settings.getProperty(PluginSettings.LOG_LEVEL)).willReturn(LogLevel.DEBUG);

        // when
        HtmlEmail email = sendMailSsl.initializeMail("recipient@example.com");

        // then
        assertThat(email, not(nullValue()));
        assertThat(email.getToAddresses(), hasSize(1));
        assertThat(email.getToAddresses().get(0).getAddress(), equalTo("recipient@example.com"));
        assertThat(email.getFromAddress().getAddress(), equalTo(senderAccount));
        assertThat(email.getFromAddress().getPersonal(), equalTo(senderName));
        assertThat(email.getHostName(), equalTo(smtpHost));
        assertThat(email.getSmtpPort(), equalTo("465"));
    }

    @Test
    public void shouldCreateEmailObjectWithAddress() throws EmailException {
        // given
        given(settings.getProperty(EmailSettings.SMTP_PORT)).willReturn(465);
        String smtpHost = "mail.example.com";
        given(settings.getProperty(EmailSettings.SMTP_HOST)).willReturn(smtpHost);
        String senderAccount = "exampleAccount";
        given(settings.getProperty(EmailSettings.MAIL_ACCOUNT)).willReturn(senderAccount);
        String senderAddress = "mail@example.com";
        given(settings.getProperty(EmailSettings.MAIL_ADDRESS)).willReturn(senderAddress);
        String senderName = "Server administration";
        given(settings.getProperty(EmailSettings.MAIL_SENDER_NAME)).willReturn(senderName);

        // when
        HtmlEmail email = sendMailSsl.initializeMail("recipient@example.com");

        // then
        assertThat(email, not(nullValue()));
        assertThat(email.getToAddresses(), hasSize(1));
        assertThat(email.getToAddresses().get(0).getAddress(), equalTo("recipient@example.com"));
        assertThat(email.getFromAddress().getAddress(), equalTo(senderAddress));
        assertThat(email.getFromAddress().getPersonal(), equalTo(senderName));
        assertThat(email.getHostName(), equalTo(smtpHost));
        assertThat(email.getSmtpPort(), equalTo("465"));
    }

    @Test
    public void shouldCreateEmailObjectWithOAuth2() throws EmailException {
        // given
        given(settings.getProperty(EmailSettings.SMTP_PORT)).willReturn(587);
        given(settings.getProperty(EmailSettings.OAUTH2_TOKEN)).willReturn("oAuth2 token");
        String smtpHost = "mail.example.com";
        given(settings.getProperty(EmailSettings.SMTP_HOST)).willReturn(smtpHost);
        String senderMail = "sender@example.org";
        given(settings.getProperty(EmailSettings.MAIL_ACCOUNT)).willReturn(senderMail);

        // when
        HtmlEmail email = sendMailSsl.initializeMail("recipient@example.com");

        // then
        assertThat(email, not(nullValue()));
        assertThat(email.getToAddresses(), hasSize(1));
        assertThat(email.getToAddresses().get(0).getAddress(), equalTo("recipient@example.com"));
        assertThat(email.getFromAddress().getAddress(), equalTo(senderMail));
        assertThat(email.getHostName(), equalTo(smtpHost));
        assertThat(email.getSmtpPort(), equalTo("587"));

        Properties mailProperties = email.getMailSession().getProperties();
        assertThat(mailProperties.getProperty("mail.smtp.auth.mechanisms"), equalTo("XOAUTH2"));
        assertThat(mailProperties.getProperty("mail.smtp.auth.plain.disable"), equalTo("true"));
        assertThat(mailProperties.getProperty(OAuth2SaslClientFactory.OAUTH_TOKEN_PROP), equalTo("oAuth2 token"));
    }
}