Produits Entr'ouvert » Chrono

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

from __future__ import unicode_literals

from django.db import migrations, models

class Migration(migrations.Migration):

    dependencies = [

        ('agendas', '0019_timeperiodexception'),

    ]

    operations = [

        migrations.AddField(

            model_name='desk',

            name='timeperiod_exceptions_remote_url',

            field=models.URLField(null=True, verbose_name='URL to fetch time period exceptions from', blank=True),

        ),

    ]

    timeperiod_exceptions_remote_url = models.URLField(_('URL to fetch time period exceptions from'),

                                    null=True, blank=True)

import requests

class DeskForm(forms.ModelForm):

        exclude = []

    def clean_timeperiod_exceptions_remote_url(self):

        if not self.cleaned_data['timeperiod_exceptions_remote_url']:

            return

        try:

            response = requests.get(self.cleaned_data['timeperiod_exceptions_remote_url'])

            response.raise_for_status()

        except requests.exceptions.HTTPError as e:

            raise ValidationError(_('Error %s is returned while trying to get file.') % e.response.status_code)

        except requests.exceptions.ConnectionError:

            raise ValidationError(_('URL is unreachable.'))

class NewDeskForm(DeskForm):

        exclude = ['slug']

# chrono - agendas system

# Copyright (C) 2016-2017  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/>.

import sys

import logging

from django.core.management.base import BaseCommand

from chrono.agendas.models import Desk, ICSError

class Command(BaseCommand):

    help = 'Synchronize time period exceptions from desks remote ics'

    def handle(self, **options):

        logger = logging.getLogger(__name__)

        for desk in Desk.objects.filter(timeperiod_exceptions_remote_url__isnull=False):

            try:

                desk.create_timeperiod_exceptions_from_remote_ics(desk.timeperiod_exceptions_remote_url)

            except ICSError as e:

                logger.warning(u'unable to create timeperiod exceptions for "%s": %s' % (desk, e))

import logging

from django.core.management import call_command

@mock.patch('chrono.agendas.models.requests.get')

def test_sync_desks_timeperiod_exceptions_from_ics(mocked_get, caplog):

    agenda = Agenda(label=u'Test 11 agenda')

    agenda.save()

    desk = Desk(label='Test 11 desk', agenda=agenda, timeperiod_exceptions_remote_url='http:example.com/sample.ics')

    desk.save()

    mocked_response = mock.Mock()

    mocked_response.status_code = 403

    mocked_get.return_value = mocked_response

    def mocked_requests_http_forbidden_error(*args, **kwargs):

        raise requests.exceptions.HTTPError(response=mocked_response)

    mocked_get.side_effect = mocked_requests_http_forbidden_error

    call_command('sync_desks_timeperiod_exceptions')

    records = caplog.records()

    assert len(records) == 1

    for record in records:

        assert record.name == 'chrono.manager.management.commands.sync_desks_timeperiod_exceptions'

        assert record.levelno == logging.WARNING

        assert record.getMessage() == 'unable to create timeperiod exceptions for "Test 11 desk": Error 403 is returned while trying to get file.'

@mock.patch('chrono.manager.forms.requests.get')

def test_meetings_agenda_add_desk_with_non_existing_exceptions_url(mocked_get, app, admin_user):

    app = login(app)

    resp = app.get('/manage/', status=200)

    resp = resp.click('New')

    resp.form['label'] = 'Foo bar'

    resp.form['kind'] = 'meetings'

    resp = resp.form.submit()

    agenda = Agenda.objects.get(slug='foo-bar')

    resp = app.get('/manage/agendas/%s/' % agenda.id, status=200)

    mocked_response = mock.Mock()

    mocked_response.status_code = 403

    mocked_get.return_value = mocked_response

    def mocked_requests_http_forbidden_error(*args, **kwargs):

        raise requests.exceptions.HTTPError(response=mocked_response)

    mocked_get.side_effect = mocked_requests_http_forbidden_error

    resp = resp.click('New Desk')

    resp.form['label'] = 'Desk A'

    resp.form['timeperiod_exceptions_remote_url'] = 'http://nowhere.com/unknown.ics'

    resp = resp.form.submit(status=200)

    assert 'Error 403 is returned while trying to get file.' in resp.text

@mock.patch('chrono.manager.forms.requests.get')

def test_meetings_agenda_add_desk_with_unreachable_exceptions_url(mocked_get, app, admin_user):

    app = login(app)

    resp = app.get('/manage/', status=200)

    resp = resp.click('New')

    resp.form['label'] = 'Foo bar'

    resp.form['kind'] = 'meetings'

    resp = resp.form.submit()

    agenda = Agenda.objects.get(slug='foo-bar')

    resp = app.get('/manage/agendas/%s/' % agenda.id, status=200)

    mocked_response = mock.Mock()

    mocked_get.return_value = mocked_response

    def mocked_requests_connection_error(*args, **kwargs):

        raise requests.exceptions.ConnectionError('unreachable')

    mocked_get.side_effect = mocked_requests_connection_error

    resp = resp.click('New Desk')

    resp.form['label'] = 'Desk A'

    resp.form['timeperiod_exceptions_remote_url'] = 'http://nowhere.com/unknown.ics'

    resp = resp.form.submit(status=200)

    assert 'URL is unreachable.' in resp.text