Python compat.text_type函数代码示例

def test_social_login_register(social_config, db_session):
    """Register fresh user and logs him in."""
    profile = {
        'accounts': [{'domain': text_type('facebook.com'), 'userid': text_type('2343')}],
        'displayName': text_type('teddy'),
        'verifiedEmail': text_type('[email protected]'),
        'preferredUsername': text_type('teddy'),
        'emails': [{'value': text_type('[email protected]')}],
        'name': text_type('ted')
    }
    credentials = {'oauthAccessToken': '7897048593434'}
    provider_name = text_type('facebook')
    provider_type = text_type('facebook')
    request = testing.DummyRequest()
    request.user = None
    request.registry = social_config.registry
    request.remote_addr = text_type('127.0.0.123')
    request.context = AuthenticationComplete(profile, credentials, provider_name, provider_type)

    request.login_perform = MagicMock(name='login_perform')
    request.login_perform.return_value = {'status': True}
    view = SocialLoginViews(request)
    out = view()
    assert out == {'status': True}
    transaction.commit()

    # read first new account
    user = db_session.query(User).one()
    assert user.is_active
    assert user.provider_id('facebook') == profile['accounts'][0]['userid']

def test_login_different_social_account(social_config, db_session, facebook_user):  # pylint:disable=unused-argument
    """
    Login with different social account than connected from same provider.

    System should let user in, but not change connection.
    """
    # profile mock response
    profile = {
        # facebook user id is different than user's
        'accounts': [{'domain': text_type('facebook.com'), 'userid': text_type('2343')}],
        'displayName': text_type('teddy'),
        'verifiedEmail': facebook_user.email,
        'preferredUsername': text_type('teddy'),
        'emails': [{'value': text_type('[email protected]')}],
        'name': text_type('ted')
    }
    request = testing.DummyRequest()
    request.user = None
    request.registry = social_config.registry
    request.remote_addr = text_type('127.0.0.123')
    request.context = AuthenticationComplete(
        profile,
        {'oauthAccessToken': '7897048593434'},
        text_type('facebook'),
        text_type('facebook'))

    request.login_perform = MagicMock(name='login_perform')
    request.login_perform.return_value = {'status': True}
    view = SocialLoginViews(request)
    out = view()
    # user should be authenticated recognized by email!
    assert out['status'] is True
    assert facebook_user.provider_id(text_type('facebook')) is not profile['accounts'][0]['userid']

def test_change_email(db_with_user):
    '''User::change_email'''
    user = db_with_user.query(User).filter(User.email == text_type('[email protected]')).one()
    user.set_new_email(text_type('[email protected]'))
    user.change_email()

    assert not user.email_change_key

def test_password_empty(db_with_user):
    """User::empty password change"""

    user = db_with_user.query(User).filter(User.username == text_type("u1")).one()

    with pytest.raises(EmptyError):
        user.password = text_type("")

def test_aftersociallogin(aftersociallogin_config, db_session):  # pylint:disable=redefined-outer-name
    """Register fresh user and logs him in and check response if redirect from AfterSocialLogIn."""
    profile = {
        'accounts': [{'domain': text_type('facebook.com'), 'userid': text_type('2343')}],
        'displayName': text_type('teddy'),
        'verifiedEmail': text_type('[email protected]'),
        'preferredUsername': text_type('teddy'),
        'emails': [{'value': text_type('[email protected]')}],
        'name': text_type('ted')
    }
    credentials = {'oauthAccessToken': '7897048593434'}
    provider_name = text_type('facebook')
    provider_type = text_type('facebook')
    request = testing.DummyRequest()
    request.user = None
    request.registry = aftersociallogin_config.registry
    request.remote_addr = text_type('127.0.0.123')
    request.context = AuthenticationComplete(profile, credentials, provider_name, provider_type)

    def login_perform(*_, **kwargs):
        return HTTPFound(location=kwargs['location'])
    request.login_perform = login_perform
    view = SocialLoginViews(request)
    out = view()
    assert out.location == EVENT_PATH.format(AfterSocialLogIn)
    transaction.commit()

    # read first new account
    user = db_session.query(User).one()
    assert user.is_active
    assert user.provider_id('facebook') == profile['accounts'][0]['userid']

