FtpSource refactoring: Message<File> -> Message<List<File>>

org.springframework.integration.adapter/src/test/java/org/springframework/integration/adapter/ftp/FtpSourceTests.java

@@ -0,0 +1,120 @@
+package org.springframework.integration.adapter.ftp;
+
+import static org.easymock.classextension.EasyMock.*;
+import static org.junit.Assert.*;
+
+import java.io.File;
+import java.io.OutputStream;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Calendar;
+import java.util.Collections;
+import java.util.List;
+
+import org.apache.commons.net.ftp.FTPClient;
+import org.apache.commons.net.ftp.FTPFile;
+import org.junit.Before;
+import org.junit.Test;
+import org.springframework.integration.message.GenericMessage;
+import org.springframework.integration.message.Message;
+import org.springframework.integration.message.MessageCreator;
+
+@SuppressWarnings("unchecked")
+public class FtpSourceTests {
+
+	private MessageCreator<List<File>, List<File>> messageCreator = createMock(MessageCreator.class);
+
+	private FTPClient ftpClient = createMock(FTPClient.class);
+
+	private FTPFile ftpFile = createMock(FTPFile.class);
+
+	private Object[] globalMocks = new Object[] { messageCreator, ftpClient, ftpFile };
+
+	private static final String HOST = "testHost";
+
+	private static final String USER = "testUser";
+
+	private static final String PASS = "testPass";
+
+	private FtpSource ftpSource;
+
+	@Before
+	public void initializeFtpSource() {
+		ftpSource = new FtpSource(messageCreator, ftpClient);
+		ftpSource.setHost(HOST);
+		ftpSource.setUsername(USER);
+		ftpSource.setPassword(PASS);
+	}
+
+	@Before
+	public void clearState() {
+		reset(globalMocks);
+	}
+
+	@Test
+	public void retrieveSingleFile() throws Exception {
+		// connect client and get file
+		expect(ftpClient.isConnected()).andReturn(false);
+		expect(ftpClient.isConnected()).andReturn(true).anyTimes();
+
+		ftpClient.connect(HOST, 21);
+		expect(ftpClient.login(USER, PASS)).andReturn(true);
+		expect(ftpClient.setFileType(anyInt())).andReturn(true);
+		expect(ftpClient.printWorkingDirectory()).andReturn("/");
+		expect(ftpClient.listFiles()).andReturn(mockedFTPFilesNamed(Calendar.getInstance(), "test"));
+		expect(ftpClient.retrieveFile(eq("test"), isA(OutputStream.class))).andReturn(true);
+		// create message
+		expect(messageCreator.createMessage(isA(List.class))).andReturn(
+				new GenericMessage(Arrays.asList(new File("test"))));
+		ftpClient.disconnect();
+		replay(globalMocks);
+		ftpSource.onSend(ftpSource.receive());
+		verify(globalMocks);
+	}
+
+	private FTPFile[] mockedFTPFilesNamed(Calendar timestamp, String... names) {
+		List<FTPFile> files = new ArrayList<FTPFile>();
+		for (String name : names) {
+			FTPFile ftpFile = createMock(FTPFile.class);
+			expect(ftpFile.getName()).andReturn(name).anyTimes();
+			expect(ftpFile.getTimestamp()).andReturn(timestamp).anyTimes();
+			expect(ftpFile.getSize()).andReturn(100l).anyTimes();
+			files.add(ftpFile);
+			replay(ftpFile);
+		}
+		return files.toArray(new FTPFile[] {});
+	}
+
+	@Test
+	public void retrieveMultipleFiles() throws Exception {
+		// connect client
+		expect(ftpClient.isConnected()).andReturn(false);
+		expect(ftpClient.isConnected()).andReturn(true).anyTimes();
+
+		ftpClient.connect(HOST, 21);
+		expect(ftpClient.login(USER, PASS)).andReturn(true);
+		expect(ftpClient.setFileType(anyInt())).andReturn(true);
+		expect(ftpClient.printWorkingDirectory()).andReturn("/");
+
+		// get files
+		Calendar timestamp = Calendar.getInstance();
+		expect(ftpClient.listFiles()).andReturn(mockedFTPFilesNamed(timestamp, "test", "test2")).anyTimes();
+
+		expect(ftpClient.retrieveFile(eq("test"), isA(OutputStream.class))).andReturn(true);
+		expect(ftpClient.retrieveFile(eq("test2"), isA(OutputStream.class))).andReturn(true);
+		// create message
+		List<File> files = Arrays.asList(new File("test"), new File("test2"));
+		expect(messageCreator.createMessage(isA(List.class))).andReturn(new GenericMessage(files));
+		ftpClient.disconnect();
+
+		replay(globalMocks);
+		Message receivedFiles = ftpSource.receive();
+		ftpSource.onSend(receivedFiles);
+		Message<List<File>> secondReceived = ftpSource.receive();
+		verify(globalMocks);
+		assertEquals(files, receivedFiles.getPayload());
+		assertNull(secondReceived);
+	}
+
+
+}