Produits Entr'ouvert » Combo

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

from __future__ import unicode_literals

from django.db import migrations, models

import jsonfield.fields

class Migration(migrations.Migration):

    dependencies = [

        ('lingo', '0028_tipipaymentformcell'),

    ]

    operations = [

        migrations.AlterModelOptions(

            name='basketitem',

            options={'ordering': ['regie', 'extra_fee', 'subject']},

        ),

        migrations.AddField(

            model_name='basketitem',

            name='extra_fee',

            field=models.BooleanField(default=False),

        ),

        migrations.AddField(

            model_name='basketitem',

            name='request_data',

            field=jsonfield.fields.JSONField(default=dict, blank=True),

        ),

        migrations.AddField(

            model_name='regie',

            name='extra_fees_ws_url',

            field=models.URLField(verbose_name='Webservice URL to compute extra fees', blank=True),

        ),

    ]

    extra_fees_ws_url = models.URLField(_('Webservice URL to compute extra fees'),

            blank=True)

    def compute_extra_fees(self, user):

        if not self.extra_fees_ws_url:

            return

        post_data = {'data': []}

        basketitems = BasketItem.objects.filter(

                user=user, regie=self,

                cancellation_date__isnull=True,

                payment_date__isnull=True)

        for basketitem in basketitems.filter(extra_fee=False):

            basketitem_data = {

                'subject': basketitem.subject,

                'source_url': basketitem.source_url,

                'details': basketitem.details,

                'amount': basketitem.amount,

                'request_data': basketitem.request_data

            }

            post_data['data'].append(basketitem_data)

        if not post_data['data']:

            basketitems.filter(extra_fee=True).delete()

            return

        response = requests.post(self.extra_fees_ws_url, remote_service='auto')

        if response.status_code != 200 or response.json().get('err'):

            logger = logging.getLogger(__name__)

            logger.error('failed to compute extra frees (user: %r)', user)

            return

        basketitems.filter(extra_fee=True).delete()

        for extra_fee in response.json().get('data'):

            BasketItem(user=user, regie=self,

                    subject=extra_fee.get('subject'),

                    amount=extra_fee.get('amount'),

                    extra_fee=True,

                    user_cancellable=False).save()

    request_data = JSONField(blank=True)

    extra_fee = models.BooleanField(default=False)

    class Meta:

        ordering = ['regie', 'extra_fee', 'subject']

        if not self.source_url:

            return

        self.regie.compute_extra_fees(user=self.user)

        self.regie.compute_extra_fees(user=self.user)

  <li><a {% if item.source_url %}href="{{ item.source_url }}{% endif %}">{{ item.subject }}</a>: {{ item.amount }} €

        if 'extra' in request_body:

            item.request_data = request_body.get('extra')

        else:

            item.request_data = request_body

        item.regie.compute_extra_fees(user=item.user)

                regie.compute_extra_fees(user=self.request.user)

        regie.compute_extra_fees(user=transaction.user)

    data['extra'] = {'amount': '22.24', 'foo': 'bar'}

    assert BasketItem.objects.filter(amount=Decimal('22.24'))[0].request_data == data['extra']

def test_extra_fees(key, regie, user):

    regie.extra_fees_ws_url = 'http://www.example.net/extra-fees'

    regie.save()

    user_email = 'foo@example.com'

    User.objects.get_or_create(email=user_email)

    amount = 42

    data = {'amount': amount, 'display_name': 'test amount'}

    with mock.patch('combo.utils.RequestsSession.request') as request:

        mock_json = mock.Mock()

        mock_json.status_code = 200

        mock_json.json.return_value = {'err': 0, 'data': [{'subject': 'Extra Fees', 'amount': '5'}]}

        request.return_value = mock_json

        url = sign_url('%s?email=%s&orig=wcs' % (reverse('api-add-basket-item'), user_email), key)

        resp = client.post(url, json.dumps(data), content_type='application/json')

    assert resp.status_code == 200

    assert json.loads(resp.content)['result'] == 'success'

    assert BasketItem.objects.filter(amount=amount).exists()

    assert BasketItem.objects.filter(amount=amount)[0].regie_id == regie.id

    assert BasketItem.objects.filter(amount=5, extra_fee=True).exists()

    assert BasketItem.objects.filter(amount=5, extra_fee=True)[0].regie_id == regie.id

    with mock.patch('combo.utils.RequestsSession.request') as request:

        mock_json = mock.Mock()

        mock_json.status_code = 200

        mock_json.json.return_value = {'err': 0, 'data': [{'subject': 'Extra Fees', 'amount': '7'}]}

        request.return_value = mock_json

        data['amount'] = 43

        url = sign_url('%s?email=%s&orig=wcs' % (reverse('api-add-basket-item'), user_email), key)

        resp = client.post(url, json.dumps(data), content_type='application/json')

    assert resp.status_code == 200

    assert json.loads(resp.content)['result'] == 'success'

    assert not BasketItem.objects.filter(amount=5, extra_fee=True).exists()

    assert BasketItem.objects.filter(amount=7, extra_fee=True).exists()

    with mock.patch('combo.utils.RequestsSession.request') as request:

        mock_json = mock.Mock()

        mock_json.status_code = 200

        mock_json.json.return_value = {'err': 0, 'data': [{'subject': 'Extra Fees', 'amount': '4'}]}

        request.return_value = mock_json

        url = sign_url('%s?email=%s&orig=wcs' % (reverse('api-remove-basket-item'), user_email), key)

        data = {'basket_item_id': BasketItem.objects.get(amount=43).id}

        resp = client.post(url, json.dumps(data), content_type='application/json')

    assert resp.status_code == 200

    assert not BasketItem.objects.filter(amount=7, extra_fee=True).exists()

    assert BasketItem.objects.filter(amount=4, extra_fee=True).exists()

    # test payment

    login()

    with mock.patch('combo.utils.RequestsSession.request') as request:

        mock_json = mock.Mock()

        mock_json.status_code = 200

        mock_json.json.return_value = {'err': 0, 'data': [{'subject': 'Extra Fees', 'amount': '2'}]}

        request.return_value = mock_json

        resp = client.post(reverse('lingo-pay'), {'regie': regie.pk})

    assert resp.status_code == 302

    location = resp.get('location')

    parsed = urlparse.urlparse(location)

    qs = urlparse.parse_qs(parsed.query)

    transaction_id = qs['transaction_id'][0]

    data = {'transaction_id': transaction_id, 'signed': True,

            'amount': qs['amount'][0], 'ok': True}

    assert data['amount'] == '44.00'

    # call callback with GET

    callback_url = reverse('lingo-callback', kwargs={'regie_pk': regie.id})

    resp = client.get(callback_url, data)

    assert resp.status_code == 200

    assert Transaction.objects.get(order_id=transaction_id).status == 3