def test_logged_social_connect_account(social_config, active_user, db_session):
    """Connect facebook account to logged in user."""
    user = db_session.merge(active_user)

    profile = {
        'accounts': [{'domain': text_type('facebook.com'), 'userid': text_type('2343')}],
        'displayName': text_type('teddy'),
        'preferredUsername': text_type('teddy'),
        'emails': [{'value': text_type('[email protected]')}],
        'name': text_type('ted')
    }
    credentials = {'oauthAccessToken': '7897048593434'}
    provider_name = text_type('facebook')
    provider_type = text_type('facebook')
    request = testing.DummyRequest()
    request.user = user
    request.registry = social_config.registry
    request.remote_addr = text_type('127.0.0.123')
    request.context = AuthenticationComplete(profile, credentials, provider_name, provider_type)
    request._ = mock_translate

    request.login_perform = MagicMock(name='login_perform')
    request.login_perform.return_value = {'status': True}
    view = SocialLoginViews(request)
    out = view()
    assert out['status'] is True

    transaction.commit()
    user = db_session.merge(user)
    assert user.provider_id(text_type('facebook')) == profile['accounts'][0]['userid']

def test_social_login_register(social_config, db_session):
    """Register fresh user and logs him in."""
    profile = {
        "accounts": [{"domain": text_type("facebook.com"), "userid": text_type("2343")}],
        "displayName": text_type("teddy"),
        "verifiedEmail": text_type("[email protected]"),
        "preferredUsername": text_type("teddy"),
        "emails": [{"value": text_type("[email protected]")}],
        "name": text_type("ted"),
    }
    credentials = {"oauthAccessToken": "7897048593434"}
    provider_name = text_type("facebook")
    provider_type = text_type("facebook")
    request = testing.DummyRequest()
    request.user = None
    request.registry = social_config.registry
    request.remote_addr = text_type("127.0.0.123")
    request.context = AuthenticationComplete(profile, credentials, provider_name, provider_type)

    request.login_perform = MagicMock(name="login_perform")
    request.login_perform.return_value = {"status": True}
    view = SocialLoginViews(request)
    out = view()
    assert out == {"status": True}
    transaction.commit()

    # read first new account
    user = db_session.query(User).one()
    assert user.is_active
    assert user.provider_id("facebook") == profile["accounts"][0]["userid"]

def db_locales(db_session):  # pylint:disable=redefined-outer-name
    """Add Languages to db_session."""
    for locale in ['pl', 'cz', 'fr']:
        locale_object = Language(name=text_type(locale),
                                 native_name=text_type(locale),
                                 language_code=text_type(locale))
        db_session.add(locale_object)
    transaction.commit()

def test_login_different_social_account(social_config, db_session, facebook_user):
    """
    Login with different social account than connected from same provider.

    System should let user in, but not change connection.
    """
    # profile mock response
    profile = {
        # facebook user id is different than user's
        "accounts": [{"domain": text_type("facebook.com"), "userid": text_type("2343")}],
        "displayName": text_type("teddy"),
        "verifiedEmail": facebook_user.email,
        "preferredUsername": text_type("teddy"),
        "emails": [{"value": text_type("[email protected]")}],
        "name": text_type("ted"),
    }
    request = testing.DummyRequest()
    request.user = None
    request.registry = social_config.registry
    request.remote_addr = text_type("127.0.0.123")
    request.context = AuthenticationComplete(
        profile, {"oauthAccessToken": "7897048593434"}, text_type("facebook"), text_type("facebook")
    )

    request.login_perform = MagicMock(name="login_perform")
    request.login_perform.return_value = {"status": True}
    view = SocialLoginViews(request)
    out = view()
    # user should be authenticated recognized by email!
    assert out["status"] is True
    assert facebook_user.provider_id("facebook") is not profile["accounts"][0]["userid"]

