summaryrefslogtreecommitdiff
path: root/addons/payment_test/models/payment_acquirer.py
blob: 965d520192fdc5a9b540c17d488d19a134d3baa9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# coding: utf-8
# Part of Odoo. See LICENSE file for full copyright and licensing details.

from datetime import datetime
from uuid import uuid4

from odoo import api, exceptions, fields, models, _


class PaymentAcquirerTest(models.Model):
    _inherit = 'payment.acquirer'

    provider = fields.Selection(selection_add=[
        ('test', 'Test')
    ], ondelete={'test': 'set default'})

    @api.model
    def create(self, values):
        if values.get('provider') == 'test' and 'state' in values and values.get('state') not in ('test', 'disabled'):
            raise exceptions.UserError(_('This acquirer should not be used for other purposes than testing.'))
        return super(PaymentAcquirerTest, self).create(values)

    def write(self, values):
        if any(rec.provider == 'test' for rec in self) and 'state' in values and values.get('state') not in ('test', 'disabled'):
            raise exceptions.UserError(_('This acquirer should not be used for other purposes than testing.'))
        return super(PaymentAcquirerTest, self).write(values)

    @api.model
    def test_s2s_form_process(self, data):
        """ Return a minimal token to allow proceeding to transaction creation. """
        payment_token = self.env['payment.token'].sudo().create({
            'acquirer_ref': uuid4(),
            'acquirer_id': int(data['acquirer_id']),
            'partner_id': int(data['partner_id']),
            'name': 'Test - XXXXXXXXXXXX%s - %s' % (data['cc_number'][-4:], data['cc_holder_name'])
        })
        return payment_token


class PaymentTransactionTest(models.Model):
    _inherit = 'payment.transaction'

    def test_create(self, values):
        """Automatically set the transaction as successful upon creation. """
        return {'date': datetime.now(), 'state': 'done'}

    def test_s2s_do_transaction(self, **kwargs):
        self.execute_callback()