refacotred PollingMailSource and added SubscribableMailSource for use with IMAP idle

org.springframework.integration.adapter/src/test/java/org/springframework/integration/adapter/mail/PollingMessageSourceTests.java

@@ -0,0 +1,89 @@
+/*
+ * Copyright 2002-2007 the original author or authors.
+ *
+ * Licensed 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.springframework.integration.adapter.mail;
+
+import static org.junit.Assert.*;
+
+import java.util.concurrent.ConcurrentLinkedQueue;
+
+import javax.mail.internet.MimeMessage;
+
+import org.easymock.classextension.EasyMock;
+import org.junit.Test;
+import org.springframework.integration.message.GenericMessage;
+import org.springframework.integration.message.Message;
+
+public class PollingMessageSourceTests {
+
+	@Test
+	public void testPolling(){
+		StubFolderConnection folderConnection = new StubFolderConnection();
+	
+		MimeMessage messageOne = EasyMock.createMock(MimeMessage.class);
+		MimeMessage messageTwo = EasyMock.createMock(MimeMessage.class);
+		MimeMessage messageThree = EasyMock.createMock(MimeMessage.class);
+		MimeMessage messageFour = EasyMock.createMock(MimeMessage.class);
+		
+		folderConnection.messages.add(new javax.mail.Message[]{messageOne});
+		folderConnection.messages.add(new javax.mail.Message[]{messageTwo,messageThree});
+		folderConnection.messages.add(new javax.mail.Message[]{messageFour});
+		
+		PollingMailSource pollingMailSource = new PollingMailSource(folderConnection);
+		pollingMailSource.setConverter(new StubMessageConvertor());
+		
+		
+		assertEquals("Wrong message for number 1", messageOne, pollingMailSource.receive().getPayload());
+		assertEquals("Wrong message for number 2", messageTwo, pollingMailSource.receive().getPayload());
+		assertEquals("Wrong message for number 3", messageThree, pollingMailSource.receive().getPayload());
+		assertEquals("Wrong message for number 4", messageFour, pollingMailSource.receive().getPayload());
+		assertNull("Expected null after exhausting all messages",pollingMailSource.receive());
+	}
+	
+	
+	
+	private static class StubFolderConnection implements FolderConnection {
+
+		ConcurrentLinkedQueue<javax.mail.Message[]> messages = new ConcurrentLinkedQueue<javax.mail.Message[]>();
+
+		
+		
+		public javax.mail.Message[] receive() {
+			return messages.poll();
+		}
+
+		public boolean isRunning() {
+			return false;
+		}
+
+		public void start() {
+
+		}
+
+		public void stop() {
+
+		}
+
+	}
+
+	private static class StubMessageConvertor implements MailMessageConverter {
+
+		@SuppressWarnings("unchecked")
+		public Message create(MimeMessage mailMessage) {
+			return new GenericMessage<MimeMessage>(mailMessage);
+		}
+	}
+
+}

org.springframework.integration.adapter/src/test/java/org/springframework/integration/adapter/mail/SubscribableMailSourceTests.java

@@ -0,0 +1,104 @@
+package org.springframework.integration.adapter.mail;
+
+import static org.junit.Assert.assertEquals;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.ConcurrentLinkedQueue;
+
+import javax.mail.internet.MimeMessage;
+
+import org.easymock.classextension.EasyMock;
+import org.junit.Before;
+import org.junit.Test;
+import org.springframework.core.task.TaskExecutor;
+import org.springframework.integration.message.GenericMessage;
+import org.springframework.integration.message.Message;
+import org.springframework.integration.message.MessageTarget;
+import org.springframework.scheduling.concurrent.ConcurrentTaskExecutor;
+
+/**
+ * 
+ * @author Jonas Partner
+ *
+ */
+public class SubscribableMailSourceTests {
+
+	TaskExecutor executor;
+
+	@Before
+	public void setUp() {
+		executor = new ConcurrentTaskExecutor();
+	}
+
+	@Test
+	public void testReceive() throws Exception {
+		javax.mail.Message message = EasyMock.createMock(MimeMessage.class);
+		StubFolderConnection folderConnection = new StubFolderConnection(
+				message);
+		StubTarget target = new StubTarget();
+
+		SubscribableMailSource mailSource = new SubscribableMailSource(
+				folderConnection, executor);
+		mailSource.subscribe(target);
+		mailSource.setConverter(new StubMessageConvertor());
+
+		mailSource.start();
+		Thread.sleep(1000);
+		mailSource.stop();
+
+		assertEquals("Wrong message count", 1, target.messages.size());
+		assertEquals("Wrong payload", message, target.messages.get(0)
+				.getPayload());
+	}
+
+	private static class StubFolderConnection implements FolderConnection {
+
+		ConcurrentLinkedQueue<javax.mail.Message> messages = new ConcurrentLinkedQueue<javax.mail.Message>();
+
+		public StubFolderConnection(javax.mail.Message message) {
+			messages.add(message);
+		}
+
+		public javax.mail.Message[] receive() {
+			javax.mail.Message msg = messages.poll();
+			if (msg == null) {
+				return new javax.mail.Message[] {};
+			}
+			return new javax.mail.Message[] { msg };
+		}
+
+		public boolean isRunning() {
+			return false;
+		}
+
+		public void start() {
+
+		}
+
+		public void stop() {
+
+		}
+
+	}
+
+	private static class StubTarget implements MessageTarget {
+
+		List<Message<?>> messages = new ArrayList<Message<?>>();
+
+		public boolean send(Message<?> message) {
+			messages.add(message);
+			return true;
+		}
+
+	}
+
+	private static class StubMessageConvertor implements MailMessageConverter {
+
+		@SuppressWarnings("unchecked")
+		public Message create(MimeMessage mailMessage) {
+			return new GenericMessage<MimeMessage>(mailMessage);
+		}
+	}
+
+}