server/test/unittest_hook.py
author Aurelien Campeas <aurelien.campeas@logilab.fr>
Tue, 03 Jun 2014 12:02:51 +0200
changeset 9758 91f3cb7db2aa
parent 9548 be001628edad
permissions -rw-r--r--
[test/hook] use the new connection API
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
b97547f5f1fa Showtime !
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
diff changeset
     1
# -*- coding: utf-8 -*-
9548
be001628edad [schema] fix composite deletion handling
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 8694
diff changeset
     2
# copyright 2003-2014 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
5421
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
     3
# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
     4
#
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
     5
# This file is part of CubicWeb.
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
     6
#
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
     7
# CubicWeb is free software: you can redistribute it and/or modify it under the
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
     8
# terms of the GNU Lesser General Public License as published by the Free
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
     9
# Software Foundation, either version 2.1 of the License, or (at your option)
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
    10
# any later version.
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
    11
#
5424
8ecbcbff9777 replace logilab-common by CubicWeb in disclaimer
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5421
diff changeset
    12
# CubicWeb is distributed in the hope that it will be useful, but WITHOUT
5421
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
    13
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
    14
# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
    15
# details.
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
    16
#
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
    17
# You should have received a copy of the GNU Lesser General Public License along
8167de96c523 proper licensing information (LGPL-2.1). Hope I get it right this time.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5093
diff changeset
    18
# with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.
5886
00a78298d30d cleanups
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5424
diff changeset
    19
"""unit/functional tests for cubicweb.server.hook"""
0
b97547f5f1fa Showtime !
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
diff changeset
    20
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    21
from logilab.common.testlib import TestCase, unittest_main, mock_object
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    22
7361
6baa4f430c9b [repo test] fix test broken by c2452cd57026 by using proper fake session
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6796
diff changeset
    23
from cubicweb.devtools import TestServerConfiguration, fake
2773
b2530e3e0afb [testlib] #345052 and #344207: major test lib refactoring/cleanup + update usage
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 2637
diff changeset
    24
from cubicweb.devtools.testlib import CubicWebTC
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    25
from cubicweb.server import hook
3413
5cd27d2cc64f update imports, rename test module
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 2968
diff changeset
    26
from cubicweb.hooks import integrity, syncschema
0
b97547f5f1fa Showtime !
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
diff changeset
    27
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    28
class OperationsTC(CubicWebTC):
1787
71c143c0ada3 fix test
sylvain.thenault@logilab.fr
parents: 1398
diff changeset
    29
0
b97547f5f1fa Showtime !
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
diff changeset
    30
    def setUp(self):
2773
b2530e3e0afb [testlib] #345052 and #344207: major test lib refactoring/cleanup + update usage
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 2637
diff changeset
    31
        CubicWebTC.setUp(self)
0
b97547f5f1fa Showtime !
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
diff changeset
    32
        self.hm = self.repo.hm
1787
71c143c0ada3 fix test
sylvain.thenault@logilab.fr
parents: 1398
diff changeset
    33
0
b97547f5f1fa Showtime !
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
diff changeset
    34
    def test_late_operation(self):
