EmailNotificationTest.java Source code

Introduction

Here is the source code for EmailNotificationTest.java

Source

/**
 * Copyright 2010 The University of North Carolina at Chapel Hill
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
import static org.junit.Assert.*;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.reset;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

import javax.annotation.Resource;
import javax.mail.internet.MimeMessage;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import org.springframework.mail.SimpleMailMessage;
import org.springframework.mail.javamail.JavaMailSender;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import cdr.forms.CachedXMIFormFactory;
import cdr.forms.DateDepositField;
import cdr.forms.Deposit;
import cdr.forms.DepositElement;
import cdr.forms.DepositEntry;
import cdr.forms.DepositField;
import cdr.forms.DepositResult;
import cdr.forms.Submission;
import cdr.forms.DepositResult.Status;
import cdr.forms.EmailDepositField;
import cdr.forms.EmailNotificationHandler;
import cdr.forms.TextDepositField;
import crosswalk.DateInputField;
import crosswalk.Form;
import crosswalk.FormElement;
import crosswalk.InputField;
import crosswalk.MetadataBlock;
import crosswalk.TextInputField;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = { "/email-test-service-context.xml" })
public class EmailNotificationTest {

    @Resource
    JavaMailSender javaMailSender = null;

    @Resource
    EmailNotificationHandler emailNotificationHandler = null;

    @Resource
    CachedXMIFormFactory formFactory = null;

    public void setMailMock() {
        // setup mail sender mock invocations
        reset(this.javaMailSender);
        when(this.javaMailSender.createMimeMessage()).thenCallRealMethod();

        Answer dumpMessage = new Answer() {
            @Override
            public Object answer(InvocationOnMock invocation) throws Throwable {
                Object arg = invocation.getArguments()[0];
                if (arg instanceof MimeMessage) {
                    MimeMessage m = (MimeMessage) arg;
                    System.out.println("EMAIL DUMP:");
                    m.writeTo(System.out);
                } else if (arg instanceof SimpleMailMessage) {
                    SimpleMailMessage m = (SimpleMailMessage) arg;
                    System.out.println("EMAIL DUMP:");
                    System.out.println(m.toString());
                } else {
                    throw new Error("Could not print email: " + arg);
                }
                return null;
            }
        };
        doAnswer(dumpMessage).when(this.javaMailSender).send(any(MimeMessage.class));
        doAnswer(dumpMessage).when(this.javaMailSender).send(any(SimpleMailMessage.class));
    }

    @Test
    public void testDepositNotification() {
        setMailMock();

        Form form = this.formFactory.getForm("test");
        Deposit deposit = buildDeposit(form);

        DepositResult result = new DepositResult();
        result.setAccessURL("http://example.org/the/deposit/url");
        result.setStatus(Status.PENDING);

        emailNotificationHandler.notifyDeposit(deposit, result);
        verify(this.javaMailSender, times(2)).send(any(MimeMessage.class));
    }

    @Test
    public void testDepositError() {
        setMailMock();

        Form form = this.formFactory.getForm("test");
        Deposit deposit = buildDeposit(form);

        DepositResult result = new DepositResult();
        result.setAccessURL("http://example.org/the/deposit/url");
        result.setStatus(Status.FAILED);
        Throwable exception = new Exception("example error trace").fillInStackTrace();
        StringWriter sw = new StringWriter();
        exception.printStackTrace(new PrintWriter(sw));
        result.setResponseBody(sw.toString());

        emailNotificationHandler.notifyError(deposit, result);
        verify(this.javaMailSender, times(1)).send(any(MimeMessage.class));
    }

    private Deposit buildDeposit(Form form) {

        form.setCurrentUser("testuser");

        Deposit deposit = new Deposit();

        deposit.setForm(form);
        deposit.setFormId("test");
        deposit.setElements(new ArrayList<DepositElement>());
        deposit.setReceiptEmailAddress("receipt@email.address");

        for (FormElement element : form.getElements()) {

            DepositElement depositElement = new DepositElement();
            depositElement.setFormElement(element);
            depositElement.setEntries(new ArrayList<DepositEntry>());

            depositElement.appendEntry();

            if (element instanceof MetadataBlock && ((MetadataBlock) element).getMaxRepeat() > 1) {
                depositElement.appendEntry();
            }

            for (DepositEntry entry : depositElement.getEntries()) {

                if (entry.getFields() != null) {

                    for (DepositField<?> field : entry.getFields()) {

                        if (field instanceof DateDepositField) {
                            ((DateDepositField) field).setValue(new Date());
                        } else if (field instanceof TextDepositField) {
                            ((TextDepositField) field).setValue("Test");
                        } else if (field instanceof EmailDepositField) {
                            ((EmailDepositField) field).setValue("email@deposit.field");
                        }

                    }

                }

            }

            deposit.getElements().add(depositElement);

        }

        return deposit;

    }

}