コード例 #1
0
ファイル: test_view.py プロジェクト: jrgrafton/tweet-debate
    def test_view_tasks_twitter_stream(self):
        if os.getenv('TWITTER_TESTS') == "FALSE":
            return # Modify in runtests.sh

        load_fixture('tests/states.json', kind={'State': State})
        load_fixture('tests/questions.json', 
                        kind={'Question': Question,'State': State})

        # Start new thread with Twitter Stream listener
        stream_thread = Thread(target=self._start_twitter_stream)
        stream_thread.daemon = True
        stream_thread.start()

        # Delete all previous tweets before proceeding
        twitter_api = TwitterAPI()
        twitter_api.delete_all_tweets()

        reponse = \
            self.app.get('/tasks/twitter_post_status' \
                '?question_cadence_minutes=1&post_to_twitter=True')
        current_question = Question.get_current_question()

        # Post reply
        twitter_status_id = twitter_api.get_last_tweet().id
        twitter_api.update_status("#yes for #WA #sway10", twitter_status_id)

        # Wait for stream to be updated
        time.sleep(2)

        # Test to see reply has been registered
        ndb.get_context().clear_cache()
        users = User.get_all()
        assert len(users) == 1
        users[0].sway_points = User.get_starting_sway_points() - 10
        users[0].votes[0].vote_text == "#yes for #WA"

        current_question = Question.get_current_question()
        assert len(current_question.state_scores) == 1
        assert current_question.state_scores[0].state_abbreviation == "WA"
        assert current_question.state_scores[0].party_score_votes[0] == 1
        assert current_question.state_scores[0].party_score_votes[1] == 0
        assert current_question.state_scores[0].party_score_sway[0] == 10
        assert current_question.state_scores[0].party_score_sway[1] == 0
コード例 #2
0
ファイル: tasks.py プロジェクト: jrgrafton/tweet-debate
def twitter_post_status():
    """ Potentially post a new status to Twitter
    """
    # Can be overriden in GET request
    question_cadence_minutes = \
        int(request.args.get('question_cadence_minutes', 60 * 6))
    post_to_twitter = request.args.get('post_to_twitter', False)

    next_question = Question.get_next_question()
    if next_question is not None \
            and __is_time_for_new_question(question_cadence_minutes):
        current_question = Question.get_current_question()

        if post_to_twitter != False:
            try:
                twitter_api = TwitterAPI()
                status = twitter_api.update_status(next_question.question_text)
                next_question.twitterid = str(status.id)
                next_question.put()
            except TweepError as e:
                pass  #TODO: do something - message to monitoring?

        if current_question is not None:
            # Update overall state scores
            State.update_state_scores_for_completed_question(current_question)

            # Update overall question scores
            Question.tally_college_and_vote_scores(current_question)

            # Update end of question user sway scores
            users = User.get_all()
            for user in users:
                __attribute_sway_points_for_user(current_question, user)
            current_question.end_time = datetime.datetime.now()
            current_question.put()

        next_question.start_time = datetime.datetime.now()
        next_question.put()

        return 'Posted new status [%s]' % next_question.question_text, 200

    return 'New question not ready', 200
コード例 #3
0
ファイル: tasks.py プロジェクト: jrgrafton/tweet-debate
def twitter_post_status():
    """ Potentially post a new status to Twitter
    """
    # Can be overriden in GET request
    question_cadence_minutes = \
        int(request.args.get('question_cadence_minutes', 60 * 6))
    post_to_twitter = request.args.get('post_to_twitter', False)
    
    next_question = Question.get_next_question()
    if next_question is not None \
            and __is_time_for_new_question(question_cadence_minutes):
        current_question = Question.get_current_question()
        
        if post_to_twitter != False:
            try:
                twitter_api = TwitterAPI()
                status = twitter_api.update_status(next_question.question_text)
                next_question.twitterid = str(status.id)
                next_question.put()
            except TweepError as e:
                pass  #TODO: do something - message to monitoring?

        if current_question is not None:
            # Update overall state scores
            State.update_state_scores_for_completed_question(current_question)
            
            # Update overall question scores
            Question.tally_college_and_vote_scores(current_question)

            # Update end of question user sway scores
            users = User.get_all()
            for user in users:
                __attribute_sway_points_for_user(current_question, user)
            current_question.end_time = datetime.datetime.now()
            current_question.put()
        
        next_question.start_time = datetime.datetime.now()
        next_question.put()

        return 'Posted new status [%s]' % next_question.question_text, 200

    return 'New question not ready', 200
