Produits Entr'ouvert » Corbo

class SendTestSMSForm(forms.Form):

    mobile = forms.CharField(label=_('Mobile number'))

    subscriptions_import, view_announce, email_announce, sms_announce, \

    menu_json

            url(r'^announce/sms/(?P<pk>\d+)/$', sms_announce,

                name='sms_announce'),

{% if sms_enabled %}

<a href="{% url 'sms_announce' pk=object.pk %}" rel="popup">{% trans 'Send test SMS' %}</a>

{% endif %}

{% extends "corbo/manage.html" %}

{% load i18n static %}

{% block breadcrumb %}

{{ block.super }}

<a href="{% url "view_announce" pk=object.pk %}">{{ object.title }}</a>

{% endblock %}

{% block appbar %}

<h2>{% trans "Send test SMS" %}</h2>

{% endblock %}

{% block content %}

<form method="post">

  {% csrf_token %}

  {{ form.as_p }}

  <div class="buttons">

    <button class="submit-button">{% trans "Send" %}</button>

    <a href="{% url "view_announce" pk=object.pk %}" class="cancel">{% trans "Cancel" %}</a>

  </div>

</form>

{% endblock %}

    SendTestEmailForm, SendTestSMSForm

        context['sms_enabled'] = settings.SMS_GATEWAY_URL

class SendAnnounceView(FormView):

        return {'email': self.request.user.email,

                'mobile': self.request.session.get('mellon_session', {}).get('mobile', '')}

    def get_context_data(self, **kwargs):

        context = super(SendAnnounceView, self).get_context_data(**kwargs)

        context['object'] = models.Announce.objects.get(pk=self.kwargs['pk'])

        return context

class EmailAnnounceView(SendAnnounceView):

    form_class = SendTestEmailForm

    template_name = 'corbo/email_test_announce_form.html'

class SMSAnnounceView(SendAnnounceView):

    form_class = SendTestSMSForm

    template_name = 'corbo/sms_test_announce_form.html'

    def form_valid(self, form):

        mobile = form.cleaned_data['mobile']

        announce = models.Announce.objects.get(pk=self.kwargs['pk'])

        if (utils.send_sms(announce.text, [mobile])):

            messages.info(self.request, _('SMS successfully sent'))

        else:

            messages.error(self.request, _('Error occured while sending SMS'))

        return super(SMSAnnounceView, self).form_valid(form)

sms_announce = SMSAnnounceView.as_view()

import logging

import mock

from django.test import override_settings

def test_email_announce(app, admin_user):

    assert 'Send test SMS' not in resp.content

@mock.patch('corbo.utils.requests.post')

def test_sms_announce(mocked_post, app, admin_user, settings):

    app = login(app)

    resp = app.get('/manage/')

    assert 'New category' in resp.content

    category_page = resp.click('New category')

    category_form = category_page.forms[0]

    category_form['name'] = 'Alerts'

    resp = category_form.submit()

    assert resp.status_int == 302

    assert resp.location == 'http://testserver/manage/'

    resp = resp.follow()

    resp = resp.click('Alerts')

    assert 'New announce' in resp.content

    announce_page = resp.click('New announce')

    announce_form = announce_page.forms[0]

    announce_form['title'] = 'First announce'

    announce_form['text'] = 'announce content'

    resp = announce_form.submit()

    assert resp.status_int == 302

    assert resp.location == 'http://testserver/manage/category/alerts/'

    resp = resp.follow()

    assert 'First announce' in resp.content

    settings.SMS_GATEWAY_URL='http:/passerelle.com'

    resp = resp.click('First announce')

    assert 'Send test SMS' in resp.content

    resp = resp.click('Send test SMS')

    send_form = resp.forms[0]

    assert 'mobile' in send_form.fields

    assert send_form.fields['mobile'][0].value == ''

    resp = send_form.submit()

    assert resp.status_int == 200

    form = resp.forms[0]

    form['mobile'] = '0607080900'

    mocked_response = mock.Mock()

    mocked_response.json.return_value = {'err': 0, 'data': True}

    mocked_post.return_value = mocked_response

    resp = form.submit()

    assert resp.location == 'http://testserver/manage/announce/1/'

    resp = resp.follow()

    assert 'SMS successfully sent' in resp.content

    resp = resp.click('Send test SMS')

    form = resp.forms[0]

    form['mobile'] = '0607080900'

    mocked_response.json.return_value = {'err': 1, 'data': None, 'err_desc': 'Destination error'}

    resp = form.submit()

    resp = resp.follow()

    assert 'Error occured while sending SMS' in resp.content

def test_sms_announce_with_invalid_gateway_url(app, admin_user, settings, caplog):

    app = login(app)

    resp = app.get('/manage/')

    assert 'New category' in resp.content

    category_page = resp.click('New category')

    category_form = category_page.forms[0]

    category_form['name'] = 'Alerts'

    resp = category_form.submit()

    resp = resp.follow()

    resp = resp.click('Alerts')

    assert 'New announce' in resp.content

    announce_page = resp.click('New announce')

    announce_form = announce_page.forms[0]

    announce_form['title'] = 'First announce'

    announce_form['text'] = 'announce content'

    resp = announce_form.submit()

    assert resp.status_int == 302

    assert resp.location == 'http://testserver/manage/category/alerts/'

    resp = resp.follow()

    assert 'First announce' in resp.content

    settings.SMS_GATEWAY_URL='invalid_url'

    resp = resp.click('First announce')

    assert 'Send test SMS' in resp.content

    resp = resp.click('Send test SMS')

    form = resp.forms[0]

    form['mobile'] = '0607080900'

    resp = form.submit()

    records = caplog.records

    assert len(records) == 1

    for record in records:

        assert record.name == 'corbo.utils'

        assert record.levelno == logging.WARNING

        assert 'Invalid URL' in record.getMessage()