com.zimbra.cs.imap.ImapProxyTest.java Source code

Introduction

Here is the source code for com.zimbra.cs.imap.ImapProxyTest.java

Source

/*
 * ***** BEGIN LICENSE BLOCK *****
 * Zimbra Collaboration Suite Server
 * Copyright (C) 2011, 2012, 2013, 2014, 2016 Synacor, Inc.
 *
 * This program is free software: you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software Foundation,
 * version 2 of the License.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 * You should have received a copy of the GNU General Public License along with this program.
 * If not, see <https://www.gnu.org/licenses/>.
 * ***** END LICENSE BLOCK *****
 */
package com.zimbra.cs.imap;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.util.regex.Pattern;

import junit.framework.Assert;

import org.apache.commons.io.output.ByteArrayOutputStream;
import org.junit.After;
import org.junit.Test;

import com.google.common.base.Charsets;
import com.zimbra.cs.util.MockTcpServer;

/**
 * Unit test for {@link ImapProxy}.
 *
 * @author ysasaki
 */
public final class ImapProxyTest {
    private static final int PORT = 9143;
    private MockTcpServer server;

    @After
    public void tearDown() throws Exception {
        if (server != null) {
            server.destroy();
        }
    }

    @Test(timeout = 30000)
    public void bye() throws Exception {
        server = MockTcpServer.scenario().sendLine("* OK server ready").recvLine() // CAPABILITY
                .sendLine("* CAPABILITY IMAP4rev1 AUTH=X-ZIMBRA")
                .reply(Pattern.compile("(.*) CAPABILITY"), "{0} OK CAPABILITY\r\n").recvLine() // ID
                .sendLine("* ID (\"NAME\" \"Zimbra\")").reply(Pattern.compile("(.*) ID"), "{0} OK ID completed\r\n")
                .recvLine() // AUTHENTICATE
                .sendLine("+ ready for literal")
                .reply(Pattern.compile("(.*) AUTHENTICATE"), "{0} OK AUTHENTICATE\r\n").recvLine() // credential
                .recvLine() // NOOP
                .reply(Pattern.compile("(.*) NOOP"), "{0} OK NOOP\r\n").sendLine("* BYE server closing connection")
                .build().start(PORT);

        MockImapHandler handler = new MockImapHandler();
        ImapProxy proxy = new ImapProxy(new InetSocketAddress(PORT), "test@zimbra.com", "secret", handler);
        proxy.proxy("001", "NOOP");
        try {
            proxy.proxy("002", "NOOP");
            Assert.fail();
        } catch (ImapProxyException expected) {
        }

        // verify BYE was not proxied
        Assert.assertEquals("001 OK NOOP\r\n", handler.output.toString());

        server.shutdown(3000);
        Assert.assertEquals("C01 CAPABILITY\r\n", server.replay());
        String id = server.replay();
        Assert.assertTrue(id,
                id.matches("C02 ID \\(\"name\" \"ZCS\" \"version\" \".*\" \"X-VIA\" \"127\\.0\\.0\\.1\"\\)\r\n"));
        Assert.assertEquals("C03 AUTHENTICATE X-ZIMBRA\r\n", server.replay());
        server.replay(); // auth token
        Assert.assertEquals("001 NOOP\r\n", server.replay());
        Assert.assertEquals(null, server.replay());
    }

    private static final class MockImapHandler extends ImapHandler {

        MockImapHandler() {
            super(new ImapConfig(false));
            output = new ByteArrayOutputStream();
        }

        @Override
        String getRemoteIp() {
            return "127.0.0.1";
        }

        @Override
        void sendLine(String line, boolean flush) throws IOException {
            output.write(line.getBytes(Charsets.UTF_8));
        }

        @Override
        void dropConnection(boolean sendBanner) {
        }

        @Override
        void close() {
        }

        @Override
        void enableInactivityTimer() throws IOException {
        }

        @Override
        void completeAuthentication() throws IOException {
        }

        @Override
        boolean doSTARTTLS(String tag) throws IOException {
            return false;
        }

        @Override
        InetSocketAddress getLocalAddress() {
            return new InetSocketAddress("localhost", 0);
        }

    }

}