def test_upvote(self, mock_vote): message = Message() message.upvote() mock_vote.assert_called_with('upvote')