From 800d5e6617612571d9066805cd425811d41e976a Mon Sep 17 00:00:00 2001 From: Oleg Zhurakousky Date: Thu, 11 Nov 2010 11:33:02 -0500 Subject: [PATCH] INT-1603 added test for search to Twitter4JTemplateTest --- .../twitter/core/Twitter4jTemplateTests.java | 29 +++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/spring-integration-twitter/src/test/java/org/springframework/integration/twitter/core/Twitter4jTemplateTests.java b/spring-integration-twitter/src/test/java/org/springframework/integration/twitter/core/Twitter4jTemplateTests.java index 7ccf455005..faf9d4094e 100644 --- a/spring-integration-twitter/src/test/java/org/springframework/integration/twitter/core/Twitter4jTemplateTests.java +++ b/spring-integration-twitter/src/test/java/org/springframework/integration/twitter/core/Twitter4jTemplateTests.java @@ -16,6 +16,7 @@ package org.springframework.integration.twitter.core; import static junit.framework.Assert.assertEquals; +import static junit.framework.Assert.assertNotNull; import static junit.framework.Assert.assertTrue; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.times; @@ -23,6 +24,8 @@ import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; import java.lang.reflect.Field; +import java.util.ArrayList; +import java.util.List; import org.junit.Before; import org.junit.Test; @@ -30,6 +33,8 @@ import org.mockito.Mockito; import org.springframework.integration.test.util.TestUtils; import twitter4j.Paging; +import twitter4j.Query; +import twitter4j.QueryResult; import twitter4j.StatusUpdate; import twitter4j.Twitter; import twitter4j.http.AccessToken; @@ -108,4 +113,28 @@ public class Twitter4jTemplateTests { template.updateStatus("writing twitter test"); verify(twitter, times(1)).updateStatus(Mockito.any(StatusUpdate.class)); } + + @Test + public void testSearch() throws Exception{ + // set up test + QueryResult result = mock(QueryResult.class); + List t4jTweets = new ArrayList(); + t4jTweets.add(mock(twitter4j.Tweet.class)); + t4jTweets.add(mock(twitter4j.Tweet.class)); + t4jTweets.add(mock(twitter4j.Tweet.class)); + + when(result.getTweets()).thenReturn(t4jTweets); + + when(twitter.search(Mockito.any(Query.class))).thenReturn(result); + // end setup test + + SearchResults results = template.search("#s2gx"); + List tweets = results.getTweets(); + assertNotNull(tweets); + assertEquals(3, tweets.size()); + assertTrue(tweets.get(0) instanceof Tweet); + assertTrue(tweets.get(1) instanceof Tweet); + assertTrue(tweets.get(2) instanceof Tweet); + + } }