def test_logged_social_connect_second_account(social_config, facebook_user, db_session):
    """Connect second facebook account to logged in user."""
    user = db_session.merge(facebook_user)

    # mock request
    profile = {
        "accounts": [{"domain": text_type("facebook.com"), "userid": text_type("2343")}],
        "displayName": text_type("teddy"),
        "preferredUsername": text_type("teddy"),
        "emails": [{"value": text_type("[email protected]")}],
        "name": text_type("ted"),
    }
    credentials = {"oauthAccessToken": "7897048593434"}
    provider_name = text_type("facebook")
    provider_type = text_type("facebook")
    request = testing.DummyRequest()
    request.user = user
    request.registry = social_config.registry
    request.remote_addr = text_type("127.0.0.123")
    request.context = AuthenticationComplete(profile, credentials, provider_name, provider_type)
    request._ = mock_translate

    request.login_perform = MagicMock(name="login_perform")
    request.login_perform.return_value = {"status": True}
    view = SocialLoginViews(request)
    out = view()
    # status should be false
    assert out["status"] is False
    assert out["msg"] == "Your account is already connected to other ${provider} account."
    assert user.provider_id("facebook") is not profile["accounts"][0]["userid"]

def test_logged_social_connect_account(social_config, active_user, db_session):
    """Connect facebook account to logged in user."""
    user = db_session.merge(active_user)

    profile = {
        "accounts": [{"domain": text_type("facebook.com"), "userid": text_type("2343")}],
        "displayName": text_type("teddy"),
        "preferredUsername": text_type("teddy"),
        "emails": [{"value": text_type("[email protected]")}],
        "name": text_type("ted"),
    }
    credentials = {"oauthAccessToken": "7897048593434"}
    provider_name = text_type("facebook")
    provider_type = text_type("facebook")
    request = testing.DummyRequest()
    request.user = user
    request.registry = social_config.registry
    request.remote_addr = text_type("127.0.0.123")
    request.context = AuthenticationComplete(profile, credentials, provider_name, provider_type)
    request._ = mock_translate

    request.login_perform = MagicMock(name="login_perform")
    request.login_perform.return_value = {"status": True}
    view = SocialLoginViews(request)
    out = view()
    assert out["status"] is True

    transaction.commit()
    user = db_session.merge(user)
    assert user.provider_id("facebook") == profile["accounts"][0]["userid"]

def test_email_from_context(profile, email):
    """Test email_from_context email getting method."""
    from velruse import AuthenticationComplete
    context = AuthenticationComplete(
        profile,
        {'oauthAccessToken': '7897048593434'},
        text_type('facebook'),
        text_type('facebook')
    )
    view = SocialLoginViews(mock.MagicMock())
    assert view._email_from_context(context) == email

    def setUp(self):
        '''
            setUp test method @see unittest.TestCase.setUp
        '''
        Base.metadata.create_all(engine)

        for locale in ['pl', 'cz', 'fr']:
            locale_object = Language(name=text_type(locale),
                                     native_name=text_type(locale),
                                     language_code=text_type(locale))
            Session.add(locale_object)

    def password_validator(self, _, password):
        """
        Validate password.

        Password validator keeps new password hashed.
        Rises Value error on empty password

        :param str key: field key
        :param str password: new password

        :returns: hashed and salted password
        :rtype: str
        :raises: pyramid_fullauth.exceptions.EmptyError

        .. note::

            If you're using this Mixin on your own User object,
            don't forget to add a listener as well, like that:

            .. code-block:: python

                from sqlalchemy.event import listen

                listen(User.password, 'set', User.password_listener, retval=True)

        .. note::

            For more information on Attribute Events in sqlalchemy see:

            :meth:`sqlalchemy.orm.events.AttributeEvents.set`

        """
        if not password:
            raise EmptyError('password-empty')

        # reading default hash_algorithm
        # pylint:disable=protected-access
        hash_algorithm = self.__class__._hash_algorithm.property.columns[0].default.arg
        # pylint:enable=protected-access

        # getting currently used hash method
        hash_method = getattr(hashlib, hash_algorithm)

        # generating salt
        salt = hash_method()
        salt.update(os.urandom(60))
        salt_value = salt.hexdigest()

        # storing used hash algorithm
        self._hash_algorithm = hash_algorithm
        self._salt = text_type(salt_value)
        return text_type(self.__class__.hash_password(password, salt_value, hash_method))

    def test_email_valid_formats(self, db, email):
        ''' Check all valid formats of Email (RFC 5321) can be set by user
        '''
        self.create_user(db, username=text_type('u1'))

        user = db.query(User).filter(User.username == text_type('u1')).one()

        user.email = email
        db.commit()

        user = db.query(User).filter(
            User.username == text_type('u1')).one()
        assert user.email == email