コード例 #4
0
ファイル: test_view.py プロジェクト: jrgrafton/tweet-debate
    def test_view_tasks_twitter_stream(self):
        if os.getenv("TWITTER_TESTS") == "FALSE":
            return  # Modify in runtests.sh

        load_fixture("tests/states.json", kind={"State": State})
        load_fixture("tests/questions.json", kind={"Question": Question, "State": State})

        # Start new thread with Twitter Stream listener
        stream_thread = Thread(target=self._start_twitter_stream)
        stream_thread.daemon = True
        stream_thread.start()

        # Delete all previous tweets before proceeding
        twitter_api = TwitterAPI()
        twitter_api.delete_all_tweets()

        reponse = self.app.get("/tasks/twitter_post_status" "?question_cadence_minutes=1&post_to_twitter=True")
        current_question = Question.get_current_question()

        # Post reply
        twitter_status_id = twitter_api.get_last_tweet().id
        twitter_api.update_status("#yes for #WA #sway10", twitter_status_id)

        # Wait for stream to be updated
        time.sleep(2)

        # Test to see reply has been registered
        ndb.get_context().clear_cache()
        users = User.get_all()
        assert len(users) == 1
        users[0].sway_points = User.get_starting_sway_points() - 10
        users[0].votes[0].vote_text == "#yes for #WA"

        current_question = Question.get_current_question()
        assert len(current_question.state_scores) == 1
        assert current_question.state_scores[0].state_abbreviation == "WA"
        assert current_question.state_scores[0].party_score_votes[0] == 1
        assert current_question.state_scores[0].party_score_votes[1] == 0
        assert current_question.state_scores[0].party_score_sway[0] == 10
        assert current_question.state_scores[0].party_score_sway[1] == 0
コード例 #5
0
ファイル: test_view.py プロジェクト: jrgrafton/tweet-debate
    def test_view_tasks_twitter_stream_listener_on_data(self):
        load_fixture('tests/states.json', kind={'State': State})
        load_fixture('tests/questions_no_votes.json', 
                        kind={'Question': Question,'State': State})

        twitter_stream = open(os.path.join(os.path.dirname(__file__),
                                          'twitter_stream.json'))
        twitter_stream = json.load(twitter_stream)
        twitter_stream_listener = TwitterStreamListener()

        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_reply_old"]))
        users = User.get_all()
        assert len(users) == 0

        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_reply_old"]))
        users = User.get_all()
        assert len(users) == 0
        
        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_reply_current_invalid"]))
        users = User.get_all()
        assert len(users) == 0
        
        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_post"]))
        users = User.get_all()
        assert len(users) == 0
        
        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_mention"]))
        users = User.get_all()
        assert len(users) == 0

        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_favourite"]))
        users = User.get_all()
        assert len(users) == 0

        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_favourite_reply"]))
        users = User.get_all()
        assert len(users) == 0

        current_question = Question.get_current_question()
        assert len(current_question.state_scores) == 0

        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_reply_current_valid_user1"]))
        users = User.get_all()
        assert len(users) == 1
        assert "https" in users[0].profile_image_url
        current_question = Question.get_current_question()
        assert len(current_question.state_scores) == 1
        assert current_question.state_scores[0].state_abbreviation == "CA"
        assert current_question.state_scores[0].party_score_votes[0] == 0
        assert current_question.state_scores[0].party_score_votes[1] == 1
        assert current_question.state_scores[0].party_score_sway[0] == 0
        assert current_question.state_scores[0].party_score_sway[1] == \
            User.get_starting_sway_points()
        assert users[0].sway_points == 0

        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_reply_current_valid_user1"]))
        users = User.get_all()
        assert len(users) == 1
        assert len(users[0].votes) == 1
        assert users[0].sway_points == 0

        current_question = Question.get_current_question()
        assert len(current_question.state_scores) == 1
        assert current_question.state_scores[0].state_abbreviation == "CA"
        assert current_question.state_scores[0].party_score_votes[0] == 0
        assert current_question.state_scores[0].party_score_votes[1] == 1
        assert current_question.state_scores[0].party_score_sway[0] == 0
        assert current_question.state_scores[0].party_score_sway[1] == \
            User.get_starting_sway_points()
        assert users[0].sway_points == 0

        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_reply_current_valid_user2"]))
        users = User.get_all()
        assert len(users) == 2
        assert "https" in users[1].profile_image_url
        current_question = Question.get_current_question()
        assert len(current_question.state_scores) == 2
        assert current_question.state_scores[1].state_abbreviation == "WA"
        assert current_question.state_scores[1].party_score_votes[0] == 1
        assert current_question.state_scores[1].party_score_votes[1] == 0
        assert current_question.state_scores[1].party_score_sway[0] == 20
        assert current_question.state_scores[1].party_score_sway[1] == 0
        assert users[1].sway_points == User.get_starting_sway_points() - 20

        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_reply_current_valid_user2"]))
        users = User.get_all()
        assert len(users) == 2
        assert current_question.state_scores[1].state_abbreviation == "WA"
        assert current_question.state_scores[1].party_score_votes[0] == 1
        assert current_question.state_scores[1].party_score_votes[1] == 0
        assert current_question.state_scores[1].party_score_sway[0] == 20
        assert current_question.state_scores[1].party_score_sway[1] == 0
        assert users[1].sway_points == User.get_starting_sway_points() - 20

        twitter_stream_listener.on_data(
            json.dumps(twitter_stream["twitter_retweet"]))
        users = User.get_all()
        assert len(users) == 3
        assert users[2].sway_points == User.get_starting_sway_points() + \
                                       sway_points_backend["rewteet_poll"]
        assert "https" in users[2].profile_image_url
