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 generateManyTweetSearches(randomTweetSearches, tweetId, searchId): tweetSearchList = ObjectList() for i in range(0, randomTweetSearches): tweetSearch = TweetSearch(randomTweetSearch(tweetId, searchId)) tweetSearchList.append(tweetSearch) return tweetSearchList
def test_addRepeatedTweetSearch(self): lb = LocalBuffer() lb.addTweetSearch(TweetSearch(randomTweetSearch(1, 1))) self.assertRaises(Exception, lb.addTweetSearch, [TweetSearch(randomTweetSearch(1, 1)),])
def test_addTweetSearch(self): lb = LocalBuffer() lb.addTweetSearch(TweetSearch(randomTweetSearch(1, 1))) self.assertEqual(len(lb.tweetSearchList.getList()), 1)