def test_update_new_result(self): dao_twitter_counter = TwitterCounter(self.client, self.database_name) dao_twitter_counter.save_new_twitter_counter(int(time.time()), int(time.time())+1000, {"PP": 30, "PSC": -50}, 12345, {"PP": 10, "PSC": -20}) self.assertEqual(dao_twitter_counter.get_size(), 1)
def test_timestamp_query(self): dao_twitter_counter = TwitterCounter(self.client, self.database_name) dao_twitter_counter.save_new_twitter_counter(int(time.time()), int(time.time())+1000, {"PP": 30, "PSC": -50}, 12345, {"PP": 10, "PSC": -20}) time.sleep(1) start_query = int(time.time()) dao_twitter_counter.save_new_twitter_counter(int(time.time()), int(time.time())+1000, {"PP": 30, "PSC": -50}, 12345, {"PP": 10, "PSC": -20}) dao_twitter_counter.save_new_twitter_counter(int(time.time()), int(time.time())+1000, {"PP": 30, "PSC": -50}, 12345, {"PP": 10, "PSC": -20}) dao_twitter_counter.save_new_twitter_counter(int(time.time()), int(time.time())+1000, {"PP": 30, "PSC": -50}, 12345, {"PP": 10, "PSC": -20}) dao_twitter_counter.save_new_twitter_counter(int(time.time()), int(time.time())+1000, {"PP": 30, "PSC": -50}, 12345, {"PP": 10, "PSC": -20}) end_query = int(time.time()) time.sleep(1) dao_twitter_counter.save_new_twitter_counter(int(time.time()), int(time.time())+1000, {"PP": 30, "PSC": -50}, 12345, {"PP": 10, "PSC": -20}) result_size = dao_twitter_counter.get_twitter_counter_by_dates(start_query, end_query).count() self.assertEqual(dao_twitter_counter.get_size()-2, result_size)