Exemple #1
0
 def test_send_epistle_by_id(self):
     # @TODO: as this test illustrates, we should make a "send_epistle" function in queries
     u1 = users.create_user(username="******", password="******")
     u2 = users.create_user(username="******", password="******")
     ep = self.create_epistle(u1, u2, u"test epistle", subject=u"a simple test")
     epd = epistle_queries.get_epistle_by_recipient_id(u1.id)[0]
     assert ep.id == epd.id
Exemple #2
0
 def test_send_epistle_by_id(self):
     #@TODO: as this test illustrates, we should make a "send_epistle" function in queries
     u1 = users.create_user(username = '******', password='******')
     u2 = users.create_user(username = '******', password='******')
     ep = self.create_epistle(u1, u2, u'test epistle', subject = u'a simple test')
     epd = epistle_queries.get_epistle_by_recipient_id(u1.id)[0]
     assert ep.id == epd.id
 def test_get_story_by_url_oldest(self):
     user01 = users.create_user(username = '******', password = '******')
     user02 = users.create_user(username = '******', password = '******')
     submission01 = self.create_submission(user = user01)
     submission02 = self.create_submission(user = user02)
     result = submission_queries.get_story_by_url_oldest('http://google.com')
     if result.__class__ == Submission:
         assert result.id == submission01.id
     else:
         assert False
 def test_get_story_by_url_oldest(self):
     user01 = users.create_user(username='******', password='******')
     user02 = users.create_user(username='******', password='******')
     submission01 = self.create_submission(user=user01)
     submission02 = self.create_submission(user=user02)
     result = submission_queries.get_story_by_url_oldest(
         'http://google.com')
     if result.__class__ == Submission:
         assert result.id == submission01.id
     else:
         assert False
Exemple #5
0
 def create_epistle(self, recipient=None, sender=None, body=None, subject=None):
     if not recipient:
         recipient = users.create_user(username="******", password="******")
     if not sender:
         sender = users.create_user(username="******", password="******")
     if not body:
         body = u"test epistle"
     if not subject:
         subject = u"a simple test"
     epistle = Epistle(recipient.id, sender.id, body, subject=subject)
     self.dbsession.add(epistle)
     self.dbsession.flush()
     return epistle
Exemple #6
0
 def create_epistle(self, recipient = None, sender = None, body = None, subject = None):
     if not recipient:
         recipient = users.create_user(username = '******', password='******')
     if not sender:
         sender = users.create_user(username = '******', password='******')
     if not body:
         body = u'test epistle'
     if not subject:
         subject = u'a simple test'
     epistle = Epistle(recipient.id, sender.id, body, subject = subject)
     self.dbsession.add(epistle)
     self.dbsession.flush()
     return epistle
    def test_domain_parse(self):
        #@TODO: we should make this accept a fake user id in test mode at least
        # so that we don't have huge cascading failures if create_user is broken
        user = users.create_user(username = '******', password = '******')
        submission = self.create_submission(user = user, url = 'http://google.com')
        assert 'google.com' == submission.get_domain_name()

        user = users.create_user(username = '******', password = '******')
        submission = self.create_submission(user = user, url = 'http://googlewww.com')
        assert 'googlewww.com' == submission.get_domain_name()

        user = users.create_user(username = '******', password = '******')
        submission = self.create_submission(user = user, url = 'https://google.com')
        assert 'google.com' == submission.get_domain_name()
    def test_domain_parse(self):
        #@TODO: we should make this accept a fake user id in test mode at least
        # so that we don't have huge cascading failures if create_user is broken
        user = users.create_user(username='******', password='******')
        submission = self.create_submission(user=user, url='http://google.com')
        assert 'google.com' == submission.get_domain_name()

        user = users.create_user(username='******', password='******')
        submission = self.create_submission(user=user,
                                            url='http://googlewww.com')
        assert 'googlewww.com' == submission.get_domain_name()

        user = users.create_user(username='******', password='******')
        submission = self.create_submission(user=user,
                                            url='https://google.com')
        assert 'google.com' == submission.get_domain_name()
