"""Registration related tests."""
try:
    from HTMLParser import HTMLParser
except ImportError:
    from html.parser import HTMLParser

import pytest
import transaction
from pyramid.compat import text_type
from sqlalchemy.orm.exc import NoResultFound

from pyramid_fullauth.models import User
from tests.tools import DEFAULT_USER


def test_register_view(default_app):
    """Display register form."""
    res = default_app.get('/register')
    assert res.form


@pytest.mark.parametrize('email, password', (
    # casual
    (DEFAULT_USER['email'], DEFAULT_USER['password']),
    # loooong password
    (DEFAULT_USER['email'], DEFAULT_USER['password'] * 10000),
))
def test_register_success(db_session, default_app, email, password):
    """Register user successfully."""
    assert db_session.query(User).count() == 0

    res = default_app.get('/register')
    res.form['email'] = email
    res.form['password'] = password
    res.form['confirm_password'] = password
    res = res.form.submit(extra_environ={'REMOTE_ADDR': '0.0.0.0'})
    transaction.commit()

    user = db_session.query(User).filter(User.email == email).one()

    # User should not have active account at this moment
    assert user.is_active is not None
    assert user.check_password(password)


def test_register_user_exists(db_session, user, default_app):  # pylint:disable=unused-argument
    """Trying to register existing user."""
    res = default_app.get('/register')
    res.form['email'] = DEFAULT_USER['email']
    res.form['password'] = DEFAULT_USER['password']
    res.form['confirm_password'] = DEFAULT_USER['password']
    res = res.form.submit(extra_environ={'REMOTE_ADDR': '0.0.0.0'})
    transaction.commit()

    assert 'User with given e-mail already exists' in res.body.decode('unicode_escape')


@pytest.mark.parametrize('email, password, confirm_password, error', (
    # no email
    (None, DEFAULT_USER['password'], DEFAULT_USER['password'], 'E-mail is empty'),
    # empty email
    ('', DEFAULT_USER['password'], DEFAULT_USER['password'], 'E-mail is empty'),
    # not matching passwords
    (DEFAULT_USER['email'], DEFAULT_USER['password'], DEFAULT_USER['password'] + 'Typo', 'Passwords don\'t match'),
    # long, incorect email
    (
        text_type('email') * 100 + text_type('@wap.pl'), DEFAULT_USER['password'], DEFAULT_USER['password'],
        'Incorrect e-mail format'
    ),
    # too short password
    (DEFAULT_USER['email'], '12', '12', 'Password is too short'),
    # empty password
    (DEFAULT_USER['email'], '', '', 'Please enter your password'),
))
def test_register_error(db_session, default_app, email, password, confirm_password, error):
    """Error in registration process."""
    assert db_session.query(User).count() == 0

    res = default_app.get('/register')
    if email is not None:
        res.form['email'] = email
    res.form['password'] = password
    res.form['confirm_password'] = confirm_password
    res = res.form.submit(extra_environ={'REMOTE_ADDR': '0.0.0.0'})
    transaction.commit()

    assert error in HTMLParser().unescape(res.body.decode('unicode_escape'))
    assert db_session.query(User).count() == 0


def test_register_wrong_csrf(db_session, default_app):
    """CSRF error during registration."""
    assert db_session.query(User).count() == 0

    res = default_app.get('/register')
    res.form['email'] = DEFAULT_USER['email']
    res.form['password'] = DEFAULT_USER['password']
    res.form['confirm_password'] = DEFAULT_USER['password']
    res.form['csrf_token'] = 'wrong_token'
    res = res.form.submit(extra_environ={'REMOTE_ADDR': '0.0.0.0'},
                          status=400)
    transaction.commit()

    assert db_session.query(User).count() == 0


def test_no_pass_confirm(db_session, nopassconfirm_app):
    """Register without password confirm option."""
    assert db_session.query(User).count() == 0

    res = nopassconfirm_app.get('/register')
    res.form['email'] = DEFAULT_USER['email']
    res.form['password'] = DEFAULT_USER['password']
    res = res.form.submit(extra_environ={'REMOTE_ADDR': '0.0.0.0'})

    assert 'Passwords don\'t match!' not in HTMLParser().unescape(res.body.decode('unicode_escape'))
    transaction.commit()

    user = db_session.query(User).filter(User.email == DEFAULT_USER['email']).one()

    # User should not have active account at this moment
    assert user.is_active is not None
    assert user.check_password(DEFAULT_USER['password'])


def test_register_action_no_password_required(db_session, nopassregister_app):
    """Register without password required."""
    with pytest.raises(NoResultFound):
        db_session.query(User).filter(User.email == DEFAULT_USER['email']).one()

    res = nopassregister_app.get('/register')
    res.form['email'] = DEFAULT_USER['email']
    res = res.form.submit(extra_environ={'REMOTE_ADDR': '0.0.0.0'})
    transaction.commit()

    db_session.query(User).filter(User.email == DEFAULT_USER['email']).one()

@@ -31,75 +31,9 @@
 %22%22%22%0A
-try:%0A    from HTMLParser import HTMLParser%0Aexcept ImportError:%0A    
+%0A
 from
@@ -52,26 +52,24 @@
  import 
-HTMLParser
+unescape
 %0A%0Aimport
@@ -2974,37 +2974,24 @@
 rt error in 
-HTMLParser().
 unescape(res
@@ -4054,21 +4054,8 @@
  in 
-HTMLParser().
 unes

Use html.unescape for Python 3.9 compatibility.