コード例 #6
0
ファイル: test_view.py プロジェクト: jrgrafton/tweet-debate
    def test_view_tasks_twitter_stream_listener_add_vote_for_screenname(self):
        load_fixture('tests/states.json', kind={'State': State})
        load_fixture('tests/questions.json', 
                        kind={'Question': Question,'State': State})
        twitter_stream_listener = TwitterStreamListener()
        current_question = Question.get_current_question()
        
        result = twitter_stream_listener.\
            add_vote_for_screenname(current_question,
                                    "Vote text #yes",
                                    "123",
                                    None, # Invalid state
                                    1,
                                    0,
                                    "jrgrafton",
                                    "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg")
        assert result == False

        result = twitter_stream_listener.\
            add_vote_for_screenname(current_question,
                                    "Vote text #CA",
                                    "123",
                                    "CA",
                                    None, # Invalid party
                                    0,
                                    "jrgrafton",
                                    "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg")
        assert result == False

        # First valid user
        result = twitter_stream_listener.\
            add_vote_for_screenname(current_question,
                                    "Vote text #CA #yes",
                                    "123",
                                    "CA",
                                    0,
                                    100,
                                    "jrgrafton",
                                    "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg")
        assert result == True
        user = User.query_by_userid("jrgrafton").get()
        assert user != None
        assert len(user.votes) == 1
        assert user.votes[0].question.get().key == current_question.key
        assert user.votes[0].vote_text == "Vote text #CA #yes"
        assert user.votes[0].replyid == "123"
        assert user.votes[0].state_abbreviation == "CA"
        assert user.votes[0].party == 0
        assert user.votes[0].sway_points == 100

        # Second vote for user on same question
        result = twitter_stream_listener.\
            add_vote_for_screenname(current_question,
                                    "Vote text #WA #no",
                                    "123",
                                    "WA",
                                    1,
                                    0,
                                    "jrgrafton",
                                    "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg")
        assert result == False

        # Second valid user
        result = twitter_stream_listener.\
            add_vote_for_screenname(current_question,
                                    "Vote text for user 2 #WA #no",
                                    "123",
                                    "WA",
                                    1,
                                    50,
                                    "jrgrafton_test",
                                    "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg")
        assert result == True
        users = User.get_all()
        assert len(users) == 2
        user = users[1]
        assert len(user.votes) == 1
        assert user.votes[0].question.get().key == current_question.key
        assert user.votes[0].vote_text == "Vote text for user 2 #WA #no"
        assert user.votes[0].replyid == "123"
        assert user.votes[0].state_abbreviation == "WA"
        assert user.votes[0].party == 1
        assert user.votes[0].sway_points == 50

        # Go to new question and test valid votes
        reponse = self.app.get('/tasks/twitter_post_status' \
                               '?question_cadence_minutes=1')
        current_question = Question.get_current_question()

        # Second vote for new question should work
        result = twitter_stream_listener.\
            add_vote_for_screenname(current_question,
                                    "Vote second question text #WA #no",
                                    "124",
                                    "WA",
                                    1,
                                    10,
                                    "jrgrafton",
                                    "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg")
        assert result == True
        user = User.query_by_userid("jrgrafton").get()
        assert user != None
        assert len(user.votes) == 2
        assert user.votes[1].vote_text == "Vote second question text #WA #no"
        assert user.votes[1].question.get().key == current_question.key
        assert user.votes[1].replyid == "124"
        assert user.votes[1].state_abbreviation == "WA"
        assert user.votes[1].party == 1
        assert user.votes[1].sway_points == 10
