hooks/test/unittest_syncschema.py
changeset 9851 42a874c666b1
parent 9675 8aabfefc8a81
child 10365 21461f80f348
equal deleted inserted replaced
9850:5ef9dd383ae2 9851:42a874c666b1
     1 # copyright 2003-2013 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
     1 # copyright 2003-2014 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
     2 # contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
     2 # contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
     3 #
     3 #
     4 # This file is part of CubicWeb.
     4 # This file is part of CubicWeb.
     5 #
     5 #
     6 # CubicWeb is free software: you can redistribute it and/or modify it under the
     6 # CubicWeb is free software: you can redistribute it and/or modify it under the
    15 #
    15 #
    16 # You should have received a copy of the GNU Lesser General Public License along
    16 # You should have received a copy of the GNU Lesser General Public License along
    17 # with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.
    17 # with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.
    18 """cubicweb.server.hooks.syncschema unit and functional tests"""
    18 """cubicweb.server.hooks.syncschema unit and functional tests"""
    19 
    19 
    20 from logilab.common.testlib import TestCase, unittest_main
    20 from logilab.common.testlib import unittest_main
    21 
    21 
    22 from cubicweb import ValidationError, Binary
    22 from cubicweb import ValidationError, Binary
    23 from cubicweb.schema import META_RTYPES
    23 from cubicweb.schema import META_RTYPES
    24 from cubicweb.devtools.testlib import CubicWebTC
    24 from cubicweb.devtools.testlib import CubicWebTC
    25 from cubicweb.server.sqlutils import SQL_PREFIX
    25 from cubicweb.server.sqlutils import SQL_PREFIX
    26 from cubicweb.devtools.repotest import schema_eids_idx, restore_schema_eids_idx
    26 from cubicweb.devtools.repotest import schema_eids_idx
    27 
    27 
    28 
    28 
    29 def tearDownModule(*args):
    29 def tearDownModule(*args):
    30     del SchemaModificationHooksTC.schema_eids
    30     del SchemaModificationHooksTC.schema_eids
    31 
    31 
    34     def setUp(self):
    34     def setUp(self):
    35         super(SchemaModificationHooksTC, self).setUp()
    35         super(SchemaModificationHooksTC, self).setUp()
    36         self.repo.set_schema(self.repo.deserialize_schema(), resetvreg=False)
    36         self.repo.set_schema(self.repo.deserialize_schema(), resetvreg=False)
    37         self.__class__.schema_eids = schema_eids_idx(self.repo.schema)
    37         self.__class__.schema_eids = schema_eids_idx(self.repo.schema)
    38 
    38 
    39     def index_exists(self, etype, attr, unique=False):
    39     def index_exists(self, cnx, etype, attr, unique=False):
    40         self.session.set_cnxset()
       
    41         dbhelper = self.repo.system_source.dbhelper
    40         dbhelper = self.repo.system_source.dbhelper
    42         sqlcursor = self.session.cnxset.cu
    41         with cnx.ensure_cnx_set:
    43         return dbhelper.index_exists(sqlcursor, SQL_PREFIX + etype, SQL_PREFIX + attr, unique=unique)
    42             sqlcursor = cnx.cnxset.cu
    44 
    43             return dbhelper.index_exists(sqlcursor,
    45     def _set_perms(self, eid):
    44                                          SQL_PREFIX + etype,
    46         self.execute('SET X read_permission G WHERE X eid %(x)s, G is CWGroup',
    45                                          SQL_PREFIX + attr,
    47                      {'x': eid})
    46                                          unique=unique)
    48         self.execute('SET X add_permission G WHERE X eid %(x)s, G is CWGroup, G name "managers"',
    47 
    49                      {'x': eid})
    48     def _set_perms(self, cnx, eid):
    50         self.execute('SET X delete_permission G WHERE X eid %(x)s, G is CWGroup, G name "owners"',
    49         cnx.execute('SET X read_permission G WHERE X eid %(x)s, G is CWGroup',
    51                      {'x': eid})
    50                     {'x': eid})
    52 
    51         cnx.execute('SET X add_permission G WHERE X eid %(x)s, G is CWGroup, '
    53     def _set_attr_perms(self, eid):
    52                     'G name "managers"', {'x': eid})
    54         self.execute('SET X read_permission G WHERE X eid %(x)s, G is CWGroup',
    53         cnx.execute('SET X delete_permission G WHERE X eid %(x)s, G is CWGroup, '
    55                      {'x': eid})
    54                     'G name "owners"', {'x': eid})
    56         self.execute('SET X update_permission G WHERE X eid %(x)s, G is CWGroup, G name "managers"',
    55 
    57                      {'x': eid})
    56     def _set_attr_perms(self, cnx, eid):
       
    57         cnx.execute('SET X read_permission G WHERE X eid %(x)s, G is CWGroup',
       
    58                     {'x': eid})
       
    59         cnx.execute('SET X update_permission G WHERE X eid %(x)s, G is CWGroup, G name "managers"',
       
    60                     {'x': eid})
    58 
    61 
    59     def test_base(self):
    62     def test_base(self):
    60         schema = self.repo.schema
    63         with self.admin_access.repo_cnx() as cnx:
    61         self.session.set_cnxset()
    64             schema = self.repo.schema
    62         dbhelper = self.repo.system_source.dbhelper
    65             self.assertFalse(schema.has_entity('Societe2'))
    63         sqlcursor = self.session.cnxset.cu
    66             self.assertFalse(schema.has_entity('concerne2'))
    64         self.assertFalse(schema.has_entity('Societe2'))
    67             # schema should be update on insertion (after commit)
    65         self.assertFalse(schema.has_entity('concerne2'))
    68             eeid = cnx.execute('INSERT CWEType X: X name "Societe2", '
    66         # schema should be update on insertion (after commit)
    69                                'X description "", X final FALSE')[0][0]
    67         eeid = self.execute('INSERT CWEType X: X name "Societe2", X description "", X final FALSE')[0][0]
    70             self._set_perms(cnx, eeid)
    68         self._set_perms(eeid)
    71             cnx.execute('INSERT CWRType X: X name "concerne2", X description "", '
    69         self.execute('INSERT CWRType X: X name "concerne2", X description "", X final FALSE, X symmetric FALSE')
    72                         'X final FALSE, X symmetric FALSE')
    70         self.assertFalse(schema.has_entity('Societe2'))
    73             self.assertFalse(schema.has_entity('Societe2'))
    71         self.assertFalse(schema.has_entity('concerne2'))
    74             self.assertFalse(schema.has_entity('concerne2'))
    72         # have to commit before adding definition relations
    75             # have to commit before adding definition relations
    73         self.commit()
    76             cnx.commit()
    74         self.assertTrue(schema.has_entity('Societe2'))
    77             self.assertTrue(schema.has_entity('Societe2'))
    75         self.assertTrue(schema.has_relation('concerne2'))
    78             self.assertTrue(schema.has_relation('concerne2'))
    76         attreid = self.execute('INSERT CWAttribute X: X cardinality "11", X defaultval %(default)s, '
    79             attreid = cnx.execute('INSERT CWAttribute X: X cardinality "11", '
    77                                '   X indexed TRUE, X relation_type RT, X from_entity E, X to_entity F '
    80                                   'X defaultval %(default)s, X indexed TRUE, '
    78                                'WHERE RT name "name", E name "Societe2", F name "String"',
    81                                   'X relation_type RT, X from_entity E, X to_entity F '
    79                                {'default': Binary.zpickle('noname')})[0][0]
    82                                    'WHERE RT name "name", E name "Societe2", '
    80         self._set_attr_perms(attreid)
    83                                   'F name "String"',
    81         concerne2_rdef_eid = self.execute(
    84                                    {'default': Binary.zpickle('noname')})[0][0]
    82             'INSERT CWRelation X: X cardinality "**", X relation_type RT, X from_entity E, X to_entity E '
    85             self._set_attr_perms(cnx, attreid)
    83             'WHERE RT name "concerne2", E name "Societe2"')[0][0]
    86             concerne2_rdef_eid = cnx.execute(
    84         self._set_perms(concerne2_rdef_eid)
    87                 'INSERT CWRelation X: X cardinality "**", X relation_type RT, '
    85         self.assertNotIn('name', schema['Societe2'].subject_relations())
    88                 'X from_entity E, X to_entity E '
    86         self.assertNotIn('concerne2', schema['Societe2'].subject_relations())
    89                 'WHERE RT name "concerne2", E name "Societe2"')[0][0]
    87         self.assertFalse(self.index_exists('Societe2', 'name'))
    90             self._set_perms(cnx, concerne2_rdef_eid)
    88         self.commit()
    91             self.assertNotIn('name', schema['Societe2'].subject_relations())
    89         self.assertIn('name', schema['Societe2'].subject_relations())
    92             self.assertNotIn('concerne2', schema['Societe2'].subject_relations())
    90         self.assertIn('concerne2', schema['Societe2'].subject_relations())
    93             self.assertFalse(self.index_exists(cnx, 'Societe2', 'name'))
    91         self.assertTrue(self.index_exists('Societe2', 'name'))
    94             cnx.commit()
    92         # now we should be able to insert and query Societe2
    95             self.assertIn('name', schema['Societe2'].subject_relations())
    93         s2eid = self.execute('INSERT Societe2 X: X name "logilab"')[0][0]
    96             self.assertIn('concerne2', schema['Societe2'].subject_relations())
    94         self.execute('Societe2 X WHERE X name "logilab"')
    97             self.assertTrue(self.index_exists(cnx, 'Societe2', 'name'))
    95         self.execute('SET X concerne2 X WHERE X name "logilab"')
    98             # now we should be able to insert and query Societe2
    96         rset = self.execute('Any X WHERE X concerne2 Y')
    99             s2eid = cnx.execute('INSERT Societe2 X: X name "logilab"')[0][0]
    97         self.assertEqual(rset.rows, [[s2eid]])
   100             cnx.execute('Societe2 X WHERE X name "logilab"')
    98         # check that when a relation definition is deleted, existing relations are deleted
   101             cnx.execute('SET X concerne2 X WHERE X name "logilab"')
    99         rdefeid = self.execute('INSERT CWRelation X: X cardinality "**", X relation_type RT, '
   102             rset = cnx.execute('Any X WHERE X concerne2 Y')
   100                                '   X from_entity E, X to_entity E '
   103             self.assertEqual(rset.rows, [[s2eid]])
   101                                'WHERE RT name "concerne2", E name "CWUser"')[0][0]
   104             # check that when a relation definition is deleted, existing relations are deleted
   102         self._set_perms(rdefeid)
   105             rdefeid = cnx.execute('INSERT CWRelation X: X cardinality "**", X relation_type RT, '
   103         self.commit()
   106                                    '   X from_entity E, X to_entity E '
   104         self.execute('DELETE CWRelation X WHERE X eid %(x)s', {'x': concerne2_rdef_eid})
   107                                    'WHERE RT name "concerne2", E name "CWUser"')[0][0]
   105         self.commit()
   108             self._set_perms(cnx, rdefeid)
   106         self.assertIn('concerne2', schema['CWUser'].subject_relations())
   109             cnx.commit()
   107         self.assertNotIn('concerne2', schema['Societe2'].subject_relations())
   110             cnx.execute('DELETE CWRelation X WHERE X eid %(x)s', {'x': concerne2_rdef_eid})
   108         self.assertFalse(self.execute('Any X WHERE X concerne2 Y'))
   111             cnx.commit()
   109         # schema should be cleaned on delete (after commit)
   112             self.assertIn('concerne2', schema['CWUser'].subject_relations())
   110         self.execute('DELETE CWEType X WHERE X name "Societe2"')
   113             self.assertNotIn('concerne2', schema['Societe2'].subject_relations())
   111         self.execute('DELETE CWRType X WHERE X name "concerne2"')
   114             self.assertFalse(cnx.execute('Any X WHERE X concerne2 Y'))
   112         self.assertTrue(self.index_exists('Societe2', 'name'))
   115             # schema should be cleaned on delete (after commit)
   113         self.assertTrue(schema.has_entity('Societe2'))
   116             cnx.execute('DELETE CWEType X WHERE X name "Societe2"')
   114         self.assertTrue(schema.has_relation('concerne2'))
   117             cnx.execute('DELETE CWRType X WHERE X name "concerne2"')
   115         self.commit()
   118             self.assertTrue(self.index_exists(cnx, 'Societe2', 'name'))
   116         self.assertFalse(self.index_exists('Societe2', 'name'))
   119             self.assertTrue(schema.has_entity('Societe2'))
   117         self.assertFalse(schema.has_entity('Societe2'))
   120             self.assertTrue(schema.has_relation('concerne2'))
   118         self.assertFalse(schema.has_entity('concerne2'))
   121             cnx.commit()
   119         self.assertNotIn('concerne2', schema['CWUser'].subject_relations())
   122             self.assertFalse(self.index_exists(cnx, 'Societe2', 'name'))
       
   123             self.assertFalse(schema.has_entity('Societe2'))
       
   124             self.assertFalse(schema.has_entity('concerne2'))
       
   125             self.assertNotIn('concerne2', schema['CWUser'].subject_relations())
   120 
   126 
   121     def test_metartype_with_nordefs(self):
   127     def test_metartype_with_nordefs(self):
   122         META_RTYPES.add('custom_meta')
   128         with self.admin_access.repo_cnx() as cnx:
   123         self.execute('INSERT CWRType X: X name "custom_meta", X description "", '
   129             META_RTYPES.add('custom_meta')
   124                      'X final FALSE, X symmetric FALSE')
   130             cnx.execute('INSERT CWRType X: X name "custom_meta", X description "", '
   125         self.commit()
   131                          'X final FALSE, X symmetric FALSE')
   126         eeid = self.execute('INSERT CWEType X: X name "NEWEtype", '
   132             cnx.commit()
   127                             'X description "", X final FALSE')[0][0]
   133             eeid = cnx.execute('INSERT CWEType X: X name "NEWEtype", '
   128         self._set_perms(eeid)
   134                                 'X description "", X final FALSE')[0][0]
   129         self.commit()
   135             self._set_perms(cnx, eeid)
   130         META_RTYPES.remove('custom_meta')
   136             cnx.commit()
       
   137             META_RTYPES.remove('custom_meta')
   131 
   138 
   132     def test_metartype_with_somerdefs(self):
   139     def test_metartype_with_somerdefs(self):
   133         META_RTYPES.add('custom_meta')
   140         with self.admin_access.repo_cnx() as cnx:
   134         self.execute('INSERT CWRType X: X name "custom_meta", X description "", '
   141             META_RTYPES.add('custom_meta')
   135                      'X final FALSE, X symmetric FALSE')
   142             cnx.execute('INSERT CWRType X: X name "custom_meta", X description "", '
   136         self.commit()
   143                          'X final FALSE, X symmetric FALSE')
   137         rdefeid = self.execute('INSERT CWRelation X: X cardinality "**", X relation_type RT, '
   144             cnx.commit()
   138                                '   X from_entity E, X to_entity E '
   145             rdefeid = cnx.execute('INSERT CWRelation X: X cardinality "**", X relation_type RT, '
   139                                'WHERE RT name "custom_meta", E name "CWUser"')[0][0]
   146                                    '   X from_entity E, X to_entity E '
   140         self._set_perms(rdefeid)
   147                                    'WHERE RT name "custom_meta", E name "CWUser"')[0][0]
   141         self.commit()
   148             self._set_perms(cnx, rdefeid)
   142         eeid = self.execute('INSERT CWEType X: X name "NEWEtype", '
   149             cnx.commit()
   143                             'X description "", X final FALSE')[0][0]
   150             eeid = cnx.execute('INSERT CWEType X: X name "NEWEtype", '
   144         self._set_perms(eeid)
   151                                 'X description "", X final FALSE')[0][0]
   145         self.commit()
   152             self._set_perms(cnx, eeid)
   146         META_RTYPES.remove('custom_meta')
   153             cnx.commit()
       
   154             META_RTYPES.remove('custom_meta')
   147 
   155 
   148     def test_is_instance_of_insertions(self):
   156     def test_is_instance_of_insertions(self):
   149         seid = self.execute('INSERT Transition T: T name "subdiv"')[0][0]
   157         with self.admin_access.repo_cnx() as cnx:
   150         is_etypes = [etype for etype, in self.execute('Any ETN WHERE X eid %s, X is ET, ET name ETN' % seid)]
   158             seid = cnx.execute('INSERT Transition T: T name "subdiv"')[0][0]
   151         self.assertEqual(is_etypes, ['Transition'])
   159             is_etypes = [etype for etype, in cnx.execute('Any ETN WHERE X eid %s, '
   152         instanceof_etypes = [etype for etype, in self.execute('Any ETN WHERE X eid %s, X is_instance_of ET, ET name ETN' % seid)]
   160                                                          'X is ET, ET name ETN' % seid)]
   153         self.assertEqual(sorted(instanceof_etypes), ['BaseTransition', 'Transition'])
   161             self.assertEqual(is_etypes, ['Transition'])
   154         snames = [name for name, in self.execute('Any N WHERE S is BaseTransition, S name N')]
   162             instanceof_etypes = [etype
   155         self.assertNotIn('subdiv', snames)
   163                                  for etype, in cnx.execute('Any ETN WHERE X eid %s, '
   156         snames = [name for name, in self.execute('Any N WHERE S is_instance_of BaseTransition, S name N')]
   164                                                            'X is_instance_of ET, ET name ETN'
   157         self.assertIn('subdiv', snames)
   165                                                            % seid)]
       
   166             self.assertEqual(sorted(instanceof_etypes), ['BaseTransition', 'Transition'])
       
   167             snames = [name for name, in cnx.execute('Any N WHERE S is BaseTransition, S name N')]
       
   168             self.assertNotIn('subdiv', snames)
       
   169             snames = [name for name, in cnx.execute('Any N WHERE S is_instance_of BaseTransition, '
       
   170                                                     'S name N')]
       
   171             self.assertIn('subdiv', snames)
   158 
   172 
   159 
   173 
   160     def test_perms_synchronization_1(self):
   174     def test_perms_synchronization_1(self):
   161         schema = self.repo.schema
   175         with self.admin_access.repo_cnx() as cnx:
   162         self.assertEqual(schema['CWUser'].get_groups('read'), set(('managers', 'users')))
   176             schema = self.repo.schema
   163         self.assertTrue(self.execute('Any X, Y WHERE X is CWEType, X name "CWUser", Y is CWGroup, Y name "users"')[0])
   177             self.assertEqual(schema['CWUser'].get_groups('read'), set(('managers', 'users')))
   164         self.execute('DELETE X read_permission Y WHERE X is CWEType, X name "CWUser", Y name "users"')
   178             self.assertTrue(cnx.execute('Any X, Y WHERE X is CWEType, X name "CWUser", '
   165         self.assertEqual(schema['CWUser'].get_groups('read'), set(('managers', 'users', )))
   179                                         'Y is CWGroup, Y name "users"')[0])
   166         self.commit()
   180             cnx.execute('DELETE X read_permission Y WHERE X is CWEType, X name "CWUser", Y name "users"')
   167         self.assertEqual(schema['CWUser'].get_groups('read'), set(('managers',)))
   181             self.assertEqual(schema['CWUser'].get_groups('read'), set(('managers', 'users', )))
   168         self.execute('SET X read_permission Y WHERE X is CWEType, X name "CWUser", Y name "users"')
   182             cnx.commit()
   169         self.commit()
   183             self.assertEqual(schema['CWUser'].get_groups('read'), set(('managers',)))
   170         self.assertEqual(schema['CWUser'].get_groups('read'), set(('managers', 'users',)))
   184             cnx.execute('SET X read_permission Y WHERE X is CWEType, '
       
   185                         'X name "CWUser", Y name "users"')
       
   186             cnx.commit()
       
   187             self.assertEqual(schema['CWUser'].get_groups('read'),
       
   188                              set(('managers', 'users',)))
   171 
   189 
   172     def test_perms_synchronization_2(self):
   190     def test_perms_synchronization_2(self):
   173         schema = self.repo.schema['in_group'].rdefs[('CWUser', 'CWGroup')]
   191         with self.admin_access.repo_cnx() as cnx:
   174         self.assertEqual(schema.get_groups('read'), set(('managers', 'users', 'guests')))
   192             schema = self.repo.schema['in_group'].rdefs[('CWUser', 'CWGroup')]
   175         self.execute('DELETE X read_permission Y WHERE X relation_type RT, RT name "in_group", Y name "guests"')
   193             self.assertEqual(schema.get_groups('read'),
   176         self.assertEqual(schema.get_groups('read'), set(('managers', 'users', 'guests')))
   194                              set(('managers', 'users', 'guests')))
   177         self.commit()
   195             cnx.execute('DELETE X read_permission Y WHERE X relation_type RT, '
   178         self.assertEqual(schema.get_groups('read'), set(('managers', 'users')))
   196                         'RT name "in_group", Y name "guests"')
   179         self.execute('SET X read_permission Y WHERE X relation_type RT, RT name "in_group", Y name "guests"')
   197             self.assertEqual(schema.get_groups('read'),
   180         self.assertEqual(schema.get_groups('read'), set(('managers', 'users')))
   198                              set(('managers', 'users', 'guests')))
   181         self.commit()
   199             cnx.commit()
   182         self.assertEqual(schema.get_groups('read'), set(('managers', 'users', 'guests')))
   200             self.assertEqual(schema.get_groups('read'),
       
   201                              set(('managers', 'users')))
       
   202             cnx.execute('SET X read_permission Y WHERE X relation_type RT, '
       
   203                         'RT name "in_group", Y name "guests"')
       
   204             self.assertEqual(schema.get_groups('read'),
       
   205                              set(('managers', 'users')))
       
   206             cnx.commit()
       
   207             self.assertEqual(schema.get_groups('read'),
       
   208                              set(('managers', 'users', 'guests')))
   183 
   209 
   184     def test_nonregr_user_edit_itself(self):
   210     def test_nonregr_user_edit_itself(self):
   185         ueid = self.session.user.eid
   211         with self.admin_access.repo_cnx() as cnx:
   186         groupeids = [eid for eid, in self.execute('CWGroup G WHERE G name in ("managers", "users")')]
   212             ueid = cnx.user.eid
   187         self.execute('DELETE X in_group Y WHERE X eid %s' % ueid)
   213             groupeids = [eid for eid, in cnx.execute('CWGroup G WHERE G name '
   188         self.execute('SET X surname "toto" WHERE X eid %s' % ueid)
   214                                                      'in ("managers", "users")')]
   189         self.execute('SET X in_group Y WHERE X eid %s, Y name "managers"' % ueid)
   215             cnx.execute('DELETE X in_group Y WHERE X eid %s' % ueid)
   190         self.commit()
   216             cnx.execute('SET X surname "toto" WHERE X eid %s' % ueid)
   191         eeid = self.execute('Any X WHERE X is CWEType, X name "CWEType"')[0][0]
   217             cnx.execute('SET X in_group Y WHERE X eid %s, Y name "managers"' % ueid)
   192         self.execute('DELETE X read_permission Y WHERE X eid %s' % eeid)
   218             cnx.commit()
   193         self.execute('SET X final FALSE WHERE X eid %s' % eeid)
   219             eeid = cnx.execute('Any X WHERE X is CWEType, X name "CWEType"')[0][0]
   194         self.execute('SET X read_permission Y WHERE X eid %s, Y eid in (%s, %s)'
   220             cnx.execute('DELETE X read_permission Y WHERE X eid %s' % eeid)
   195                      % (eeid, groupeids[0], groupeids[1]))
   221             cnx.execute('SET X final FALSE WHERE X eid %s' % eeid)
   196         self.commit()
   222             cnx.execute('SET X read_permission Y WHERE X eid %s, Y eid in (%s, %s)'
   197         self.execute('Any X WHERE X is CWEType, X name "CWEType"')
   223                          % (eeid, groupeids[0], groupeids[1]))
       
   224             cnx.commit()
       
   225             cnx.execute('Any X WHERE X is CWEType, X name "CWEType"')
   198 
   226 
   199     # schema modification hooks tests #########################################
   227     # schema modification hooks tests #########################################
   200 
   228 
   201     def test_uninline_relation(self):
   229     def test_uninline_relation(self):
   202         self.session.set_cnxset()
   230         with self.admin_access.repo_cnx() as cnx:
   203         dbhelper = self.repo.system_source.dbhelper
   231             try:
   204         sqlcursor = self.session.cnxset.cu
   232                 self.assertTrue(self.schema['state_of'].inlined)
   205         self.assertTrue(self.schema['state_of'].inlined)
   233                 cnx.execute('SET X inlined FALSE WHERE X name "state_of"')
   206         try:
   234                 self.assertTrue(self.schema['state_of'].inlined)
   207             self.execute('SET X inlined FALSE WHERE X name "state_of"')
   235                 cnx.commit()
   208             self.assertTrue(self.schema['state_of'].inlined)
   236                 self.assertFalse(self.schema['state_of'].inlined)
   209             self.commit()
   237                 self.assertFalse(self.index_exists(cnx, 'State', 'state_of'))
   210             self.assertFalse(self.schema['state_of'].inlined)
   238                 rset = cnx.execute('Any X, Y WHERE X state_of Y')
   211             self.assertFalse(self.index_exists('State', 'state_of'))
   239                 self.assertEqual(len(rset), 2) # user states
   212             rset = self.execute('Any X, Y WHERE X state_of Y')
   240             finally:
   213             self.assertEqual(len(rset), 2) # user states
   241                 cnx.execute('SET X inlined TRUE WHERE X name "state_of"')
   214         except Exception:
   242                 self.assertFalse(self.schema['state_of'].inlined)
   215             import traceback
   243                 cnx.commit()
   216             traceback.print_exc()
   244                 self.assertTrue(self.schema['state_of'].inlined)
   217         finally:
   245                 self.assertTrue(self.index_exists(cnx, 'State', 'state_of'))
   218             self.execute('SET X inlined TRUE WHERE X name "state_of"')
   246                 rset = cnx.execute('Any X, Y WHERE X state_of Y')
   219             self.assertFalse(self.schema['state_of'].inlined)
   247                 self.assertEqual(len(rset), 2)
   220             self.commit()
       
   221             self.assertTrue(self.schema['state_of'].inlined)
       
   222             self.assertTrue(self.index_exists('State', 'state_of'))
       
   223             rset = self.execute('Any X, Y WHERE X state_of Y')
       
   224             self.assertEqual(len(rset), 2)
       
   225 
   248 
   226     def test_indexed_change(self):
   249     def test_indexed_change(self):
   227         self.session.set_cnxset()
   250         with self.admin_access.repo_cnx() as cnx:
   228         dbhelper = self.repo.system_source.dbhelper
   251             try:
   229         sqlcursor = self.session.cnxset.cu
   252                 cnx.execute('SET X indexed FALSE WHERE X relation_type R, R name "name"')
   230         try:
   253                 self.assertTrue(self.schema['name'].rdef('Workflow', 'String').indexed)
   231             self.execute('SET X indexed FALSE WHERE X relation_type R, R name "name"')
   254                 self.assertTrue(self.index_exists(cnx, 'Workflow', 'name'))
   232             self.assertTrue(self.schema['name'].rdef('Workflow', 'String').indexed)
   255                 cnx.commit()
   233             self.assertTrue(self.index_exists('Workflow', 'name'))
   256                 self.assertFalse(self.schema['name'].rdef('Workflow', 'String').indexed)
   234             self.commit()
   257                 self.assertFalse(self.index_exists(cnx, 'Workflow', 'name'))
   235             self.assertFalse(self.schema['name'].rdef('Workflow', 'String').indexed)
   258             finally:
   236             self.assertFalse(self.index_exists('Workflow', 'name'))
   259                 cnx.execute('SET X indexed TRUE WHERE X relation_type R, R name "name"')
   237         finally:
   260                 self.assertFalse(self.schema['name'].rdef('Workflow', 'String').indexed)
   238             self.execute('SET X indexed TRUE WHERE X relation_type R, R name "name"')
   261                 self.assertFalse(self.index_exists(cnx, 'Workflow', 'name'))
   239             self.assertFalse(self.schema['name'].rdef('Workflow', 'String').indexed)
   262                 cnx.commit()
   240             self.assertFalse(self.index_exists('Workflow', 'name'))
   263                 self.assertTrue(self.schema['name'].rdef('Workflow', 'String').indexed)
   241             self.commit()
   264                 self.assertTrue(self.index_exists(cnx, 'Workflow', 'name'))
   242             self.assertTrue(self.schema['name'].rdef('Workflow', 'String').indexed)
       
   243             self.assertTrue(self.index_exists('Workflow', 'name'))
       
   244 
   265 
   245     def test_unique_change(self):
   266     def test_unique_change(self):
   246         self.session.set_cnxset()
   267         with self.admin_access.repo_cnx() as cnx:
   247         dbhelper = self.repo.system_source.dbhelper
   268             try:
   248         sqlcursor = self.session.cnxset.cu
   269                 eid = cnx.execute('INSERT CWConstraint X: X cstrtype CT, DEF constrained_by X '
   249         try:
   270                                   'WHERE CT name "UniqueConstraint", DEF relation_type RT, '
   250             eid = self.execute('INSERT CWConstraint X: X cstrtype CT, DEF constrained_by X '
   271                                   'DEF from_entity E, RT name "name", '
   251                                'WHERE CT name "UniqueConstraint", DEF relation_type RT, DEF from_entity E,'
   272                                   'E name "Workflow"').rows[0][0]
   252                                'RT name "name", E name "Workflow"').rows[0][0]
   273                 self.assertFalse(self.schema['Workflow'].has_unique_values('name'))
   253             self.assertFalse(self.schema['Workflow'].has_unique_values('name'))
   274                 self.assertFalse(self.index_exists(cnx, 'Workflow', 'name', unique=True))
   254             self.assertFalse(self.index_exists('Workflow', 'name', unique=True))
   275                 cnx.commit()
   255             self.commit()
   276                 self.assertTrue(self.schema['Workflow'].has_unique_values('name'))
   256             self.assertTrue(self.schema['Workflow'].has_unique_values('name'))
   277                 self.assertTrue(self.index_exists(cnx, 'Workflow', 'name', unique=True))
   257             self.assertTrue(self.index_exists('Workflow', 'name', unique=True))
   278             finally:
   258         finally:
   279                 cnx.execute('DELETE CWConstraint C WHERE C eid %(eid)s', {'eid': eid})
   259             self.execute('DELETE CWConstraint C WHERE C eid %(eid)s', {'eid': eid})
   280                 cnx.commit()
   260             self.commit()
   281                 self.assertFalse(self.schema['Workflow'].has_unique_values('name'))
   261             self.assertFalse(self.schema['Workflow'].has_unique_values('name'))
   282                 self.assertFalse(self.index_exists(cnx, 'Workflow', 'name', unique=True))
   262             self.assertFalse(self.index_exists('Workflow', 'name', unique=True))
       
   263 
   283 
   264     def test_required_change_1(self):
   284     def test_required_change_1(self):
   265         self.execute('SET DEF cardinality "?1" '
   285         with self.admin_access.repo_cnx() as cnx:
   266                      'WHERE DEF relation_type RT, DEF from_entity E,'
   286             cnx.execute('SET DEF cardinality "?1" '
   267                      'RT name "title", E name "Bookmark"')
   287                          'WHERE DEF relation_type RT, DEF from_entity E,'
   268         self.commit()
   288                          'RT name "title", E name "Bookmark"')
   269         # should now be able to add bookmark without title
   289             cnx.commit()
   270         self.execute('INSERT Bookmark X: X path "/view"')
   290             # should now be able to add bookmark without title
   271         self.commit()
   291             cnx.execute('INSERT Bookmark X: X path "/view"')
       
   292             cnx.commit()
   272 
   293 
   273     def test_required_change_2(self):
   294     def test_required_change_2(self):
   274         self.execute('SET DEF cardinality "11" '
   295         with self.admin_access.repo_cnx() as cnx:
   275                      'WHERE DEF relation_type RT, DEF from_entity E,'
   296             cnx.execute('SET DEF cardinality "11" '
   276                      'RT name "surname", E name "CWUser"')
   297                          'WHERE DEF relation_type RT, DEF from_entity E,'
   277         self.commit()
   298                          'RT name "surname", E name "CWUser"')
   278         # should not be able anymore to add cwuser without surname
   299             cnx.commit()
   279         req = self.request()
   300             # should not be able anymore to add cwuser without surname
   280         self.assertRaises(ValidationError, self.create_user, req, "toto")
   301             self.assertRaises(ValidationError, self.create_user, cnx, "toto")
   281         self.rollback()
   302             cnx.rollback()
   282         self.execute('SET DEF cardinality "?1" '
   303             cnx.execute('SET DEF cardinality "?1" '
   283                      'WHERE DEF relation_type RT, DEF from_entity E,'
   304                          'WHERE DEF relation_type RT, DEF from_entity E,'
   284                      'RT name "surname", E name "CWUser"')
   305                          'RT name "surname", E name "CWUser"')
   285         self.commit()
   306             cnx.commit()
   286 
       
   287 
   307 
   288     def test_add_attribute_to_base_class(self):
   308     def test_add_attribute_to_base_class(self):
   289         attreid = self.execute('INSERT CWAttribute X: X cardinality "11", X defaultval %(default)s, '
   309         with self.admin_access.repo_cnx() as cnx:
   290                                'X indexed TRUE, X relation_type RT, X from_entity E, X to_entity F '
   310             attreid = cnx.execute('INSERT CWAttribute X: X cardinality "11", X defaultval %(default)s, '
   291                                'WHERE RT name "messageid", E name "BaseTransition", F name "String"',
   311                                    'X indexed TRUE, X relation_type RT, X from_entity E, X to_entity F '
   292                                {'default': Binary.zpickle('noname')})[0][0]
   312                                    'WHERE RT name "messageid", E name "BaseTransition", F name "String"',
   293         assert self.execute('SET X read_permission Y WHERE X eid %(x)s, Y name "managers"',
   313                                    {'default': Binary.zpickle('noname')})[0][0]
   294                      {'x': attreid})
   314             assert cnx.execute('SET X read_permission Y WHERE X eid %(x)s, Y name "managers"',
   295         self.commit()
   315                                {'x': attreid})
   296         self.schema.rebuild_infered_relations()
   316             cnx.commit()
   297         self.assertIn('Transition', self.schema['messageid'].subjects())
   317             self.schema.rebuild_infered_relations()
   298         self.assertIn('WorkflowTransition', self.schema['messageid'].subjects())
   318             self.assertIn('Transition', self.schema['messageid'].subjects())
   299         self.execute('Any X WHERE X is_instance_of BaseTransition, X messageid "hop"')
   319             self.assertIn('WorkflowTransition', self.schema['messageid'].subjects())
       
   320             cnx.execute('Any X WHERE X is_instance_of BaseTransition, X messageid "hop"')
   300 
   321 
   301     def test_change_fulltextindexed(self):
   322     def test_change_fulltextindexed(self):
   302         req = self.request()
   323         with self.admin_access.repo_cnx() as cnx:
   303         target = req.create_entity(u'Email', messageid=u'1234',
   324             target = cnx.create_entity(u'Email', messageid=u'1234',
   304                                    subject=u'rick.roll@dance.com')
   325                                        subject=u'rick.roll@dance.com')
   305         self.commit()
   326             cnx.commit()
   306         rset = req.execute('Any X WHERE X has_text "rick.roll"')
   327             rset = cnx.execute('Any X WHERE X has_text "rick.roll"')
   307         self.assertIn(target.eid, [item[0] for item in rset])
   328             self.assertIn(target.eid, [item[0] for item in rset])
   308         assert req.execute('SET A fulltextindexed FALSE '
   329             assert cnx.execute('SET A fulltextindexed FALSE '
   309                             'WHERE E is CWEType, E name "Email", A is CWAttribute,'
   330                                'WHERE E is CWEType, E name "Email", A is CWAttribute,'
   310                             'A from_entity E, A relation_type R, R name "subject"')
   331                                'A from_entity E, A relation_type R, R name "subject"')
   311         self.commit()
   332             cnx.commit()
   312         rset = req.execute('Any X WHERE X has_text "rick.roll"')
   333             rset = cnx.execute('Any X WHERE X has_text "rick.roll"')
   313         self.assertFalse(rset)
   334             self.assertFalse(rset)
   314         assert req.execute('SET A fulltextindexed TRUE '
   335             assert cnx.execute('SET A fulltextindexed TRUE '
   315                            'WHERE A from_entity E, A relation_type R, '
   336                                'WHERE A from_entity E, A relation_type R, '
   316                            'E name "Email", R name "subject"')
   337                                'E name "Email", R name "subject"')
   317         self.commit()
   338             cnx.commit()
   318         rset = req.execute('Any X WHERE X has_text "rick.roll"')
   339             rset = cnx.execute('Any X WHERE X has_text "rick.roll"')
   319         self.assertIn(target.eid, [item[0] for item in rset])
   340             self.assertIn(target.eid, [item[0] for item in rset])
   320 
   341 
   321     def test_change_fulltext_container(self):
   342     def test_change_fulltext_container(self):
   322         req = self.request()
   343         with self.admin_access.repo_cnx() as cnx:
   323         target = req.create_entity(u'EmailAddress', address=u'rick.roll@dance.com')
   344             target = cnx.create_entity(u'EmailAddress', address=u'rick.roll@dance.com')
   324         target.cw_set(reverse_use_email=req.user)
   345             target.cw_set(reverse_use_email=cnx.user)
   325         self.commit()
   346             cnx.commit()
   326         rset = req.execute('Any X WHERE X has_text "rick.roll"')
   347             rset = cnx.execute('Any X WHERE X has_text "rick.roll"')
   327         self.assertIn(req.user.eid, [item[0] for item in rset])
   348             self.assertIn(cnx.user.eid, [item[0] for item in rset])
   328         assert self.execute('SET R fulltext_container NULL '
   349             assert cnx.execute('SET R fulltext_container NULL '
   329                             'WHERE R name "use_email"')
   350                                 'WHERE R name "use_email"')
   330         self.commit()
   351             cnx.commit()
   331         rset = self.execute('Any X WHERE X has_text "rick.roll"')
   352             rset = cnx.execute('Any X WHERE X has_text "rick.roll"')
   332         self.assertIn(target.eid, [item[0] for item in rset])
   353             self.assertIn(target.eid, [item[0] for item in rset])
   333         assert self.execute('SET R fulltext_container "subject" '
   354             assert cnx.execute('SET R fulltext_container "subject" '
   334                             'WHERE R name "use_email"')
   355                                 'WHERE R name "use_email"')
   335         self.commit()
   356             cnx.commit()
   336         rset = req.execute('Any X WHERE X has_text "rick.roll"')
   357             rset = cnx.execute('Any X WHERE X has_text "rick.roll"')
   337         self.assertIn(req.user.eid, [item[0] for item in rset])
   358             self.assertIn(cnx.user.eid, [item[0] for item in rset])
   338 
   359 
   339     def test_update_constraint(self):
   360     def test_update_constraint(self):
   340         rdef = self.schema['Transition'].rdef('type')
   361         with self.admin_access.repo_cnx() as cnx:
   341         cstr = rdef.constraint_by_type('StaticVocabularyConstraint')
   362             rdef = self.schema['Transition'].rdef('type')
   342         if not getattr(cstr, 'eid', None):
   363             cstr = rdef.constraint_by_type('StaticVocabularyConstraint')
   343             self.skipTest('start me alone') # bug in schema reloading, constraint's eid not restored
   364             if not getattr(cstr, 'eid', None):
   344         self.execute('SET X value %(v)s WHERE X eid %(x)s',
   365                 # bug in schema reloading, constraint's eid not restored
   345                      {'x': cstr.eid, 'v': u"u'normal', u'auto', u'new'"})
   366                 self.skipTest('start me alone')
   346         self.execute('INSERT CWConstraint X: X value %(value)s, X cstrtype CT, EDEF constrained_by X '
   367             cnx.execute('SET X value %(v)s WHERE X eid %(x)s',
   347                      'WHERE CT name %(ct)s, EDEF eid %(x)s',
   368                          {'x': cstr.eid, 'v': u"u'normal', u'auto', u'new'"})
   348                      {'ct': 'SizeConstraint', 'value': u'max=10', 'x': rdef.eid})
   369             cnx.execute('INSERT CWConstraint X: X value %(value)s, X cstrtype CT, '
   349         self.commit()
   370                         'EDEF constrained_by X WHERE CT name %(ct)s, EDEF eid %(x)s',
   350         cstr = rdef.constraint_by_type('StaticVocabularyConstraint')
   371                          {'ct': 'SizeConstraint', 'value': u'max=10', 'x': rdef.eid})
   351         self.assertEqual(cstr.values, (u'normal', u'auto', u'new'))
   372             cnx.commit()
   352         self.execute('INSERT Transition T: T name "hop", T type "new"')
   373             cstr = rdef.constraint_by_type('StaticVocabularyConstraint')
       
   374             self.assertEqual(cstr.values, (u'normal', u'auto', u'new'))
       
   375             cnx.execute('INSERT Transition T: T name "hop", T type "new"')
   353 
   376 
   354 if __name__ == '__main__':
   377 if __name__ == '__main__':
   355     unittest_main()
   378     unittest_main()