linearprocessortest.java

java mail,java mailjava mailjava mailjava mail

/****************************************************************
 * Licensed to the Apache Software Foundation (ASF) under one   *
 * or more contributor license agreements.  See the NOTICE file *
 * distributed with this work for additional information        *
 * regarding copyright ownership.  The ASF licenses this file   *
 * to you 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.                                           *
 ****************************************************************/
package org.apache.james.transport;

import org.apache.avalon.framework.logger.ConsoleLogger;
import org.apache.avalon.framework.logger.Logger;
import org.apache.james.core.MailImpl;
import org.apache.james.core.MimeMessageCopyOnWriteProxy;
import org.apache.james.core.MimeMessageInputStreamSource;
import org.apache.james.test.mock.james.MockSpoolRepository;
import org.apache.james.test.mock.mailet.MockMailContext;
import org.apache.james.test.mock.mailet.MockMailetConfig;
import org.apache.james.transport.mailets.debug.DumpSystemErr;
import org.apache.james.transport.matchers.All;
import org.apache.james.transport.matchers.RecipientIs;
import org.apache.mailet.GenericMailet;
import org.apache.mailet.Mail;
import org.apache.mailet.MailAddress;
import org.apache.mailet.Mailet;
import org.apache.mailet.MailetConfig;
import org.apache.mailet.MailetContext;
import org.apache.mailet.Matcher;
import org.apache.mailet.MatcherConfig;

import javax.mail.MessagingException;
import javax.mail.internet.MimeMessage;
import javax.mail.util.SharedByteArrayInputStream;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;

import junit.framework.TestCase;

/**
 * Currently here as a proof of JAMES-421.
 * Fixing code will follow
 */
public class LinearProcessorTest extends TestCase {
    Collection c;
    LinearProcessor t;
    MimeMessage mw = null;
    String content = "Subject: test\r\nContent-Transfer-Encoding: plain";
    String sep = "\r\n\r\n";
    String body = "original body\r\n.\r\n";
    MailetContext       mockContext = new MockMailContext();

    public static int counter = 0;
    
    
    
    private class CheckerMailet extends GenericMailet {
        
        public ArrayList receivedMails = new ArrayList();

        public void service(Mail mail) throws MessagingException {
            Mail m2 = new MailImpl(mail,mail.getName());
            m2.setState(mail.getState());
            receivedMails.add(m2);
        }
        
    }
    
    private MailetConfig mockMailetConfig = new MockMailetConfig("Dummy",mockContext);
    
    private CheckerMailet checkerMailet;
    
    private class MyMailet extends GenericMailet {

        public void service(Mail mail) throws MessagingException {
      try {
        MimeMessage message = mail.getMessage () ;
        //Set the header name and value (supplied at init time).
        String newText =  "new text "+counter++;
        System.err.println("Setting body to "+newText);
        message.addHeader("x-Header", newText);
        message.setText(newText);
        message.setSubject(newText);
        message.saveChanges();
      } catch (javax.mail.MessagingException me) {
         log (me.getMessage());
      }
        }
    }

    public LinearProcessorTest(String arg0) throws Exception {
        super(arg0);

        MimeMessageInputStreamSource mmis = null;
        try {
            mmis = new MimeMessageInputStreamSource("test", new SharedByteArrayInputStream((content+sep+body).getBytes()));
        } catch (MessagingException e) {
        }
        mw = new MimeMessageCopyOnWriteProxy(mmis);
        setUp();
  }

    private class DummyMatcherConfig implements MatcherConfig {
        private String condition;

        public DummyMatcherConfig(String config) {
            this.condition = config;
        }
        public String getCondition() {
            return condition;
        }
        
        public MailetContext getMailetContext() {
            return mockContext;
        }

        public String getMatcherName() {
            return "All";
        }
    }
    
    public void testCopyOnWrite() throws IOException, MessagingException {
        t.setSpool(new MockSpoolRepository());
        Matcher recipientIs = new RecipientIs();
        recipientIs.init(new DummyMatcherConfig("[e-mail]"));
        
        Matcher all = new All();
        all.init(new DummyMatcherConfig(""));
        
        Mailet changeBody = new MyMailet();
        Mailet changeBody2 = new MyMailet();

        changeBody.init(mockMailetConfig);
        changeBody2.init(mockMailetConfig);
        
        Mailet dumpSystemErr = new DumpSystemErr();
        changeBody.init(mockMailetConfig);
        
        checkerMailet = new CheckerMailet();
        t.add(recipientIs,changeBody);
        t.add(all,changeBody);
        t.add(all,dumpSystemErr);
        t.add(all,checkerMailet);
        t.closeProcessorLists();

        Collection recipients = new ArrayList();
        recipients.add(new MailAddress("rec1","domain.com"));
        recipients.add(new MailAddress("rec2","domain.com"));
        try {
            MailImpl m = new MailImpl("mail1",new MailAddress("sender","domain.com"),recipients,mw);
            t.service(m);
            ArrayList a = checkerMailet.receivedMails;
            assertEquals(2,a.size());
            MimeMessage m1 = ((Mail) a.get(0)).getMessage();
            MimeMessage m2 = ((Mail) a.get(1)).getMessage();
            assertNotSame(m1,m2);
            assertEquals(m1.getSubject(),"new text 1");
            assertEquals(m2.getSubject(),"new text 2");
            m.dispose();
        } catch (MessagingException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    public void testStateChange() throws IOException, MessagingException {
        t.setSpool(new MockSpoolRepository() {
            public void store(Mail mc) throws MessagingException {
                assertEquals("MYSTATE",mc.getState());
                super.store(mc);
            }
        });
        
        Matcher recipientIs = new RecipientIs();
        recipientIs.init(new DummyMatcherConfig("[e-mail]"));
        
        Matcher all = new All();
        all.init(new DummyMatcherConfig(""));
        
        Mailet dumpSystemErr = new DumpSystemErr();
        
        checkerMailet = new CheckerMailet();
        t.add(recipientIs,dumpSystemErr);
        t.add(all,dumpSystemErr);
        t.add(all,checkerMailet);
        t.closeProcessorLists();

        Collection recipients = new ArrayList();
        recipients.add(new MailAddress("rec1","domain.com"));
        recipients.add(new MailAddress("rec2","domain.com"));
        try {
            MailImpl m = new MailImpl("mail1",new MailAddress("sender","domain.com"),recipients,mw);
            m.setState("MYSTATE");
            t.service(m);
            ArrayList a = checkerMailet.receivedMails;
            assertEquals(2,a.size());
            MimeMessage m1 = ((Mail) a.get(0)).getMessage();
            MimeMessage m2 = ((Mail) a.get(1)).getMessage();
            assertNotSame(m1,m2);
            assertEquals("MYSTATE",((Mail) a.get(0)).getState());
            assertEquals("MYSTATE",((Mail) a.get(1)).getState());
            m.dispose();
        } catch (MessagingException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    public void setUp() throws Exception {
        super.setUp();
        t = new LinearProcessor();
        Logger l = new ConsoleLogger();
        t.enableLogging(l);
        t.initialize();
        
    }
    
    public void tearDown() throws Exception {
        t.dispose();
        super.tearDown();
    }
    
}