コード例 #7
0
ファイル: test_view.py プロジェクト: jrgrafton/tweet-debate
    def test_view_tasks_twitter_stream_listener_process_retweet_from_screename(self):
        load_fixture('tests/states.json', kind={'State': State})
        load_fixture('tests/questions.json', 
                        kind={'Question': Question,'State': State})
        twitter_stream_listener = TwitterStreamListener()
        current_question = Question.get_current_question()
        
        # Invalid retweet id
        twitter_stream_listener.\
            process_retweet_from_screename(current_question,
                                            -1,
                                           "jrgrafton",
                                           "https://pbs.twimg.com// \
                                            profile_images//440698495// \
                                            9929_128442162134_504357134_ \
                                            2556732_5649977_n_bigger.jpg")
        users = User.get_all()
        assert len(users) == 0

        # Valid retweet id - new user
        twitter_stream_listener.\
            process_retweet_from_screename(current_question,
                                           current_question.twitterid,
                                           "jrgrafton",
                                           "https://pbs.twimg.com// \
                                            profile_images//440698495// \
                                            9929_128442162134_504357134_ \
                                            2556732_5649977_n_bigger.jpg")
        users = User.get_all()
        assert len(users) == 1
        assert users[0].sway_points == User.get_starting_sway_points() + \
                                       sway_points_backend["rewteet_poll"]
        assert "https" in users[0].profile_image_url

        # Ensure no extra sway given for retweet
        twitter_stream_listener.\
            process_retweet_from_screename(current_question,
                                           current_question.twitterid,
                                           "jrgrafton",
                                           "https://pbs.twimg.com// \
                                            profile_images//440698495// \
                                            9929_128442162134_504357134_ \
                                            2556732_5649977_n_bigger.jpg")
        users = User.get_all()
        assert len(users) == 1
        assert users[0].sway_points == User.get_starting_sway_points() + \
                                       sway_points_backend["rewteet_poll"]

        # Test existing user reweet
        user = User(
            userid = "jrgrafton2",
            sway_points = 30
        )
        user.put()

        # Valid retweet id - new user
        twitter_stream_listener.\
            process_retweet_from_screename(current_question,
                                           current_question.twitterid,
                                           "jrgrafton2",
                                           "https://pbs.twimg.com// \
                                            profile_images//440698495// \
                                            9929_128442162134_504357134_ \
                                            2556732_5649977_n_bigger.jpg")
        users = User.get_all()
        assert len(users) == 2
        assert users[1].sway_points == 30 + sway_points_backend["rewteet_poll"]
