org.jvnet.mock_javamail.MockStore Maven / Gradle / Ivy

Go to download
package org.jvnet.mock_javamail;

import javax.mail.Folder;
import javax.mail.MessagingException;
import javax.mail.Session;
import javax.mail.Store;
import javax.mail.URLName;

/**
 * {@link Store} backed by {@link Mailbox}.
 *
 * @author Kohsuke Kawaguchi
 */
public class MockStore extends Store {
    private MockFolder folder;
    private String address;

    public MockStore(Session session, URLName urlname) {
        super(session, urlname);
    }

    public void connect() throws MessagingException {
        connect(url.getHost(), url.getPort(), url.getUsername(), url.getPassword());
    }

    protected boolean protocolConnect(String host, int port, String user, String password) throws MessagingException {
        address = user+'@'+host;
        Mailbox mailbox = Mailbox.get(Aliases.getInstance().resolve(address));
        folder = new MockFolder(this, mailbox);
        if(mailbox.isError())
            throw new MessagingException("Simulated error connecting to "+address);
        return true;
    }

    public Folder getDefaultFolder() throws MessagingException {
        return folder;
    }

    public Folder getFolder(String name) throws MessagingException {
        return folder;
    }

    public Folder getFolder(URLName url) throws MessagingException {
        return folder;
    }
}