partner-contact/partner_revision/tests/test_revision_flow.py

# -*- coding: utf-8 -*-
#
#
#    Authors: Guewen Baconnier
#    Copyright 2015 Camptocamp SA
#
#    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 datetime import datetime, timedelta

from openerp import fields, exceptions
from openerp.tests import common
from .common import ChangesetMixin


class TestChangesetFlow(ChangesetMixin, common.TransactionCase):
    """ Check how changeset are generated and applied based on the rules.

    We do not really care about the types of the fields in this test
    suite, so we only use 'char' fields.  We have to ensure that the
    general changeset flows work as expected, that is:

    * create a 'done' changeset when a manual/system write is made on partner
    * create a changeset according to the changeset rules when the key
      '__changeset_rules' is passed in the context
    * apply a changeset change writes the value on the partner
    * apply a whole changeset writes all the changes' values on the partner
    * changes in state 'cancel' or 'done' do not write on the partner
    * when all the changes are either 'cancel' or 'done', the changeset
      becomes 'done'
    """

    def _setup_rules(self):
        ChangesetFieldRule = self.env['changeset.field.rule']
        partner_model_id = self.env.ref('base.model_res_partner').id
        self.field_name = self.env.ref('base.field_res_partner_name')
        self.field_street = self.env.ref('base.field_res_partner_street')
        self.field_street2 = self.env.ref('base.field_res_partner_street2')
        ChangesetFieldRule.create({
            'model_id': partner_model_id,
            'field_id': self.field_name.id,
            'action': 'auto',
        })
        ChangesetFieldRule.create({
            'model_id': partner_model_id,
            'field_id': self.field_street.id,
            'action': 'validate',
        })
        ChangesetFieldRule.create({
            'model_id': partner_model_id,
            'field_id': self.field_street2.id,
            'action': 'never',
        })

    def setUp(self):
        super(TestChangesetFlow, self).setUp()
        self._setup_rules()
        self.partner = self.env['res.partner'].create({
            'name': 'X',
            'street': 'street X',
            'street2': 'street2 X',
        })

    def test_new_changeset(self):
        """ Add a new changeset on a partner

        A new changeset is created when we write on a partner with
        ``__changeset_rules`` in the context.
        """
        self.partner.with_context(__changeset_rules=True).write({
            'name': 'Y',
            'street': 'street Y',
            'street2': 'street2 Y',
        })
        self.assert_changeset(
            self.partner,
            [(self.field_name, 'X', 'Y', 'done'),
             (self.field_street, 'street X', 'street Y', 'draft'),
             (self.field_street2, 'street2 X', 'street2 Y', 'cancel'),
             ],
        )
        self.assertEqual(self.partner.name, 'Y')
        self.assertEqual(self.partner.street, 'street X')
        self.assertEqual(self.partner.street2, 'street2 X')

    def test_new_changeset_empty_value(self):
        """ Create a changeset change that empty a value """
        self.partner.with_context(__changeset_rules=True).write({
            'street': False,
        })
        self.assert_changeset(
            self.partner,
            [(self.field_street, 'street X', False, 'draft')]
        )

    def test_manual_edition(self):
        """ A manual edition of a partner should always be applied

        But should create a 'done' changeset
        """
        self.partner.write({
            'name': 'Y',
            'street': 'street Y',
            'street2': 'street2 Y',
        })
        self.assert_changeset(
            self.partner,
            [(self.field_name, 'X', 'Y', 'done'),
             (self.field_street, 'street X', 'street Y', 'done'),
             (self.field_street2, 'street2 X', 'street2 Y', 'done'),
             ],
        )
        self.assertEqual(self.partner.name, 'Y')
        self.assertEqual(self.partner.street, 'street Y')
        self.assertEqual(self.partner.street2, 'street2 Y')

    def test_apply_change(self):
        """ Apply a changeset change on a partner """
        changes = [
            (self.field_name, 'Y', 'draft'),
        ]
        changeset = self._create_changeset(self.partner, changes)
        changeset.change_ids.apply()
        self.assertEqual(self.partner.name, 'Y')
        self.assertEqual(changeset.change_ids.state, 'done')

    def test_apply_done_change(self):
        """ Done changes do not apply (already applied) """
        changes = [
            (self.field_name, 'Y', 'done'),
        ]
        changeset = self._create_changeset(self.partner, changes)
        changeset.change_ids.apply()
        self.assertEqual(self.partner.name, 'X')

    def test_apply_cancel_change(self):
        """ Cancel changes do not apply """
        changes = [
            (self.field_name, 'Y', 'cancel'),
        ]
        changeset = self._create_changeset(self.partner, changes)
        changeset.change_ids.apply()
        self.assertEqual(self.partner.name, 'X')

    def test_apply_empty_value(self):
        """ Apply a change that empty a value """
        changes = [
            (self.field_street, False, 'draft'),
        ]
        changeset = self._create_changeset(self.partner, changes)
        changeset.change_ids.apply()
        self.assertFalse(self.partner.street)

    def test_apply_change_loop(self):
        """ Test @api.multi on the changes """
        changes = [
            (self.field_name, 'Y', 'draft'),
            (self.field_street, 'street Y', 'draft'),
            (self.field_street2, 'street2 Y', 'draft'),
        ]
        changeset = self._create_changeset(self.partner, changes)
        changeset.change_ids.apply()
        self.assertEqual(self.partner.name, 'Y')
        self.assertEqual(self.partner.street, 'street Y')
        self.assertEqual(self.partner.street2, 'street2 Y')

    def test_apply(self):
        """ Apply a full changeset on a partner """
        changes = [
            (self.field_name, 'Y', 'draft'),
            (self.field_street, 'street Y', 'draft'),
            (self.field_street2, 'street2 Y', 'draft'),
        ]
        changeset = self._create_changeset(self.partner, changes)
        changeset.apply()
        self.assertEqual(self.partner.name, 'Y')
        self.assertEqual(self.partner.street, 'street Y')
        self.assertEqual(self.partner.street2, 'street2 Y')

    def test_changeset_state_on_done(self):
        """ Check that changeset state becomes done when changes are done """
        changes = [(self.field_name, 'Y', 'draft')]
        changeset = self._create_changeset(self.partner, changes)
        self.assertEqual(changeset.state, 'draft')
        changeset.change_ids.apply()
        self.assertEqual(changeset.state, 'done')

    def test_changeset_state_on_cancel(self):
        """ Check that rev. state becomes done when changes are canceled """
        changes = [(self.field_name, 'Y', 'draft')]
        changeset = self._create_changeset(self.partner, changes)
        self.assertEqual(changeset.state, 'draft')
        changeset.change_ids.cancel()
        self.assertEqual(changeset.state, 'done')

    def test_changeset_state(self):
        """ Check that changeset state becomes done with multiple changes """
        changes = [
            (self.field_name, 'Y', 'draft'),
            (self.field_street, 'street Y', 'draft'),
            (self.field_street2, 'street2 Y', 'draft'),
        ]
        changeset = self._create_changeset(self.partner, changes)
        self.assertEqual(changeset.state, 'draft')
        changeset.apply()
        self.assertEqual(changeset.state, 'done')

    def test_apply_changeset_with_other_pending(self):
        """ Error when applying when previous pending changesets exist """
        changes = [(self.field_name, 'Y', 'draft')]
        old_changeset = self._create_changeset(self.partner, changes)
        # if the date is the same, both changeset can be applied
        to_string = fields.Datetime.to_string
        old_changeset.date = to_string(datetime.now() - timedelta(days=1))
        changes = [(self.field_name, 'Z', 'draft')]
        changeset = self._create_changeset(self.partner, changes)
        with self.assertRaises(exceptions.Warning):
            changeset.change_ids.apply()

    def test_apply_different_changesets(self):
        """ Apply different changesets at once """
        partner2 = self.env['res.partner'].create({'name': 'P2'})
        changes = [
            (self.field_name, 'Y', 'draft'),
            (self.field_street, 'street Y', 'draft'),
            (self.field_street2, 'street2 Y', 'draft'),
        ]
        changeset = self._create_changeset(self.partner, changes)
        changeset2 = self._create_changeset(partner2, changes)
        self.assertEqual(changeset.state, 'draft')
        self.assertEqual(changeset2.state, 'draft')
        (changeset + changeset2).apply()
        self.assertEqual(self.partner.name, 'Y')
        self.assertEqual(self.partner.street, 'street Y')
        self.assertEqual(self.partner.street2, 'street2 Y')
        self.assertEqual(partner2.name, 'Y')
        self.assertEqual(partner2.street, 'street Y')
        self.assertEqual(partner2.street2, 'street2 Y')
        self.assertEqual(changeset.state, 'done')
        self.assertEqual(changeset2.state, 'done')