31 def test_hooks_control(self): |
31 def test_hooks_control(self): |
32 session = self.session |
32 session = self.session |
33 self.assertEqual(HOOKS_ALLOW_ALL, session.hooks_mode) |
33 self.assertEqual(HOOKS_ALLOW_ALL, session.hooks_mode) |
34 self.assertEqual(set(), session.disabled_hook_categories) |
34 self.assertEqual(set(), session.disabled_hook_categories) |
35 self.assertEqual(set(), session.enabled_hook_categories) |
35 self.assertEqual(set(), session.enabled_hook_categories) |
36 self.assertEqual(1, len(session._txs)) |
36 self.assertEqual(1, len(session._cnxs)) |
37 with session.deny_all_hooks_but('metadata'): |
37 with session.deny_all_hooks_but('metadata'): |
38 self.assertEqual(HOOKS_DENY_ALL, session.hooks_mode) |
38 self.assertEqual(HOOKS_DENY_ALL, session.hooks_mode) |
39 self.assertEqual(set(), session.disabled_hook_categories) |
39 self.assertEqual(set(), session.disabled_hook_categories) |
40 self.assertEqual(set(('metadata',)), session.enabled_hook_categories) |
40 self.assertEqual(set(('metadata',)), session.enabled_hook_categories) |
41 session.commit() |
41 session.commit() |
53 self.assertEqual(HOOKS_DENY_ALL, session.hooks_mode) |
53 self.assertEqual(HOOKS_DENY_ALL, session.hooks_mode) |
54 self.assertEqual(set(), session.disabled_hook_categories) |
54 self.assertEqual(set(), session.disabled_hook_categories) |
55 self.assertEqual(set(('metadata',)), session.enabled_hook_categories) |
55 self.assertEqual(set(('metadata',)), session.enabled_hook_categories) |
56 # leaving context manager with no transaction running should reset the |
56 # leaving context manager with no transaction running should reset the |
57 # transaction local storage (and associated cnxset) |
57 # transaction local storage (and associated cnxset) |
58 self.assertEqual({}, session._txs) |
58 self.assertEqual({}, session._cnxs) |
59 self.assertEqual(None, session.cnxset) |
59 self.assertEqual(None, session.cnxset) |
60 self.assertEqual(HOOKS_ALLOW_ALL, session.hooks_mode, session.HOOKS_ALLOW_ALL) |
60 self.assertEqual(HOOKS_ALLOW_ALL, session.hooks_mode, session.HOOKS_ALLOW_ALL) |
61 self.assertEqual(set(), session.disabled_hook_categories) |
61 self.assertEqual(set(), session.disabled_hook_categories) |
62 self.assertEqual(set(), session.enabled_hook_categories) |
62 self.assertEqual(set(), session.enabled_hook_categories) |
63 |
63 |