Produits Entr'ouvert » Authentic 2

from django.core.validators import RegexValidator, EmailValidator as DjangoEmailValidator

    def query_mxs(self, domain):

        except dns.resolver.NXDOMAIN:

            return []

        except dns.resolver.NoAnswer:

            pass

            pass

        for record_type in ('AAAA', 'A'):

                mxs = dns.resolver.query(domain, record_type)

                mxs = [str(mx.address).rstrip('.') for mx in mxs]

                return mxs

            except dns.resolver.NXDOMAIN:

            except dns.resolver.NoAnswer:

                pass

            except dns.exception.DNSException:

    def check_rcpt(self, value, mxs):

        for server in mxs:

            try:

                smtp = smtplib.SMTP()

                smtp.connect(server)

                status = smtp.helo()

                if status[0] != 250:

                    continue

                smtp.mail('')

                status = smtp.rcpt(value)

                if status[0] // 100 == 5:

                    raise ValidationError(_('Invalid email address.'), code='rcpt-check-failed')

                break

            except smtplib.SMTPServerDisconnected:

                continue

            except smtplib.SMTPConnectError:

                continue

    def __call__(self, value):

        DjangoEmailValidator()(value)

        localpart, hostname = value.split('@', 1)

        if app_settings.A2_VALIDATE_EMAIL_DOMAIN:

            mxs = self.query_mxs(hostname)

            if not mxs:

                raise ValidationError(_('Email domain is invalid'), code='invalid-domain')

            if self.rcpt_check and app_settings.A2_VALIDATE_EMAIL:

                self.check_rcpt(value, mxs)

    monkeypatch.setattr(EmailValidator, 'query_mxs', lambda x, y: [])

    monkeypatch.setattr(EmailValidator, 'query_mxs', lambda x, y: ['mx1.entrouvert.org'])

    monkeypatch.setattr(EmailValidator, 'query_mxs', lambda x, y: ['mx1.entrouvert.org'])

    monkeypatch.setattr(EmailValidator, 'query_mxs', lambda x, y: [])

# -*- coding: utf-8 -*-

# authentic2 - versatile identity manager

# Copyright (C) 2010-2019 Entr'ouvert

#

# This program is free software: you can redistribute it and/or modify it

# under the terms of the GNU Affero General Public License as published

# by the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Affero General Public License for more details.

#

# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from __future__ import unicode_literals

import smtplib

import mock

import pytest

from django.core.exceptions import ValidationError

from authentic2.validators import validate_password, EmailValidator

def test_validate_password():

    with pytest.raises(ValidationError):

        validate_password('aaaaaZZZZZZ')

    with pytest.raises(ValidationError):

        validate_password('00000aaaaaa')

    with pytest.raises(ValidationError):

        validate_password('00000ZZZZZZ')

    validate_password('000aaaaZZZZ')

def test_digits_password_policy(settings):

    settings.A2_PASSWORD_POLICY_REGEX = '^[0-9]{8}$'

    settings.A2_PASSWORD_POLICY_REGEX_ERROR_MSG = 'pasbon'

    settings.A2_PASSWORD_POLICY_MIN_LENGTH = 0

    settings.A2_PASSWORD_POLICY_MIN_CLASSES = 0

    with pytest.raises(ValidationError):

        validate_password('aaa')

    validate_password('12345678')

def test_email_validator():

    with pytest.raises(ValidationError):

        EmailValidator()('nok')

    with pytest.raises(ValidationError):

        EmailValidator()('@nok.com')

    with pytest.raises(ValidationError):

        EmailValidator()('foo@bar\x00')

    EmailValidator()('ok@ok.com')

def test_email_validator_domain(settings):

    settings.A2_VALIDATE_EMAIL_DOMAIN = True

    with mock.patch('authentic2.validators.EmailValidator.query_mxs', return_value=[]) as query_mxs:

        with pytest.raises(ValidationError):

            EmailValidator()('ok@ok.com')

    assert query_mxs.call_count == 1

    with mock.patch('authentic2.validators.EmailValidator.query_mxs', return_value=['ok']) as query_mxs:

        EmailValidator()('ok@ok.com')

    assert query_mxs.call_count == 1

@pytest.fixture

def smtp():

    smtp = mock.Mock()

    smtp.helo.return_value = 250, None

    with mock.patch('smtplib.SMTP', return_value=smtp):

        yield smtp

def test_email_validator_rcpt_check(settings, smtp):

    settings.A2_VALIDATE_EMAIL_DOMAIN = True

    settings.A2_VALIDATE_EMAIL = True

    validator = EmailValidator(rcpt_check=True)

    with mock.patch('authentic2.validators.EmailValidator.query_mxs', return_value=['ok']):

        smtp.rcpt.return_value = 100, None

        validator('ok@ok.com')

        smtp.rcpt.return_value = 500, None

        with pytest.raises(ValidationError):

            validator('ok@ok.com')

        smtp.rcpt.return_value = 100, None

        smtp.rcpt.side_effect = smtplib.SMTPServerDisconnected

        validator('ok@ok.com')

        smtp.rcpt.return_value = 100, None

        smtp.connect.side_effect = smtplib.SMTPConnectError(1,2)

        validator('ok@ok.com')

    assert smtp.connect.call_count == 4

    assert smtp.rcpt.call_count == 3