|
17 | 17 | package org.springframework.integration.file.remote; |
18 | 18 |
|
19 | 19 | import static org.assertj.core.api.Assertions.assertThat; |
| 20 | +import static org.assertj.core.api.Assertions.assertThatExceptionOfType; |
20 | 21 | import static org.mockito.ArgumentMatchers.anyString; |
21 | 22 | import static org.mockito.Mockito.mock; |
22 | 23 | import static org.mockito.Mockito.when; |
23 | 24 |
|
24 | 25 | import java.io.IOException; |
25 | 26 | import java.io.InputStream; |
| 27 | +import java.io.UncheckedIOException; |
26 | 28 | import java.util.Collection; |
27 | 29 | import java.util.Comparator; |
28 | 30 | import java.util.List; |
29 | 31 | import java.util.stream.Collectors; |
30 | 32 |
|
31 | | -import org.junit.Test; |
| 33 | +import org.junit.jupiter.api.Test; |
32 | 34 |
|
33 | 35 | import org.springframework.beans.factory.BeanFactory; |
34 | 36 | import org.springframework.integration.file.filters.FileListFilter; |
| 37 | +import org.springframework.integration.file.remote.session.CachingSessionFactory; |
35 | 38 | import org.springframework.integration.file.remote.session.Session; |
| 39 | +import org.springframework.integration.file.remote.session.SessionFactory; |
36 | 40 |
|
37 | 41 | /** |
38 | 42 | * @author Lukas Gemela |
@@ -69,6 +73,32 @@ public void filterOutFilesNotAcceptedByFilter() throws IOException { |
69 | 73 | assertThat(testRemoteFileStreamingMessageSource.doReceive()).isNull(); |
70 | 74 | } |
71 | 75 |
|
| 76 | + @Test |
| 77 | + @SuppressWarnings("unchecked") |
| 78 | + public void sessionReturnedToCacheProperlyOnDoReceive() throws IOException { |
| 79 | + Session<String> session = mock(Session.class); |
| 80 | + when(session.readRaw(anyString())).thenThrow(IOException.class); |
| 81 | + when(session.list("remoteDirectory")).thenReturn(new String[] { "file1" }); |
| 82 | + |
| 83 | + SessionFactory<String> sessionFactory = mock(SessionFactory.class); |
| 84 | + when(sessionFactory.getSession()).thenReturn(session); |
| 85 | + |
| 86 | + CachingSessionFactory<String> cachingSessionFactory = new CachingSessionFactory<>(sessionFactory, 1); |
| 87 | + RemoteFileTemplate<String> remoteFileTemplate = new RemoteFileTemplate<>(cachingSessionFactory); |
| 88 | + |
| 89 | + TestRemoteFileStreamingMessageSource testRemoteFileStreamingMessageSource = |
| 90 | + new TestRemoteFileStreamingMessageSource(remoteFileTemplate, null); |
| 91 | + |
| 92 | + testRemoteFileStreamingMessageSource.setRemoteDirectory("remoteDirectory"); |
| 93 | + testRemoteFileStreamingMessageSource.setBeanFactory(mock(BeanFactory.class)); |
| 94 | + testRemoteFileStreamingMessageSource.start(); |
| 95 | + |
| 96 | + assertThatExceptionOfType(UncheckedIOException.class) |
| 97 | + .isThrownBy(testRemoteFileStreamingMessageSource::doReceive); |
| 98 | + |
| 99 | + assertThat(cachingSessionFactory.getSession()).isNotNull(); |
| 100 | + } |
| 101 | + |
72 | 102 | static class TestRemoteFileStreamingMessageSource extends AbstractRemoteFileStreamingMessageSource<String> { |
73 | 103 |
|
74 | 104 | TestRemoteFileStreamingMessageSource(RemoteFileTemplate<String> template, Comparator<String> comparator) { |
|
0 commit comments