Exemple #9
0
def create_local_user(fb_info, local_username, request = None): 
    u = users.create_user(origination='facebook', remote_object=fb_info, username=local_username)
    if request:
        profile_picture = request.session['u_fbgraph'].get_connections(fb_info['id'], "picture")
        up = users.add_user_picture("{0}-fbprofile.jpg".format(fb_info['id']), fb_info['id'], request.registry.settings['user.picture_upload_directory'], profile_picture)
        u.picture = up
        dbsession.add(u)
    return u
Exemple #10
0
 def create_user(self, username=None):
     if not username:
         username = '******'
     user = users.create_user(
         username=username,
         password=username,
     )
     return user
 def create_submission(self, user = None, url = None, title = None, description = None):
     if not user:
         user = users.create_user(username = '******', password = '******')
     if not url:
         url = 'http://google.com'
     if not title:
         title = 'test'
     if not description:
         description = 'test'
     submission = Submission(title, description, url, user.id)
     self.dbsession.add(submission)
     self.dbsession.flush()
     return submission
Exemple #12
0
def create_local_user(fb_info, local_username, request=None):
    u = users.create_user(origination='facebook',
                          remote_object=fb_info,
                          username=local_username)
    if request:
        profile_picture = request.session['u_fbgraph'].get_connections(
            fb_info['id'], "picture")
        up = users.add_user_picture(
            "{0}-fbprofile.jpg".format(fb_info['id']), fb_info['id'],
            request.registry.settings['user.picture_upload_directory'],
            profile_picture)
        u.picture = up
        dbsession.add(u)
    return u
Exemple #13
0
def complete_auth(request, auth_toks):
    # create a twython object with our request-specific tokens
    # these tokens are used only to learn if the user accepted our request
    # for permissions.
    tmp_twit = Twython(
        twitter_token=request.registry.settings['twitter.app_key'],
        twitter_secret=request.registry.settings['twitter.app_secret'],
        oauth_token=auth_toks['oauth_token'],
        oauth_token_secret=auth_toks['oauth_token_secret'])

    final_toks = tmp_twit.get_authorized_tokens()

    # create a session-permanent twython object containing the permanent tokens for this user.
    # this object must be used to read any data from the user's account.
    u_twit = Twython(
        twitter_token=request.registry.settings['twitter.app_key'],
        twitter_secret=request.registry.settings['twitter.app_secret'],
        oauth_token=final_toks['oauth_token'],
        oauth_token_secret=final_toks['oauth_token_secret'])

    request.session['u_twit'] = u_twit

    username = "******".format(final_toks['oauth_token'])
    screen_name = final_toks['screen_name']

    # check if user already exists; if not, please create
    try:
        u = users.get_user_by_name(username)
        #@TODO: add something to ensure we are in sync with the twitter profile picture
        # unless specifically overridden by the user
    except sqlalchemy.orm.exc.NoResultFound:
        u = users.create_user(origination='twitter',
                              username=username,
                              remote_object=final_toks)
        import urllib2
        image_data = urllib2.urlopen(
            "http://api.twitter.com/1/users/profile_image/{0}.json".format(
                screen_name))
        orig_filename = "{0}-twitter-pic.png".format(screen_name)
        up_dir = request.registry.settings['user.picture_upload_directory']

        u.picture = users.add_user_picture(orig_filename,
                                           str(u.id)[:7], up_dir, image_data)

        dbsession.add(u)

    return {'final_toks': final_toks, 'u': u}
 def create_submission(self,
                       user=None,
                       url=None,
                       title=None,
                       description=None):
     if not user:
         user = users.create_user(username='******', password='******')
     if not url:
         url = 'http://google.com'
     if not title:
         title = 'test'
     if not description:
         description = 'test'
     submission = Submission(title, description, url, user.id)
     self.dbsession.add(submission)
     self.dbsession.flush()
     return submission
