def test_crons(self): from payparrot_scripts.crons.cron1 import main as cron1 cron1() from payparrot_scripts.crons.cron2 import main as cron2 cron2() sleep(2) parrot = Parrots.findOne(self.db, {}) twitter = Twitter() twitter.create_client(parrot.oauth_token, parrot.oauth_token_secret) headers, body = twitter.get("http://api.twitter.com/1/statuses/user_timeline.json?user_id=%s&include_rts=false&count=1" % parrot.twitter_info.get('screen_name')) json_twitter_status = json.loads(body) message_start = self.message.text received_message = json_twitter_status[0].get('text') self.assertEqual(self.message.text, received_message[0:len(message_start)]) queue = Queue.get_queue('payments') self.assertEqual(0, queue.count()) from payparrot_scripts.crons.cron3 import main as cron3 cron3() raw_input("Check? ")
def tear_down(db, app=None, queue = False): db.accounts.drop() db.messages.drop() db.accounts_sessions.drop() db.subscriptions.drop() db.sessions.drop() db.parrots.drop() db.notifications.drop() db.payments.drop() if app: app.get('/logout') if queue: sleep(1) for queue_name in ['notifications', 'payments']: queue = Queue.get_queue(queue_name) while queue.count(): for message in queue.get_messages(num_messages=10, visibility_timeout=60): Queue.delete_message(queue_name, message)