9758
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    35
        with self.admin_access.repo_cnx() as cnx:
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    36
            l1 = hook.LateOperation(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    37
            l2 = hook.LateOperation(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    38
            l3 = hook.Operation(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    39
            self.assertEqual(cnx.pending_operations, [l3, l1, l2])
1787
71c143c0ada3 fix test
sylvain.thenault@logilab.fr
parents: 1398
diff changeset
    40
0
b97547f5f1fa Showtime !
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
diff changeset
    41
    def test_single_last_operation(self):
9758
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    42
        with self.admin_access.repo_cnx() as cnx:
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    43
            l0 = hook.SingleLastOperation(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    44
            l1 = hook.LateOperation(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    45
            l2 = hook.LateOperation(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    46
            l3 = hook.Operation(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    47
            self.assertEqual(cnx.pending_operations, [l3, l1, l2, l0])
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    48
            l4 = hook.SingleLastOperation(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    49
            self.assertEqual(cnx.pending_operations, [l3, l1, l2, l4])
1787
71c143c0ada3 fix test
sylvain.thenault@logilab.fr
parents: 1398
diff changeset
    50
0
b97547f5f1fa Showtime !
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
diff changeset
    51
    def test_global_operation_order(self):
9758
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    52
        with self.admin_access.repo_cnx() as cnx:
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    53
            op1 = syncschema.RDefDelOp(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    54
            op2 = integrity._CheckORelationOp(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    55
            op3 = syncschema.MemSchemaNotifyChanges(cnx)
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
    56
            self.assertEqual([op1, op2, op3], cnx.pending_operations)
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    57
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    58
class HookCalled(Exception): pass
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    59
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    60
config = TestServerConfiguration('data')
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    61
config.bootstrap_cubes()
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    62
schema = config.load_schema()
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    63
6781
5062d86d6ffe [unittest2] use unittest2 module fixture api
Julien Jehannet <julien.jehannet@logilab.fr>
parents: 6340
diff changeset
    64
def tearDownModule(*args):
4766
162b2b127b15 [test] get a chance to get proper garbage collection when running pytest on whole cw
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 4252
diff changeset
    65
    global config, schema
162b2b127b15 [test] get a chance to get proper garbage collection when running pytest on whole cw
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 4252
diff changeset
    66
    del config, schema
162b2b127b15 [test] get a chance to get proper garbage collection when running pytest on whole cw
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 4252
diff changeset
    67
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    68
class AddAnyHook(hook.Hook):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    69
    __regid__ = 'addany'
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    70
    category = 'cat1'
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    71
    events = ('before_add_entity',)
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    72
    def __call__(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    73
        raise HookCalled()
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    74
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    75
5093
8d073d2e089d [optimization] improve massive write performance by optimizing hooks selection
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5029
diff changeset
    76
class HooksRegistryTC(TestCase):
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    77
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    78
    def setUp(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    79
        """ called before each test from this class """
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    80
        self.vreg = mock_object(config=config, schema=schema)
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    81
        self.o = hook.HooksRegistry(self.vreg)
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    82
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    83
    def test_register_bad_hook1(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    84
        class _Hook(hook.Hook):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    85
            events = ('before_add_entiti',)
6796
e70ca9abfc51 [unittest2] update to unittest2 assertRaises api
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6781
diff changeset
    86
        with self.assertRaises(Exception) as cm:
e70ca9abfc51 [unittest2] update to unittest2 assertRaises api
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6781
diff changeset
    87
            self.o.register(_Hook)
e70ca9abfc51 [unittest2] update to unittest2 assertRaises api
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6781
diff changeset
    88
        self.assertEqual(str(cm.exception), 'bad event before_add_entiti on %s._Hook' % __name__)
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    89
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    90
    def test_register_bad_hook2(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    91
        class _Hook(hook.Hook):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    92
            events = None
6796
e70ca9abfc51 [unittest2] update to unittest2 assertRaises api
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6781
diff changeset
    93
        with self.assertRaises(Exception) as cm:
e70ca9abfc51 [unittest2] update to unittest2 assertRaises api
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6781
diff changeset
    94
            self.o.register(_Hook)
e70ca9abfc51 [unittest2] update to unittest2 assertRaises api
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6781
diff changeset
    95
        self.assertEqual(str(cm.exception), 'bad .events attribute None on %s._Hook' % __name__)
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    96
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    97
    def test_register_bad_hook3(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    98
        class _Hook(hook.Hook):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
    99
            events = 'before_add_entity'
6796
e70ca9abfc51 [unittest2] update to unittest2 assertRaises api
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6781
diff changeset
   100
        with self.assertRaises(Exception) as cm:
e70ca9abfc51 [unittest2] update to unittest2 assertRaises api
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6781
diff changeset
   101
            self.o.register(_Hook)
e70ca9abfc51 [unittest2] update to unittest2 assertRaises api
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6781
diff changeset
   102
        self.assertEqual(str(cm.exception), 'bad event b on %s._Hook' % __name__)
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   103
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   104
    def test_call_hook(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   105
        self.o.register(AddAnyHook)
4834
b718626a0e60 move hooks activation control on session object, so we can have a per transaction control. Added a new `hooks_control` context manager for usual modification of hooks activation.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 4766
diff changeset
   106
        dis = set()
7361
6baa4f430c9b [repo test] fix test broken by c2452cd57026 by using proper fake session
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6796
diff changeset
   107
        cw = fake.FakeSession()
6baa4f430c9b [repo test] fix test broken by c2452cd57026 by using proper fake session
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 6796
diff changeset
   108
        cw.is_hook_activated = lambda cls: cls.category not in dis
4834
b718626a0e60 move hooks activation control on session object, so we can have a per transaction control. Added a new `hooks_control` context manager for usual modification of hooks activation.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 4766
diff changeset
   109
        self.assertRaises(HookCalled,
b718626a0e60 move hooks activation control on session object, so we can have a per transaction control. Added a new `hooks_control` context manager for usual modification of hooks activation.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 4766
diff changeset
   110
                          self.o.call_hooks, 'before_add_entity', cw)
b718626a0e60 move hooks activation control on session object, so we can have a per transaction control. Added a new `hooks_control` context manager for usual modification of hooks activation.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 4766
diff changeset
   111
        dis.add('cat1')
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   112
        self.o.call_hooks('before_add_entity', cw) # disabled hooks category, not called
4834
b718626a0e60 move hooks activation control on session object, so we can have a per transaction control. Added a new `hooks_control` context manager for usual modification of hooks activation.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 4766
diff changeset
   113
        dis.remove('cat1')
b718626a0e60 move hooks activation control on session object, so we can have a per transaction control. Added a new `hooks_control` context manager for usual modification of hooks activation.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 4766
diff changeset
   114
        self.assertRaises(HookCalled,
b718626a0e60 move hooks activation control on session object, so we can have a per transaction control. Added a new `hooks_control` context manager for usual modification of hooks activation.
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 4766
diff changeset
   115
                          self.o.call_hooks, 'before_add_entity', cw)
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   116
        self.o.unregister(AddAnyHook)
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   117
        self.o.call_hooks('before_add_entity', cw) # nothing to call
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   118
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   119
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   120
class SystemHooksTC(CubicWebTC):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   121
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   122
    def test_startup_shutdown(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   123
        import hooks # cubicweb/server/test/data/hooks.py
6340
470d8e828fda [test] update test to unittest2 api (still using lgc.testlib though)
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5921
diff changeset
   124
        self.assertEqual(hooks.CALLED_EVENTS['server_startup'], True)
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   125
        # don't actually call repository.shutdown !
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   126
        self.repo.hm.call_hooks('server_shutdown', repo=self.repo)
6340
470d8e828fda [test] update test to unittest2 api (still using lgc.testlib though)
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5921
diff changeset
   127
        self.assertEqual(hooks.CALLED_EVENTS['server_shutdown'], True)
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   128
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   129
    def test_session_open_close(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   130
        import hooks # cubicweb/server/test/data/hooks.py
9758
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
   131
        anonaccess = self.new_access('anon')
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
   132
        with anonaccess.repo_cnx() as cnx:
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
   133
            self.assertEqual(hooks.CALLED_EVENTS['session_open'], 'anon')
91f3cb7db2aa [test/hook] use the new connection API
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents: 9548
diff changeset
   134
        anonaccess.close()
6340
470d8e828fda [test] update test to unittest2 api (still using lgc.testlib though)
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5921
diff changeset
   135
        self.assertEqual(hooks.CALLED_EVENTS['session_close'], 'anon')
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   136
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   137
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   138
# class RelationHookTC(TestCase):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   139
#     """testcase for relation hooks grouping"""
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   140
#     def setUp(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   141
#         """ called before each test from this class """
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   142
#         self.o = HooksManager(schema)
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   143
#         self.called = []
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   144
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   145
#     def test_before_add_relation(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   146
#         """make sure before_xxx_relation hooks are called directly"""
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   147
#         self.o.register(self._before_relation_hook,
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   148
#                              'before_add_relation', 'concerne')
6340
470d8e828fda [test] update test to unittest2 api (still using lgc.testlib though)
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5921
diff changeset
   149
#         self.assertEqual(self.called, [])
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   150
#         self.o.call_hooks('before_add_relation', 'concerne', 'USER',
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   151
#                           1, 'concerne', 2)
6340
470d8e828fda [test] update test to unittest2 api (still using lgc.testlib though)
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5921
diff changeset
   152
#         self.assertEqual(self.called, [(1, 'concerne', 2)])
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   153
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   154
#     def test_after_add_relation(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   155
#         """make sure after_xxx_relation hooks are deferred"""
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   156
#         self.o.register(self._after_relation_hook,
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   157
#                              'after_add_relation', 'concerne')
6340
470d8e828fda [test] update test to unittest2 api (still using lgc.testlib though)
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5921
diff changeset
   158
#         self.assertEqual(self.called, [])
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   159
#         self.o.call_hooks('after_add_relation', 'concerne', 'USER',
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   160
#                           1, 'concerne', 2)
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   161
#         self.o.call_hooks('after_add_relation', 'concerne', 'USER',
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   162
#                           3, 'concerne', 4)
6340
470d8e828fda [test] update test to unittest2 api (still using lgc.testlib though)
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5921
diff changeset
   163
#         self.assertEqual(self.called, [(1, 'concerne', 2), (3, 'concerne', 4)])
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   164
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   165
#     def test_before_delete_relation(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   166
#         """make sure before_xxx_relation hooks are called directly"""
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   167
#         self.o.register(self._before_relation_hook,
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   168
#                              'before_delete_relation', 'concerne')
6340
470d8e828fda [test] update test to unittest2 api (still using lgc.testlib though)
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5921
diff changeset
   169
#         self.assertEqual(self.called, [])
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   170
#         self.o.call_hooks('before_delete_relation', 'concerne', 'USER',
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   171
#                           1, 'concerne', 2)
6340
470d8e828fda [test] update test to unittest2 api (still using lgc.testlib though)
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5921
diff changeset
   172
#         self.assertEqual(self.called, [(1, 'concerne', 2)])
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   173
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   174
#     def test_after_delete_relation(self):
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   175
#         """make sure after_xxx_relation hooks are deferred"""
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   176
#         self.o.register(self._after_relation_hook,
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   177
#                         'after_delete_relation', 'concerne')
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   178
#         self.o.call_hooks('after_delete_relation', 'concerne', 'USER',
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   179
#                           1, 'concerne', 2)
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   180
#         self.o.call_hooks('after_delete_relation', 'concerne', 'USER',
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   181
#                           3, 'concerne', 4)
6340
470d8e828fda [test] update test to unittest2 api (still using lgc.testlib though)
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 5921
diff changeset
   182
#         self.assertEqual(self.called, [(1, 'concerne', 2), (3, 'concerne', 4)])
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   183
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   184
7398
26695dd703d8 [repository api] definitly kill usage of word 'pool' to refer to connections set used by a session
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 7361
diff changeset
   185
#     def _before_relation_hook(self, cnxset, subject, r_type, object):
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   186
#         self.called.append((subject, r_type, object))
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   187
7398
26695dd703d8 [repository api] definitly kill usage of word 'pool' to refer to connections set used by a session
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 7361
diff changeset
   188
#     def _after_relation_hook(self, cnxset, subject, r_type, object):
3414
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   189
#         self.called.append((subject, r_type, object))
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   190
fe6c99b96701 no more HooksManager but a HooksRegistry: update test accordingly
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents: 3413
diff changeset
   191
0
b97547f5f1fa Showtime !
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
diff changeset
   192
if __name__ == '__main__':
b97547f5f1fa Showtime !
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
diff changeset
   193
    unittest_main()