fixing broken Twitter tests

This commit is contained in:
Oleg Zhurakousky
2010-11-18 14:35:44 -05:00
parent 2d9af9b849
commit f549ccbdd9
2 changed files with 6 additions and 12 deletions

View File

@@ -20,10 +20,8 @@ import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.Queue;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.integration.Message;
@@ -57,8 +55,6 @@ abstract class AbstractTwitterMessageSource<T> extends AbstractEndpoint implemen
private volatile long lastPollForTweet;
private final CountDownLatch latch = new CountDownLatch(1);
private final TwitterPollingTask twitterPoller = new TwitterPollingTask();
private volatile MetadataStore metadataStore;
@@ -135,18 +131,15 @@ abstract class AbstractTwitterMessageSource<T> extends AbstractEndpoint implemen
if (tweet == null){
long currentTime = System.currentTimeMillis();
long diff = currentTime - lastPollForTweet;
lastPollForTweet = currentTime;
if (diff < 15000){
try {
long waitTime = 15000-diff;
this.latch.await(waitTime, TimeUnit.MILLISECONDS);
} catch (Exception e) {
//ignore
}
return null;
}
twitterPoller.run();
tweet = this.tweets.poll();
lastPollForTweet = currentTime;
}
tweet = this.tweets.poll();
if (tweet != null) {
this.lastProcessedId = tweet.getId();
this.metadataStore.put(this.metadataKey, String.valueOf(this.lastProcessedId));

View File

@@ -167,6 +167,7 @@ public class DirectMessageReceivingMessageSourceTests {
Queue msg = (Queue) TestUtils.getPropertyValue(source, "tweets");
assertTrue(!CollectionUtils.isEmpty(msg));
assertEquals(1, msg.size());
//Thread.sleep(15000);
Message message = source.receive();
Tweet tweet = (Tweet) message.getPayload();
assertEquals(2000, tweet.getId());