davmail/src/test/davmail/smtp/TestSmtp.java

117 lines
4.2 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.smtp;
import davmail.AbstractDavMailTestCase;
import davmail.DavGateway;
import davmail.Settings;
import davmail.exchange.ExchangeSessionFactory;
import org.apache.commons.codec.binary.Base64;
import javax.mail.MessagingException;
import javax.mail.Session;
import javax.mail.internet.MimeMessage;
import java.io.*;
import java.net.Socket;
/**
* Test smtp send message.
*/
public class TestSmtp extends AbstractDavMailTestCase {
static Socket clientSocket;
static BufferedWriter socketWriter;
static BufferedReader socketReader;
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.smtpPort"));
socketWriter = new BufferedWriter(new OutputStreamWriter(clientSocket.getOutputStream()));
socketReader = new BufferedReader(new InputStreamReader(clientSocket.getInputStream()));
}
if (session == null) {
session = ExchangeSessionFactory.getInstance(Settings.getProperty("davmail.username"), Settings.getProperty("davmail.password"));
}
}
public void testBanner() throws IOException {
String banner = socketReader.readLine();
assertNotNull(banner);
}
public void testLogin() throws IOException {
String credentials = (char) 0+Settings.getProperty("davmail.username")+ (char) 0 +Settings.getProperty("davmail.password");
writeLine("AUTH PLAIN " + new String(new Base64().encode(credentials.getBytes())));
assertEquals("235 OK Authenticated", socketReader.readLine());
}
public void testCreateMessage() throws IOException, MessagingException {
MimeMessage mimeMessage = new MimeMessage((Session) null);
mimeMessage.addHeader("To", Settings.getProperty("davmail.to"));
mimeMessage.setText("Test message");
mimeMessage.setSubject("Test subject");
ByteArrayOutputStream baos = new ByteArrayOutputStream();
mimeMessage.writeTo(baos);
byte[] content = baos.toByteArray();
writeLine("MAIL FROM:"+session.getEmail());
readLine();
writeLine("RCPT TO:"+Settings.getProperty("davmail.to"));
readLine();
writeLine("DATA");
assertEquals("354 Start mail input; end with <CRLF>.<CRLF>", readLine());
writeLine(new String(content));
writeLine(".");
assertEquals("250 Queued mail for delivery", readLine());
}
public void testQuit() throws IOException {
writeLine("QUIT");
assertEquals("221 Closing connection", readLine());
}
}