org.sipfoundry.voicemail.EmailFormatterTest.java Source code

Introduction

Here is the source code for org.sipfoundry.voicemail.EmailFormatterTest.java

Source

/*
 * Copyright (C) 2010 Avaya, certain elements licensed under a Contributor Agreement.
 * Contributors retain copyright to elements licensed under a Contributor Agreement.
 * Licensed to the User under the LGPL license.
 */
package org.sipfoundry.voicemail;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.text.MessageFormat;
import java.util.Date;

import javax.sound.sampled.AudioFileFormat;
import javax.sound.sampled.AudioFormat;
import javax.sound.sampled.AudioInputStream;
import javax.sound.sampled.AudioSystem;

import junit.framework.TestCase;

import org.apache.commons.io.FileUtils;
import org.sipfoundry.commons.userdb.User;
import org.sipfoundry.commons.userdb.User.EmailFormats;
import org.sipfoundry.sipxivr.IvrConfiguration;
import org.sipfoundry.sipxivr.Mailbox;
import org.sipfoundry.voicemail.MessageDescriptor.Priority;

public class EmailFormatterTest extends TestCase {

    User m_user;
    File m_testdir;
    File m_mailstore;
    Mailbox m_mailbox;
    IvrConfiguration m_ivrConfig;

    protected void setUp() throws Exception {
        super.setUp();
        m_user = new User();
        m_user.setUserName("woof");
        m_user.setDisplayName("Fuzzy Puppy");
        m_user.setIdentity("woof@pingtel.com");
        m_user.setUri("\"Fuzzy Puppy\" <woof@pingtel.com>");
        m_testdir = new File("/tmp/MessageTest/");
        if (m_testdir.isDirectory()) {
            FileUtils.forceDelete(m_testdir);
        }
        m_mailstore = new File(m_testdir.getPath() + "/mailstore/");
        m_mailstore.mkdirs();
        m_mailbox = new Mailbox(m_user, m_mailstore.getPath());
        Mailbox.createDirsIfNeeded(m_mailbox);
        Mwi.setJustTesting(true);
        m_ivrConfig = IvrConfiguration.getTest();
        m_ivrConfig.setConfigUrl("https://configserver/");
        Emailer.setJustTesting(true);
        if (false) {
            // Test actually sending e-mail.  Don't use normally, as I don't want to get flooded with mail!
            Emailer.init(m_ivrConfig);
            Emailer.setJustTesting(false);
            m_mailbox.getUser().setEmailFormat(EmailFormats.FORMAT_FULL);
            m_mailbox.getUser().setAttachAudioToEmail(true);
            m_mailbox.getUser().setEmailAddress("woof@iwoof.org");
        }
    }

    protected void tearDown() throws Exception {
        super.tearDown();
        if (m_testdir.isDirectory()) {
            FileUtils.forceDelete(m_testdir);
        }
    }

    private void makeWaves(File wavFile, byte filler, int length) throws IOException {
        byte[] fill = new byte[length];
        for (int i = 0; i < length; i++) {
            fill[i] = filler;
        }
        AudioInputStream ais = new AudioInputStream(new ByteArrayInputStream(fill),
                new AudioFormat(8000, 16, 1, true, false), fill.length);
        AudioSystem.write(ais, AudioFileFormat.Type.WAVE, wavFile);
    }

    public void testFormats() throws IOException {
        File wavFile = new File(m_testdir.getPath() + "fake.wav");
        makeWaves(wavFile, (byte) 0, 8000 * (60 * 2 + 42)); // two minutes, 42 seconds
        Message message = Message.newMessage(m_mailbox, wavFile, "\"El Nio\" <sip:5551212@pingtel.com>",
                Priority.NORMAL, null);
        Date stamp = new Date(message.getTimestamp());
        message.storeInInbox();
        if (!Emailer.isJustTesting()) {
            try {
                Thread.sleep(2000); // Need to allow e-mail to be sent
            } catch (InterruptedException e) {
            }
        }
        EmailFormatter eft = EmailFormatter.getEmailFormatter(EmailFormats.FORMAT_FULL, m_ivrConfig, m_mailbox,
                message.getVmMessage());
        assertEquals("2:42 Voice Message from 5551212 (El Nio)", eft.getSubject());
        assertEquals("Voicemail Notification Service <postmaster@localhost>", eft.getSender());

        EmailFormatter smsFt = EmailFormatter.getEmailFormatter(EmailFormats.FORMAT_BRIEF, m_ivrConfig, m_mailbox,
                message.getVmMessage());
        assertEquals("Voice Message: El Nio", smsFt.getSubject());
        assertEquals("Msg from 5551212 duration 2:42 at " + MessageFormat.format("{0,date,short}", stamp),
                smsFt.getTextBody());
    }

}