コード例 #8
0
ファイル: test_view.py プロジェクト: jrgrafton/tweet-debate
    def test_view_tasks_twitter_stream_listener_on_data(self):
        load_fixture("tests/states.json", kind={"State": State})
        load_fixture("tests/questions_no_votes.json", kind={"Question": Question, "State": State})

        twitter_stream = open(os.path.join(os.path.dirname(__file__), "twitter_stream.json"))
        twitter_stream = json.load(twitter_stream)
        twitter_stream_listener = TwitterStreamListener()

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_reply_old"]))
        users = User.get_all()
        assert len(users) == 0

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_reply_old"]))
        users = User.get_all()
        assert len(users) == 0

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_reply_current_invalid"]))
        users = User.get_all()
        assert len(users) == 0

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_post"]))
        users = User.get_all()
        assert len(users) == 0

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_mention"]))
        users = User.get_all()
        assert len(users) == 0

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_favourite"]))
        users = User.get_all()
        assert len(users) == 0

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_favourite_reply"]))
        users = User.get_all()
        assert len(users) == 0

        current_question = Question.get_current_question()
        assert len(current_question.state_scores) == 0

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_reply_current_valid_user1"]))
        users = User.get_all()
        assert len(users) == 1
        assert "https" in users[0].profile_image_url
        current_question = Question.get_current_question()
        assert len(current_question.state_scores) == 1
        assert current_question.state_scores[0].state_abbreviation == "CA"
        assert current_question.state_scores[0].party_score_votes[0] == 0
        assert current_question.state_scores[0].party_score_votes[1] == 1
        assert current_question.state_scores[0].party_score_sway[0] == 0
        assert current_question.state_scores[0].party_score_sway[1] == User.get_starting_sway_points()
        assert users[0].sway_points == 0

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_reply_current_valid_user1"]))
        users = User.get_all()
        assert len(users) == 1
        assert len(users[0].votes) == 1
        assert users[0].sway_points == 0

        current_question = Question.get_current_question()
        assert len(current_question.state_scores) == 1
        assert current_question.state_scores[0].state_abbreviation == "CA"
        assert current_question.state_scores[0].party_score_votes[0] == 0
        assert current_question.state_scores[0].party_score_votes[1] == 1
        assert current_question.state_scores[0].party_score_sway[0] == 0
        assert current_question.state_scores[0].party_score_sway[1] == User.get_starting_sway_points()
        assert users[0].sway_points == 0

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_reply_current_valid_user2"]))
        users = User.get_all()
        assert len(users) == 2
        assert "https" in users[1].profile_image_url
        current_question = Question.get_current_question()
        assert len(current_question.state_scores) == 2
        assert current_question.state_scores[1].state_abbreviation == "WA"
        assert current_question.state_scores[1].party_score_votes[0] == 1
        assert current_question.state_scores[1].party_score_votes[1] == 0
        assert current_question.state_scores[1].party_score_sway[0] == 20
        assert current_question.state_scores[1].party_score_sway[1] == 0
        assert users[1].sway_points == User.get_starting_sway_points() - 20

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_reply_current_valid_user2"]))
        users = User.get_all()
        assert len(users) == 2
        assert current_question.state_scores[1].state_abbreviation == "WA"
        assert current_question.state_scores[1].party_score_votes[0] == 1
        assert current_question.state_scores[1].party_score_votes[1] == 0
        assert current_question.state_scores[1].party_score_sway[0] == 20
        assert current_question.state_scores[1].party_score_sway[1] == 0
        assert users[1].sway_points == User.get_starting_sway_points() - 20

        twitter_stream_listener.on_data(json.dumps(twitter_stream["twitter_retweet"]))
        users = User.get_all()
        assert len(users) == 3
        assert users[2].sway_points == User.get_starting_sway_points() + sway_points_backend["rewteet_poll"]
        assert "https" in users[2].profile_image_url