Exemple #15
0
def complete_auth(request, auth_toks):
    # create a twython object with our request-specific tokens
    # these tokens are used only to learn if the user accepted our request
    # for permissions.
    tmp_twit = Twython(twitter_token = request.registry.settings['twitter.app_key'],
                       twitter_secret = request.registry.settings['twitter.app_secret'],
                       oauth_token = auth_toks['oauth_token'],
                       oauth_token_secret = auth_toks['oauth_token_secret'])

    final_toks = tmp_twit.get_authorized_tokens()

    # create a session-permanent twython object containing the permanent tokens for this user.
    # this object must be used to read any data from the user's account.
    u_twit = Twython(twitter_token = request.registry.settings['twitter.app_key'],
                       twitter_secret = request.registry.settings['twitter.app_secret'],
                       oauth_token = final_toks['oauth_token'],
                       oauth_token_secret = final_toks['oauth_token_secret'])

    request.session['u_twit'] = u_twit

    username = "******".format(final_toks['oauth_token'])
    screen_name = final_toks['screen_name']

    # check if user already exists; if not, please create
    try:
        u = users.get_user_by_name(username)
        #@TODO: add something to ensure we are in sync with the twitter profile picture
        # unless specifically overridden by the user
    except sqlalchemy.orm.exc.NoResultFound:
        u = users.create_user(origination='twitter', username=username, remote_object=final_toks)
        import urllib2
        image_data = urllib2.urlopen("http://api.twitter.com/1/users/profile_image/{0}.json".format(screen_name))
        orig_filename = "{0}-twitter-pic.png".format(screen_name)
        up_dir = request.registry.settings['user.picture_upload_directory']

        u.picture = users.add_user_picture(orig_filename, str(u.id)[:7], up_dir, image_data)

        dbsession.add(u)

    return {'final_toks': final_toks, 'u': u} 
Exemple #16
0
 def test_get_epistle_by_recipient_name(self):
     user = users.create_user(username="******", password="******")
     epistle = self.create_epistle(recipient=user)
     result = epistle_queries.get_epistle_by_recipient_name(user.name)[0]
     assert epistle.id == result.id
Exemple #17
0
 def create_user(self, username = None):
     if not username:
         username = '******'
     user = users.create_user(username = username, password = username,)
     return user
Exemple #18
0
 def test_find_by_id_user(self):
     # depends on functional test_create_user
     u = users.create_user(username = '******', password='******')
     res = general.find_by_id(u.id)
     assert res.id == u.id
     assert res.name == u.name
Exemple #19
0
 def test_get_user_by_name(self):
     u = users.create_user(username = '******', password='******')
     res = users.get_user_by_name(u.name)
     assert u.id == res.id
Exemple #20
0
 def test_get_user_by_name(self):
     u = users.create_user(username='******', password='******')
     res = users.get_user_by_name(u.name)
     assert u.id == res.id
Exemple #21
0
 def test_get_new_message_num(self):
     recipient = users.create_user(username="******", password="******")
     epistle = self.create_epistle(recipient=recipient)
     result = epistle_queries.get_new_message_num(recipient.id)
     assert result == 1
Exemple #22
0
 def test_get_new_message_num(self):
     recipient = users.create_user(username = '******', password = '******')
     epistle = self.create_epistle(recipient = recipient)
     result = epistle_queries.get_new_message_num(recipient.id)
     assert result == 1
Exemple #23
0
 def test_get_epistle_by_recipient_name(self):
     user = users.create_user(username = '******', password = '******')
     epistle = self.create_epistle(recipient = user)
     result = epistle_queries.get_epistle_by_recipient_name(user.name)[0]
     assert epistle.id == result.id
