com.it.j2ee.email.MailServiceTest.java Source code

Introduction

Here is the source code for com.it.j2ee.email.MailServiceTest.java

Source

/*******************************************************************************
 * Copyright (c) 2005, 2014 springside.github.io
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 *******************************************************************************/
package com.it.j2ee.email;

import static org.assertj.core.api.Assertions.assertThat;

import java.io.IOException;

import javax.mail.MessagingException;
import javax.mail.Multipart;
import javax.mail.Part;
import javax.mail.internet.MimeMessage;
import javax.mail.internet.MimeMultipart;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ActiveProfiles;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import com.icegreen.greenmail.util.GreenMail;
import com.icegreen.greenmail.util.GreenMailUtil;
import com.it.j2ee.modules.email.MimeMailService;
import com.it.j2ee.modules.email.SimpleMailService;
import com.it.j2ee.spring.SpringContextTestCase;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = { "classpath:applicationContext.xml",
        "classpath:email/applicationContext-email.xml" })
@ActiveProfiles("dev")
public class MailServiceTest extends SpringContextTestCase {

    @Autowired
    private MimeMailService mimeMailService;

    @Autowired
    private SimpleMailService simpleMailService;

    @Autowired
    private GreenMail greenMail;

    @Test
    public void sendSimpleMail() throws MessagingException, InterruptedException, IOException {
        simpleMailService.sendNotificationMail("calvin");

        greenMail.waitForIncomingEmail(2000, 1);

        MimeMessage[] messages = greenMail.getReceivedMessages();
        MimeMessage message = messages[messages.length - 1];

        assertThat(message.getFrom()[0].toString()).isEqualTo("springside3.demo@gmail.com");
        assertThat(message.getSubject()).isEqualTo("");
        // text?
        System.out.println(message.getContent());
        assertThat(((String) message.getContent())).contains("");
    }

    @Test
    public void sendMimeMail() throws InterruptedException, MessagingException, IOException {
        mimeMailService.sendNotificationMail("calvin");

        greenMail.waitForIncomingEmail(2000, 1);
        MimeMessage[] messages = greenMail.getReceivedMessages();
        MimeMessage message = messages[messages.length - 1];

        assertThat(message.getFrom()[0].toString()).isEqualTo("springside3.demo@gmail.com");
        assertThat(message.getSubject()).isEqualTo("");

        MimeMultipart mimeMultipart = (MimeMultipart) message.getContent();

        assertThat(mimeMultipart.getCount()).isEqualTo(2);

        // Html?
        String mainPartText = getMainPartText(mimeMultipart.getBodyPart(0));
        System.out.println(mainPartText);
        assertThat(mainPartText).contains("<h1>calvin.</h1>");

        // 
        assertThat(GreenMailUtil.getBody(mimeMultipart.getBodyPart(1)).trim())
                .isEqualTo("Hello,i am a attachment.");
    }

    private String getMainPartText(Part mainPart) throws MessagingException, IOException {
        return (String) ((Multipart) mainPart.getContent()).getBodyPart(0).getContent();
    }
}