コード例 #9
0
ファイル: test_view.py プロジェクト: jrgrafton/tweet-debate
    def test_view_tasks_twitter_stream_listener_add_vote_for_screenname(self):
        load_fixture("tests/states.json", kind={"State": State})
        load_fixture("tests/questions.json", kind={"Question": Question, "State": State})
        twitter_stream_listener = TwitterStreamListener()
        current_question = Question.get_current_question()

        result = twitter_stream_listener.add_vote_for_screenname(
            current_question,
            "Vote text #yes",
            "123",
            None,  # Invalid state
            1,
            0,
            "jrgrafton",
            "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg",
        )
        assert result == False

        result = twitter_stream_listener.add_vote_for_screenname(
            current_question,
            "Vote text #CA",
            "123",
            "CA",
            None,  # Invalid party
            0,
            "jrgrafton",
            "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg",
        )
        assert result == False

        # First valid user
        result = twitter_stream_listener.add_vote_for_screenname(
            current_question,
            "Vote text #CA #yes",
            "123",
            "CA",
            0,
            100,
            "jrgrafton",
            "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg",
        )
        assert result == True
        user = User.query_by_userid("jrgrafton").get()
        assert user != None
        assert len(user.votes) == 1
        assert user.votes[0].question.get().key == current_question.key
        assert user.votes[0].vote_text == "Vote text #CA #yes"
        assert user.votes[0].replyid == "123"
        assert user.votes[0].state_abbreviation == "CA"
        assert user.votes[0].party == 0
        assert user.votes[0].sway_points == 100

        # Second vote for user on same question
        result = twitter_stream_listener.add_vote_for_screenname(
            current_question,
            "Vote text #WA #no",
            "123",
            "WA",
            1,
            0,
            "jrgrafton",
            "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg",
        )
        assert result == False

        # Second valid user
        result = twitter_stream_listener.add_vote_for_screenname(
            current_question,
            "Vote text for user 2 #WA #no",
            "123",
            "WA",
            1,
            50,
            "jrgrafton_test",
            "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg",
        )
        assert result == True
        users = User.get_all()
        assert len(users) == 2
        user = users[1]
        assert len(user.votes) == 1
        assert user.votes[0].question.get().key == current_question.key
        assert user.votes[0].vote_text == "Vote text for user 2 #WA #no"
        assert user.votes[0].replyid == "123"
        assert user.votes[0].state_abbreviation == "WA"
        assert user.votes[0].party == 1
        assert user.votes[0].sway_points == 50

        # Go to new question and test valid votes
        reponse = self.app.get("/tasks/twitter_post_status" "?question_cadence_minutes=1")
        current_question = Question.get_current_question()

        # Second vote for new question should work
        result = twitter_stream_listener.add_vote_for_screenname(
            current_question,
            "Vote second question text #WA #no",
            "124",
            "WA",
            1,
            10,
            "jrgrafton",
            "https://pbs.twimg.com// \
                                    profile_images//440698495// \
                                    9929_128442162134_504357134_ \
                                    2556732_5649977_n_bigger.jpg",
        )
        assert result == True
        user = User.query_by_userid("jrgrafton").get()
        assert user != None
        assert len(user.votes) == 2
        assert user.votes[1].vote_text == "Vote second question text #WA #no"
        assert user.votes[1].question.get().key == current_question.key
        assert user.votes[1].replyid == "124"
        assert user.votes[1].state_abbreviation == "WA"
        assert user.votes[1].party == 1
        assert user.votes[1].sway_points == 10
コード例 #10
0
ファイル: test_view.py プロジェクト: jrgrafton/tweet-debate
    def test_view_tasks_twitter_stream_listener_process_retweet_from_screename(self):
        load_fixture("tests/states.json", kind={"State": State})
        load_fixture("tests/questions.json", kind={"Question": Question, "State": State})
        twitter_stream_listener = TwitterStreamListener()
        current_question = Question.get_current_question()

        # Invalid retweet id
        twitter_stream_listener.process_retweet_from_screename(
            current_question,
            -1,
            "jrgrafton",
            "https://pbs.twimg.com// \
                                            profile_images//440698495// \
                                            9929_128442162134_504357134_ \
                                            2556732_5649977_n_bigger.jpg",
        )
        users = User.get_all()
        assert len(users) == 0

        # Valid retweet id - new user
        twitter_stream_listener.process_retweet_from_screename(
            current_question,
            current_question.twitterid,
            "jrgrafton",
            "https://pbs.twimg.com// \
                                            profile_images//440698495// \
                                            9929_128442162134_504357134_ \
                                            2556732_5649977_n_bigger.jpg",
        )
        users = User.get_all()
        assert len(users) == 1
        assert users[0].sway_points == User.get_starting_sway_points() + sway_points_backend["rewteet_poll"]
        assert "https" in users[0].profile_image_url

        # Ensure no extra sway given for retweet
        twitter_stream_listener.process_retweet_from_screename(
            current_question,
            current_question.twitterid,
            "jrgrafton",
            "https://pbs.twimg.com// \
                                            profile_images//440698495// \
                                            9929_128442162134_504357134_ \
                                            2556732_5649977_n_bigger.jpg",
        )
        users = User.get_all()
        assert len(users) == 1
        assert users[0].sway_points == User.get_starting_sway_points() + sway_points_backend["rewteet_poll"]

        # Test existing user reweet
        user = User(userid="jrgrafton2", sway_points=30)
        user.put()

        # Valid retweet id - new user
        twitter_stream_listener.process_retweet_from_screename(
            current_question,
            current_question.twitterid,
            "jrgrafton2",
            "https://pbs.twimg.com// \
                                            profile_images//440698495// \
                                            9929_128442162134_504357134_ \
                                            2556732_5649977_n_bigger.jpg",
        )
        users = User.get_all()
        assert len(users) == 2
        assert users[1].sway_points == 30 + sway_points_backend["rewteet_poll"]