265 lines
11 KiB
Java
265 lines
11 KiB
Java
/*
|
|
* DavMail POP/IMAP/SMTP/CalDav/LDAP Exchange Gateway
|
|
* Copyright (C) 2010 Mickael Guessant
|
|
*
|
|
* 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; either version 2
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* 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, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*/
|
|
package davmail.imap;
|
|
|
|
import davmail.AbstractDavMailTestCase;
|
|
import davmail.DavGateway;
|
|
import davmail.Settings;
|
|
|
|
import javax.mail.MessagingException;
|
|
import javax.mail.Session;
|
|
import javax.mail.internet.MimeMessage;
|
|
import java.io.*;
|
|
import java.net.Socket;
|
|
|
|
/**
|
|
* IMAP tests, an instance of DavMail Gateway must be available
|
|
*/
|
|
public class TestImap extends AbstractDavMailTestCase {
|
|
static Socket clientSocket;
|
|
static BufferedWriter socketWriter;
|
|
static BufferedReader socketReader;
|
|
|
|
static String messageUid = null;
|
|
|
|
protected void write(String line) throws IOException {
|
|
socketWriter.write(line);
|
|
socketWriter.flush();
|
|
}
|
|
|
|
protected void writeLine(String line) throws IOException {
|
|
socketWriter.write(line);
|
|
socketWriter.newLine();
|
|
socketWriter.flush();
|
|
}
|
|
|
|
protected String readLine() throws IOException {
|
|
return socketReader.readLine();
|
|
}
|
|
|
|
protected String readFullAnswer(String prefix) throws IOException {
|
|
String line = socketReader.readLine();
|
|
while (!line.startsWith(prefix)) {
|
|
line = socketReader.readLine();
|
|
}
|
|
return line;
|
|
}
|
|
|
|
@Override
|
|
public void setUp() throws IOException {
|
|
super.setUp();
|
|
if (clientSocket == null) {
|
|
// start gateway
|
|
DavGateway.start();
|
|
clientSocket = new Socket("localhost", Settings.getIntProperty("davmail.imapPort"));
|
|
socketWriter = new BufferedWriter(new OutputStreamWriter(clientSocket.getOutputStream()));
|
|
socketReader = new BufferedReader(new InputStreamReader(clientSocket.getInputStream()));
|
|
}
|
|
}
|
|
|
|
public void testBanner() throws IOException {
|
|
String banner = socketReader.readLine();
|
|
assertNotNull(banner);
|
|
}
|
|
|
|
public void testLogin() throws IOException {
|
|
writeLine(". LOGIN " + Settings.getProperty("davmail.username").replaceAll("\\\\", "\\\\\\\\") + ' ' + Settings.getProperty("davmail.password"));
|
|
assertEquals(". OK Authenticated", socketReader.readLine());
|
|
}
|
|
|
|
public void testSelectInbox() throws IOException {
|
|
writeLine(". SELECT INBOX");
|
|
assertEquals(". OK [READ-WRITE] SELECT completed", readFullAnswer("."));
|
|
}
|
|
|
|
public void testFetchFlags() throws IOException {
|
|
writeLine(". UID FETCH 1:* (FLAGS)");
|
|
assertEquals(". OK UID FETCH completed", readFullAnswer("."));
|
|
}
|
|
|
|
public void testStoreDelete() throws IOException {
|
|
writeLine(". UID STORE 10 +FLAGS (\\Deleted)");
|
|
readFullAnswer(".");
|
|
}
|
|
|
|
public void testUidSearchDeleted() throws IOException {
|
|
writeLine(". UID SEARCH UNDELETED");
|
|
assertEquals(". OK SEARCH completed", readFullAnswer("."));
|
|
}
|
|
|
|
public void testUidSearchUndeleted() throws IOException {
|
|
writeLine(". UID SEARCH DELETED");
|
|
assertEquals(". OK SEARCH completed", readFullAnswer("."));
|
|
}
|
|
|
|
public void testStoreUndelete() throws IOException {
|
|
writeLine(". UID STORE 10 -FLAGS (\\Deleted)");
|
|
readFullAnswer(".");
|
|
}
|
|
|
|
public void testCreateFolder() throws IOException {
|
|
writeLine(". DELETE testfolder");
|
|
readFullAnswer(".");
|
|
writeLine(". CREATE testfolder");
|
|
assertEquals(". OK folder created", readFullAnswer("."));
|
|
}
|
|
|
|
public void testSelectFolder() throws IOException {
|
|
writeLine(". SELECT testfolder");
|
|
assertEquals(". OK [READ-WRITE] SELECT completed", readFullAnswer("."));
|
|
}
|
|
|
|
public void testCreateMessage() throws IOException, MessagingException {
|
|
|
|
MimeMessage mimeMessage = new MimeMessage((Session) null);
|
|
mimeMessage.addHeader("To", "test@test.local");
|
|
mimeMessage.setText("Test message");
|
|
mimeMessage.setSubject("Test subject");
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
mimeMessage.writeTo(baos);
|
|
byte[] content = baos.toByteArray();
|
|
writeLine(". APPEND testfolder (\\Draft) {" + content.length + '}');
|
|
assertEquals("+ send literal data", readLine());
|
|
writeLine(new String(content));
|
|
assertEquals(". OK APPEND completed", readFullAnswer("."));
|
|
writeLine(". NOOP");
|
|
assertEquals(". OK NOOP completed", readFullAnswer("."));
|
|
|
|
// fetch message uid
|
|
writeLine(". UID FETCH 1:* (FLAGS)");
|
|
String messageLine = readLine();
|
|
int uidIndex = messageLine.indexOf("UID ") + 4;
|
|
messageUid = messageLine.substring(uidIndex, messageLine.indexOf(' ', uidIndex));
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
assertNotNull(messageUid);
|
|
}
|
|
|
|
public void testUidStoreDeletedFlag() throws IOException {
|
|
|
|
// test deleted flag
|
|
writeLine(". UID STORE "+messageUid+" +FLAGS (\\Deleted)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (\\Seen \\Deleted \\Draft))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
|
|
// remove deleted flag
|
|
writeLine(". UID STORE "+messageUid+" -FLAGS (\\Deleted)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (\\Seen \\Draft))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
|
|
}
|
|
|
|
public void testUidRemoveSeenFlag() throws IOException {
|
|
// remove seen flag
|
|
writeLine(". UID STORE "+messageUid+" FLAGS (\\Draft)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (\\Draft))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
}
|
|
|
|
public void testUidStoreForwardedFlag() throws IOException {
|
|
// add forwarded flag
|
|
writeLine(". UID STORE "+messageUid+" +FLAGS ($Forwarded)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (\\Draft $Forwarded))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
|
|
// remove forwarded flag
|
|
writeLine(". UID STORE "+messageUid+" -FLAGS ($Forwarded)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (\\Draft))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
}
|
|
|
|
public void testUidStoreAnsweredFlag() throws IOException {
|
|
// add answered flag
|
|
writeLine(". UID STORE "+messageUid+" +FLAGS (\\Answered)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (\\Draft \\Answered))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
|
|
// remove answered flag
|
|
writeLine(". UID STORE "+messageUid+" -FLAGS (\\Answered)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (\\Draft))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
}
|
|
|
|
public void testUidStoreJunkFlag() throws IOException {
|
|
// add Junk flag
|
|
writeLine(". UID STORE "+messageUid+" +FLAGS (Junk)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (Junk \\Draft))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
|
|
// remove Junk flag
|
|
writeLine(". UID STORE "+messageUid+" -FLAGS (Junk)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (\\Draft))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
}
|
|
|
|
public void testUidStoreSeenFlag() throws IOException {
|
|
// add Junk flag
|
|
writeLine(". UID STORE "+messageUid+" +FLAGS (\\Seen)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (\\Seen \\Draft))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
|
|
// remove Junk flag
|
|
writeLine(". UID STORE "+messageUid+" -FLAGS (\\Seen)");
|
|
assertEquals(". OK STORE completed",readFullAnswer("."));
|
|
writeLine(". UID FETCH "+messageUid+" (FLAGS)");
|
|
assertEquals("* 1 FETCH (UID "+messageUid+" FLAGS (\\Draft))", readLine());
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
}
|
|
|
|
public void testPartialFetch() throws IOException {
|
|
writeLine(". UID FETCH "+messageUid+" (BODY.PEEK[1.MIME])");
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
}
|
|
|
|
public void testFetchInternalDate() throws IOException {
|
|
writeLine(". UID FETCH "+messageUid+" (INTERNALDATE)");
|
|
assertEquals(". OK UID FETCH completed",readFullAnswer("."));
|
|
}
|
|
|
|
|
|
public void testDeleteFolder() throws IOException {
|
|
writeLine(". DELETE testfolder");
|
|
assertEquals(". OK folder deleted",readFullAnswer("."));
|
|
}
|
|
|
|
public void testLogout() throws IOException {
|
|
writeLine(". LOGOUT");
|
|
assertEquals("* BYE Closing connection", socketReader.readLine());
|
|
}
|
|
}
|