def test_get_user_activation(self):
from pyramid_signup.models import Activation
from pyramid_signup.managers import ActivationManager
from pyramid_signup.managers import UserManager
from pyramid_signup.models import User
user1 = User(username='******',
password='******',
email='*****@*****.**')
user2 = User(username='******',
password='******',
email='*****@*****.**')
activation = Activation()
user2.activation = activation
self.session.add(user1)
self.session.add(user2)
self.session.commit()
request = testing.DummyRequest()
mgr = ActivationManager(request)
user_mgr = UserManager(request)
new_user = user_mgr.get_by_username('sontek2')
new_activation = mgr.get_by_code(activation.code)
assert activation == new_activation
assert new_user.activation == new_activation
def test_get_activation(self):
from pyramid_signup.models import Activation
from pyramid_signup.managers import ActivationManager
activation = Activation()
self.session.add(activation)
self.session.commit()
request = testing.DummyRequest()
mgr = ActivationManager(request)
new_activation = mgr.get_by_code(activation.code)
assert activation == new_activation
def test_activate_multiple_users(self):
from pyramid_signup.views import RegisterController
from pyramid_signup.models import User
from pyramid_signup.models import Activation
from pyramid_mailer.interfaces import IMailer
from pyramid_mailer.mailer import DummyMailer
from pyramid_signup.managers import UserManager
from pyramid_signup.managers import ActivationManager
self.config.include('pyramid_signup')
self.config.add_route('index', '/')
self.config.registry.registerUtility(DummyMailer(), IMailer)
user = User(username='******', password='******')
user.activation = Activation()
user1 = User(username='******', password='******')
user1.activation = Activation()
self.session.add(user)
self.session.add(user1)
self.session.flush()
request = testing.DummyRequest()
request.matchdict = Mock()
get = Mock()
get.return_value = user1.activation.code
request.matchdict.get = get
controller = RegisterController(request)
response = controller.activate()
mgr = UserManager(request)
user = mgr.get_by_username('sontek1')
act_mgr = ActivationManager(request)
activations = act_mgr.get_all()
assert len(activations) == 1
assert user.activated
assert response.status_int == 302
def activate(self):
code = self.request.matchdict.get('code', None)
act_mgr = ActivationManager(self.request)
user_mgr = UserManager(self.request)
activation = act_mgr.get_by_code(code)
if activation:
user = user_mgr.get_by_activation(activation)
if user:
self.db.delete(activation)
user.activated = True
self.db.add(user)
self.db.flush()
self.request.registry.notify(
RegistrationActivatedEvent(self.request, user, activation))
self.request.session.flash(
_('Your e-mail address has been verified.'), 'success')
return HTTPFound(location=self.activate_redirect_view)
return HTTPNotFound()