Exemple #24
0
def login(request):
    #@FIXME: this uses a request handling method with success with which I was experimenting
    # it is not used elsewhere and is a pain to read and write
    # success = False causes a page to stop drawing and "error out"
    # some error conditions therefore don't set success to false because it's more convenient
    # to draw the rest of the page.
    #
    # someone should adapt this to be less success-centric and read less branchy.
    s = request.session

    success = True

    # check for facebook login, provided by Facebook's JS SDK
    try:
        fb_cookie = fb.extract_from_cookie(request)
        try:
            u = users.get_user_by_name(fb_cookie['local_username'])
        except sqlalchemy.orm.exc.NoResultFound:
            u = fb.create_local_user(fb_cookie['info'],
                                     fb_cookie['local_username'],
                                     request=request)
        try:
            users.login_user(request, u, None, bypass_password=True)
        except LoginAdapterExc:
            pass
    except LoginAdapterExc:
        pass

    if 'logout' in request.session['safe_params']:
        if 'logged_in' in s:
            del s['logged_in']
            del s['users.id']
            if 'u_fbgraph' in s:
                del s['u_fbgraph']
                del s['u_fbinfo']
            if 'u_twit' in s:
                del s['u_twit']
            s['message'] = "You have been logged out, thanks."
            success = True
        else:
            s['message'] = "You are not logged in."
            success = True
    else:
        logged_in = False
        if 'logged_in' in s:
            s['message'] = "You are already logged in."
            logged_in = True
        else:
            if 'message' not in s:
                if 'last_login_status' in s:
                    s['message'] = s['last_login_status']
                    del s['last_login_status']
                else:
                    s['message'] = "Please log in."
        p = request.session['safe_post']
        prm = request.session['safe_params']
        username = None
        if 'username' in prm:
            username = general.strip_all_html(prm['username'])
        if p:
            dbsession = DBSession()
            if request.session['safe_get']['act'] == 'register':
                if logged_in:
                    try:
                        u = users.get_user_by_id(s['users.id'])
                        if u.temporary:
                            users.create_user(temp_to_perm=True,
                                              extant_id=s['users.id'],
                                              username=username,
                                              password=p['password'],
                                              email=p['email'],
                                              origination='site')
                            s['message'] = "Your anonymous profile has been converted, thanks."
                        else:
                            s['message'] = "You can't register while you're logged in."
                    except sqlalchemy.exc.IntegrityError:
                        s['message'] = "This username is already registered, sorry."
                        dbsession.rollback()
                else:
                    try:
                        users.create_user(username=username,
                                          password=p['password'],
                                          email=p['email'],
                                          origination='site')
                        s['message'] = "Successfully registered."
                        success = True
                    except sqlalchemy.exc.IntegrityError:
                        s['message'] = "This username is already registered, sorry."
                        success = False
                        dbsession.rollback()
            elif request.session['safe_get']['act'] == 'update_pw':
                if p['new_password'] != p['new_password_confirm']:
                    s['message'] = 'New password doesn\'t match confirmation, please try again.'
                else:
                    u = None

                    if s['logged_in_admin']:
                        if 'user_id' in prm:
                            u = users.get_user_by_id(prm['user_id'])

                    if u == None:
                        u = users.get_user_by_id(s['users.id'])

                    if u.verify_pw(p['old_password']) or s['logged_in_admin']:
                        u.password = u.hash_pw(p['new_password'])
                        dbsession.add(u)
                        s['message'] = 'Password updated.'
                        success = True
                    else:
                        s['message'] = 'Old password invalid.'
            elif request.session['safe_get']['act'] == 'forgot_pass':
                user = users.get_user_by_email(p['email'])
                if not user:
                    s['message'] = "That email isn't registered"
                else:
                    s['message'] = "Check your mail for a confirmation message."
                    users.send_lost_password_verify_email(request, user)
            else:
                try:
                    u = users.get_user_by_name(username)
                    try:
                        users.login_user(request, u, p['password'])
                        s['message'] = "Good, logged in"
                        success = True
                        return HTTPFound(request.route_url('post'))
                    except LoginAdapterExc:
                        s['message'] = "Incorrect password."
                        success = False
                except sqlalchemy.orm.exc.NoResultFound:
                    s['message'] = "Sorry, I don't know you."
                    success = False

    return {
        'success': success,
    }
Exemple #25
0
 def test_find_by_id_user(self):
     # depends on functional test_create_user
     u = users.create_user(username='******', password='******')
     res = general.find_by_id(u.id)
     assert res.id == u.id
     assert res.name == u.name