def test_sendData(self): userList = generateOneUser(userId = 1) tweetList = generateManyTweets(randomInteger(100), userId = 1) tweetStreamingList = ObjectList() tweetSearchList = ObjectList() for tweet in tweetList.getList(): tweetStreaming = TweetStreaming(randomTweetStreaming(tweet.id, 1)) tweetStreamingList.append(tweetStreaming) tweetSearch = TweetSearch(randomTweetSearch(tweet.id, 1)) tweetSearchList.append(tweetSearch) self.wsDb.sendData(tweetList, userList, tweetStreamingList, tweetSearchList)
def createData(base): status = getOneStatusTS() randomTweets = base + randomInteger(99) + 1 tweetList = ObjectList() userList = ObjectList() streamingList = ObjectList() searchList = ObjectList() for i in range(base, randomTweets): status["id"] = i status["user"]["id"] = i ps = ParserStatus(status) tweet = Tweet(ps.getTweet()) user = User(ps.getUser()) tweetList.append(tweet) userList.append(user) streamingList.append(TweetStreaming(randomTweetStreaming(i, 1))) searchList.append(TweetSearch(randomTweetSearch(i, 1))) return tweetList, userList, streamingList, searchList
def generateManyTweetStreamings(randomTweetStreamings, tweetId, streamingId): tweetStreamingList = ObjectList() for i in range(0, randomTweetStreamings): tweetStreaming = TweetStreaming(randomTweetStreaming(tweetId, streamingId)) tweetStreamingList.append(tweetStreaming) return tweetStreamingList
def generateOneTweetStreaming(tweetId, streamingId): tweetStreaming = TweetStreaming(randomTweetStreaming(tweetId, streamingId)) tweetStreamingList = ObjectList() tweetStreamingList.append(tweetStreaming) return tweetStreamingList
def test_addRepeatedTweetStreaming(self): lb = LocalBuffer() lb.addTweetStreaming(TweetStreaming(randomTweetStreaming(1, 1))) self.assertRaises(Exception, lb.addTweetStreaming, [TweetStreaming(randomTweetStreaming(1, 1)),])
def test_addTweetStreaming(self): lb = LocalBuffer() lb.addTweetStreaming(TweetStreaming(randomTweetStreaming(1, 1))) self.assertEqual(len(lb.tweetStreamingList.getList()), 1)