INT-1527 MetadataStore refactoring

spring-integration-twitter/src/test/java/org/springframework/integration/twitter/inbound/InboundDirectMessageStatusEndpointTests.java

@@ -13,6 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
+
 package org.springframework.integration.twitter.inbound;
 
 import static junit.framework.Assert.assertEquals;
@@ -28,6 +29,7 @@ import java.util.List;
 import org.junit.Before;
 import org.junit.Test;
 import org.mockito.Mockito;
+
 import org.springframework.integration.Message;
 import org.springframework.integration.channel.QueueChannel;
 import org.springframework.integration.twitter.oauth.OAuthConfiguration;
@@ -41,36 +43,18 @@ import twitter4j.Twitter;
 
 /**
  * @author Oleg Zhurakousky
- *
  */
 public class InboundDirectMessageStatusEndpointTests {
+
 	private DirectMessage firstMessage;
+
 	private DirectMessage secondMessage;
+
 	private Twitter twitter;
 
-	@Test
-	public void testTwitterMockedUpdates() throws Exception{
-		QueueChannel channel = new QueueChannel();
-		
-		InboundDirectMessageEndpoint endpoint = new InboundDirectMessageEndpoint();
-		endpoint.setOutputChannel(channel);
-		ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
-		scheduler.afterPropertiesSet();
-		endpoint.setTaskScheduler(scheduler);
-		
-		endpoint.setConfiguration(this.getTestConfigurationForDirectMessages());
-		endpoint.afterPropertiesSet();
-		endpoint.start();
-		Message message1 = channel.receive(3000);
-		assertNotNull(message1);
-		// should be second message since its timestamp is newer
-		assertEquals(secondMessage.getId(), ((DirectMessage)message1.getPayload()).getId());
-		Message message2 = channel.receive(100);
-		assertNull(message2); // should be null, since 
-	}
-	
+
 	@Before
-	public void prepare(){
+	public void prepare() {
 		twitter = mock(Twitter.class);
 		firstMessage = mock(DirectMessage.class);
 		when(firstMessage.getCreatedAt()).thenReturn(new Date(5555555555L));
@@ -79,27 +63,47 @@ public class InboundDirectMessageStatusEndpointTests {
 		when(secondMessage.getCreatedAt()).thenReturn(new Date(2222222222L));
 		when(secondMessage.getId()).thenReturn(2000);
 	}
-	
+
+
+	@Test
+	public void testTwitterMockedUpdates() throws Exception{
+		QueueChannel channel = new QueueChannel();
+		InboundDirectMessageEndpoint endpoint = new InboundDirectMessageEndpoint();
+		endpoint.setOutputChannel(channel);
+		ThreadPoolTaskScheduler scheduler = new ThreadPoolTaskScheduler();
+		scheduler.afterPropertiesSet();
+		endpoint.setTaskScheduler(scheduler);
+		endpoint.setConfiguration(this.getTestConfigurationForDirectMessages());
+		endpoint.setBeanName("twitterEndpoint");
+		endpoint.afterPropertiesSet();
+		endpoint.start();
+		Message<?> message1 = channel.receive(3000);
+		assertNotNull(message1);
+		// should be second message since its timestamp is newer
+		assertEquals(secondMessage.getId(), ((DirectMessage)message1.getPayload()).getId());
+		Message<?> message2 = channel.receive(100);
+		assertNull(message2); // should be null, since 
+	}
+
+
 	@SuppressWarnings("unchecked")
 	private OAuthConfiguration getTestConfigurationForDirectMessages() throws Exception{
 		OAuthConfiguration configuration = mock(OAuthConfiguration.class);
 		RateLimitStatus rateLimitStatus = mock(RateLimitStatus.class);
 		when(twitter.getRateLimitStatus()).thenReturn(rateLimitStatus);
 		when(configuration.getTwitter()).thenReturn(twitter);
-		
 		when(rateLimitStatus.getSecondsUntilReset()).thenReturn(2464);
 		when(rateLimitStatus.getRemainingHits()).thenReturn(250);
-		
+
 		ResponseList<DirectMessage> responses = mock(ResponseList.class);
 		List<DirectMessage> testMessages = new ArrayList<DirectMessage>();
-		
 		testMessages.add(firstMessage);
 		testMessages.add(secondMessage);
-		
+
 		when(responses.iterator()).thenReturn(testMessages.iterator());
 		when(twitter.getDirectMessages()).thenReturn(responses);
-		
 		when(twitter.getDirectMessages(Mockito.any(Paging.class))).thenReturn(responses);
 		return configuration;
 	}
+
 }