ordr2/tests/events.py

''' Tests for ordr2.security '''

import pytest

from datetime import datetime
from pyramid.testing import DummyRequest
from pyramid_mailer import get_mailer

from . import app_config, get_user


def test_user_notification_init():
    ''' test creation of user notification events '''
    from ordr2.events import UserNotification

    notification = UserNotification('a', 'b', 'c')

    assert notification.request == 'a'
    assert notification.account == 'b'
    assert notification.data == 'c'


def test_notify_user(app_config):
    from ordr2.events import CompleteRegistration, notify_user
    from ordr2.models.account import Token

    request = DummyRequest()
    user = get_user('user')
    token = Token(expires=datetime.utcnow(), hash='some_hash')
    notification = CompleteRegistration(request, user, {'token': token})
    notify_user(notification)
    mailer = get_mailer(request.registry)
    last_mail = mailer.outbox[-1]

    assert 'Please verify your email address ' in last_mail.html
    assert 'http://example.com//some_hash' in last_mail.html