90 source.synchronize() |
90 source.synchronize() |
91 |
91 |
92 def test_base(self): |
92 def test_base(self): |
93 # check a known one |
93 # check a known one |
94 e = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT}).get_entity(0, 0) |
94 e = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT}).get_entity(0, 0) |
95 self.assertEquals(e.login, SYT) |
95 self.assertEqual(e.login, SYT) |
96 e.complete() |
96 e.complete() |
97 self.assertEquals(e.creation_date, None) |
97 self.assertEqual(e.creation_date, None) |
98 self.assertEquals(e.modification_date, None) |
98 self.assertEqual(e.modification_date, None) |
99 self.assertEquals(e.firstname, None) |
99 self.assertEqual(e.firstname, None) |
100 self.assertEquals(e.surname, None) |
100 self.assertEqual(e.surname, None) |
101 self.assertEquals(e.in_group[0].name, 'users') |
101 self.assertEqual(e.in_group[0].name, 'users') |
102 self.assertEquals(e.owned_by[0].login, SYT) |
102 self.assertEqual(e.owned_by[0].login, SYT) |
103 self.assertEquals(e.created_by, ()) |
103 self.assertEqual(e.created_by, ()) |
104 self.assertEquals(e.primary_email[0].address, 'Sylvain Thenault') |
104 self.assertEqual(e.primary_email[0].address, 'Sylvain Thenault') |
105 # email content should be indexed on the user |
105 # email content should be indexed on the user |
106 rset = self.sexecute('CWUser X WHERE X has_text "thenault"') |
106 rset = self.sexecute('CWUser X WHERE X has_text "thenault"') |
107 self.assertEquals(rset.rows, [[e.eid]]) |
107 self.assertEqual(rset.rows, [[e.eid]]) |
108 |
108 |
109 def test_not(self): |
109 def test_not(self): |
110 eid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
110 eid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
111 rset = self.sexecute('CWUser X WHERE NOT X eid %s' % eid) |
111 rset = self.sexecute('CWUser X WHERE NOT X eid %s' % eid) |
112 self.assert_(rset) |
112 self.assert_(rset) |
115 def test_multiple(self): |
115 def test_multiple(self): |
116 seid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
116 seid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
117 aeid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': ADIM})[0][0] |
117 aeid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': ADIM})[0][0] |
118 rset = self.sexecute('CWUser X, Y WHERE X login %(syt)s, Y login %(adim)s', |
118 rset = self.sexecute('CWUser X, Y WHERE X login %(syt)s, Y login %(adim)s', |
119 {'syt': SYT, 'adim': ADIM}) |
119 {'syt': SYT, 'adim': ADIM}) |
120 self.assertEquals(rset.rows, [[seid, aeid]]) |
120 self.assertEqual(rset.rows, [[seid, aeid]]) |
121 rset = self.sexecute('Any X,Y,L WHERE X login L, X login %(syt)s, Y login %(adim)s', |
121 rset = self.sexecute('Any X,Y,L WHERE X login L, X login %(syt)s, Y login %(adim)s', |
122 {'syt': SYT, 'adim': ADIM}) |
122 {'syt': SYT, 'adim': ADIM}) |
123 self.assertEquals(rset.rows, [[seid, aeid, SYT]]) |
123 self.assertEqual(rset.rows, [[seid, aeid, SYT]]) |
124 |
124 |
125 def test_in(self): |
125 def test_in(self): |
126 seid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
126 seid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
127 aeid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': ADIM})[0][0] |
127 aeid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': ADIM})[0][0] |
128 rset = self.sexecute('Any X,L ORDERBY L WHERE X login IN("%s", "%s"), X login L' % (SYT, ADIM)) |
128 rset = self.sexecute('Any X,L ORDERBY L WHERE X login IN("%s", "%s"), X login L' % (SYT, ADIM)) |
129 self.assertEquals(rset.rows, [[aeid, ADIM], [seid, SYT]]) |
129 self.assertEqual(rset.rows, [[aeid, ADIM], [seid, SYT]]) |
130 |
130 |
131 def test_relations(self): |
131 def test_relations(self): |
132 eid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
132 eid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
133 rset = self.sexecute('Any X,E WHERE X is CWUser, X login L, X primary_email E') |
133 rset = self.sexecute('Any X,E WHERE X is CWUser, X login L, X primary_email E') |
134 self.assert_(eid in (r[0] for r in rset)) |
134 self.assert_(eid in (r[0] for r in rset)) |
142 self.assert_(nbusers < 30, nbusers) |
142 self.assert_(nbusers < 30, nbusers) |
143 |
143 |
144 def test_upper(self): |
144 def test_upper(self): |
145 eid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
145 eid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
146 rset = self.sexecute('Any UPPER(L) WHERE X eid %s, X login L' % eid) |
146 rset = self.sexecute('Any UPPER(L) WHERE X eid %s, X login L' % eid) |
147 self.assertEquals(rset[0][0], SYT.upper()) |
147 self.assertEqual(rset[0][0], SYT.upper()) |
148 |
148 |
149 def test_unknown_attr(self): |
149 def test_unknown_attr(self): |
150 eid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
150 eid = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT})[0][0] |
151 rset = self.sexecute('Any L,C,M WHERE X eid %s, X login L, ' |
151 rset = self.sexecute('Any L,C,M WHERE X eid %s, X login L, ' |
152 'X creation_date C, X modification_date M' % eid) |
152 'X creation_date C, X modification_date M' % eid) |
153 self.assertEquals(rset[0][0], SYT) |
153 self.assertEqual(rset[0][0], SYT) |
154 self.assertEquals(rset[0][1], None) |
154 self.assertEqual(rset[0][1], None) |
155 self.assertEquals(rset[0][2], None) |
155 self.assertEqual(rset[0][2], None) |
156 |
156 |
157 def test_sort(self): |
157 def test_sort(self): |
158 logins = [l for l, in self.sexecute('Any L ORDERBY L WHERE X login L')] |
158 logins = [l for l, in self.sexecute('Any L ORDERBY L WHERE X login L')] |
159 self.assertEquals(logins, sorted(logins)) |
159 self.assertEqual(logins, sorted(logins)) |
160 |
160 |
161 def test_lower_sort(self): |
161 def test_lower_sort(self): |
162 logins = [l for l, in self.sexecute('Any L ORDERBY lower(L) WHERE X login L')] |
162 logins = [l for l, in self.sexecute('Any L ORDERBY lower(L) WHERE X login L')] |
163 self.assertEquals(logins, sorted(logins)) |
163 self.assertEqual(logins, sorted(logins)) |
164 |
164 |
165 def test_or(self): |
165 def test_or(self): |
166 rset = self.sexecute('DISTINCT Any X WHERE X login %(login)s OR (X in_group G, G name "managers")', |
166 rset = self.sexecute('DISTINCT Any X WHERE X login %(login)s OR (X in_group G, G name "managers")', |
167 {'login': SYT}) |
167 {'login': SYT}) |
168 self.assertEquals(len(rset), 2, rset.rows) # syt + admin |
168 self.assertEqual(len(rset), 2, rset.rows) # syt + admin |
169 |
169 |
170 def test_nonregr_set_owned_by(self): |
170 def test_nonregr_set_owned_by(self): |
171 # test that when a user coming from ldap is triggering a transition |
171 # test that when a user coming from ldap is triggering a transition |
172 # the related TrInfo has correct owner information |
172 # the related TrInfo has correct owner information |
173 self.sexecute('SET X in_group G WHERE X login %(syt)s, G name "managers"', {'syt': SYT}) |
173 self.sexecute('SET X in_group G WHERE X login %(syt)s, G name "managers"', {'syt': SYT}) |
174 self.commit() |
174 self.commit() |
175 syt = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT}).get_entity(0, 0) |
175 syt = self.sexecute('CWUser X WHERE X login %(login)s', {'login': SYT}).get_entity(0, 0) |
176 self.assertEquals([g.name for g in syt.in_group], ['managers', 'users']) |
176 self.assertEqual([g.name for g in syt.in_group], ['managers', 'users']) |
177 self.patch_authenticate() |
177 self.patch_authenticate() |
178 cnx = self.login(SYT, password='dummypassword') |
178 cnx = self.login(SYT, password='dummypassword') |
179 cu = cnx.cursor() |
179 cu = cnx.cursor() |
180 adim = cu.execute('CWUser X WHERE X login %(login)s', {'login': ADIM}).get_entity(0, 0) |
180 adim = cu.execute('CWUser X WHERE X login %(login)s', {'login': ADIM}).get_entity(0, 0) |
181 iworkflowable = adim.cw_adapt_to('IWorkflowable') |
181 iworkflowable = adim.cw_adapt_to('IWorkflowable') |
182 iworkflowable.fire_transition('deactivate') |
182 iworkflowable.fire_transition('deactivate') |
183 try: |
183 try: |
184 cnx.commit() |
184 cnx.commit() |
185 adim.clear_all_caches() |
185 adim.clear_all_caches() |
186 self.assertEquals(adim.in_state[0].name, 'deactivated') |
186 self.assertEqual(adim.in_state[0].name, 'deactivated') |
187 trinfo = iworkflowable.latest_trinfo() |
187 trinfo = iworkflowable.latest_trinfo() |
188 self.assertEquals(trinfo.owned_by[0].login, SYT) |
188 self.assertEqual(trinfo.owned_by[0].login, SYT) |
189 # select from_state to skip the user's creation TrInfo |
189 # select from_state to skip the user's creation TrInfo |
190 rset = self.sexecute('Any U ORDERBY D DESC WHERE WF wf_info_for X,' |
190 rset = self.sexecute('Any U ORDERBY D DESC WHERE WF wf_info_for X,' |
191 'WF creation_date D, WF from_state FS,' |
191 'WF creation_date D, WF from_state FS,' |
192 'WF owned_by U?, X eid %(x)s', |
192 'WF owned_by U?, X eid %(x)s', |
193 {'x': adim.eid}) |
193 {'x': adim.eid}) |
194 self.assertEquals(rset.rows, [[syt.eid]]) |
194 self.assertEqual(rset.rows, [[syt.eid]]) |
195 finally: |
195 finally: |
196 # restore db state |
196 # restore db state |
197 self.restore_connection() |
197 self.restore_connection() |
198 adim = self.sexecute('CWUser X WHERE X login %(login)s', {'login': ADIM}).get_entity(0, 0) |
198 adim = self.sexecute('CWUser X WHERE X login %(login)s', {'login': ADIM}).get_entity(0, 0) |
199 adim.cw_adapt_to('IWorkflowable').fire_transition('activate') |
199 adim.cw_adapt_to('IWorkflowable').fire_transition('activate') |
211 self.session.create_entity('CWGroup', name=u'bougloup1') |
211 self.session.create_entity('CWGroup', name=u'bougloup1') |
212 self.session.create_entity('CWGroup', name=u'bougloup2') |
212 self.session.create_entity('CWGroup', name=u'bougloup2') |
213 self.sexecute('SET U in_group G WHERE G name ~= "bougloup%", U login "admin"') |
213 self.sexecute('SET U in_group G WHERE G name ~= "bougloup%", U login "admin"') |
214 self.sexecute('SET U in_group G WHERE G name = "bougloup1", U login %(syt)s', {'syt': SYT}) |
214 self.sexecute('SET U in_group G WHERE G name = "bougloup1", U login %(syt)s', {'syt': SYT}) |
215 rset = self.sexecute('Any L,SN ORDERBY L WHERE X in_state S, S name SN, X login L, EXISTS(X in_group G, G name ~= "bougloup%")') |
215 rset = self.sexecute('Any L,SN ORDERBY L WHERE X in_state S, S name SN, X login L, EXISTS(X in_group G, G name ~= "bougloup%")') |
216 self.assertEquals(rset.rows, [['admin', 'activated'], [SYT, 'activated']]) |
216 self.assertEqual(rset.rows, [['admin', 'activated'], [SYT, 'activated']]) |
217 |
217 |
218 def test_exists2(self): |
218 def test_exists2(self): |
219 self.create_user('comme') |
219 self.create_user('comme') |
220 self.create_user('cochon') |
220 self.create_user('cochon') |
221 self.sexecute('SET X copain Y WHERE X login "comme", Y login "cochon"') |
221 self.sexecute('SET X copain Y WHERE X login "comme", Y login "cochon"') |
222 rset = self.sexecute('Any GN ORDERBY GN WHERE X in_group G, G name GN, (G name "managers" OR EXISTS(X copain T, T login in ("comme", "cochon")))') |
222 rset = self.sexecute('Any GN ORDERBY GN WHERE X in_group G, G name GN, (G name "managers" OR EXISTS(X copain T, T login in ("comme", "cochon")))') |
223 self.assertEquals(rset.rows, [['managers'], ['users']]) |
223 self.assertEqual(rset.rows, [['managers'], ['users']]) |
224 |
224 |
225 def test_exists3(self): |
225 def test_exists3(self): |
226 self.create_user('comme') |
226 self.create_user('comme') |
227 self.create_user('cochon') |
227 self.create_user('cochon') |
228 self.sexecute('SET X copain Y WHERE X login "comme", Y login "cochon"') |
228 self.sexecute('SET X copain Y WHERE X login "comme", Y login "cochon"') |
229 self.failUnless(self.sexecute('Any X, Y WHERE X copain Y, X login "comme", Y login "cochon"')) |
229 self.failUnless(self.sexecute('Any X, Y WHERE X copain Y, X login "comme", Y login "cochon"')) |
230 self.sexecute('SET X copain Y WHERE X login %(syt)s, Y login "cochon"', {'syt': SYT}) |
230 self.sexecute('SET X copain Y WHERE X login %(syt)s, Y login "cochon"', {'syt': SYT}) |
231 self.failUnless(self.sexecute('Any X, Y WHERE X copain Y, X login %(syt)s, Y login "cochon"', {'syt': SYT})) |
231 self.failUnless(self.sexecute('Any X, Y WHERE X copain Y, X login %(syt)s, Y login "cochon"', {'syt': SYT})) |
232 rset = self.sexecute('Any GN,L WHERE X in_group G, X login L, G name GN, G name "managers" OR EXISTS(X copain T, T login in ("comme", "cochon"))') |
232 rset = self.sexecute('Any GN,L WHERE X in_group G, X login L, G name GN, G name "managers" OR EXISTS(X copain T, T login in ("comme", "cochon"))') |
233 self.assertEquals(sorted(rset.rows), [['managers', 'admin'], ['users', 'comme'], ['users', SYT]]) |
233 self.assertEqual(sorted(rset.rows), [['managers', 'admin'], ['users', 'comme'], ['users', SYT]]) |
234 |
234 |
235 def test_exists4(self): |
235 def test_exists4(self): |
236 self.create_user('comme') |
236 self.create_user('comme') |
237 self.create_user('cochon', groups=('users', 'guests')) |
237 self.create_user('cochon', groups=('users', 'guests')) |
238 self.create_user('billy') |
238 self.create_user('billy') |
250 'EXISTS(X copain T, T login L, T login in ("comme", "cochon")) OR ' |
250 'EXISTS(X copain T, T login L, T login in ("comme", "cochon")) OR ' |
251 'EXISTS(X in_state S, S name "activated", NOT X copain T2, T2 login "billy")') |
251 'EXISTS(X in_state S, S name "activated", NOT X copain T2, T2 login "billy")') |
252 all = self.sexecute('Any GN, L WHERE X in_group G, X login L, G name GN') |
252 all = self.sexecute('Any GN, L WHERE X in_group G, X login L, G name GN') |
253 all.rows.remove(['users', 'comme']) |
253 all.rows.remove(['users', 'comme']) |
254 all.rows.remove(['users', SYT]) |
254 all.rows.remove(['users', SYT]) |
255 self.assertEquals(sorted(rset.rows), sorted(all.rows)) |
255 self.assertEqual(sorted(rset.rows), sorted(all.rows)) |
256 |
256 |
257 def test_exists5(self): |
257 def test_exists5(self): |
258 self.create_user('comme') |
258 self.create_user('comme') |
259 self.create_user('cochon', groups=('users', 'guests')) |
259 self.create_user('cochon', groups=('users', 'guests')) |
260 self.create_user('billy') |
260 self.create_user('billy') |
263 self.sexecute('SET X copain Y WHERE X login "comme", Y login "billy"') |
263 self.sexecute('SET X copain Y WHERE X login "comme", Y login "billy"') |
264 self.sexecute('SET X copain Y WHERE X login %(syt)s, Y login "cochon"', {'syt': SYT}) |
264 self.sexecute('SET X copain Y WHERE X login %(syt)s, Y login "cochon"', {'syt': SYT}) |
265 rset= self.sexecute('Any L WHERE X login L, ' |
265 rset= self.sexecute('Any L WHERE X login L, ' |
266 'EXISTS(X copain T, T login in ("comme", "cochon")) AND ' |
266 'EXISTS(X copain T, T login in ("comme", "cochon")) AND ' |
267 'NOT EXISTS(X copain T2, T2 login "billy")') |
267 'NOT EXISTS(X copain T2, T2 login "billy")') |
268 self.assertEquals(sorted(rset.rows), [['cochon'], [SYT]]) |
268 self.assertEqual(sorted(rset.rows), [['cochon'], [SYT]]) |
269 rset= self.sexecute('Any GN,L WHERE X in_group G, X login L, G name GN, ' |
269 rset= self.sexecute('Any GN,L WHERE X in_group G, X login L, G name GN, ' |
270 'EXISTS(X copain T, T login in ("comme", "cochon")) AND ' |
270 'EXISTS(X copain T, T login in ("comme", "cochon")) AND ' |
271 'NOT EXISTS(X copain T2, T2 login "billy")') |
271 'NOT EXISTS(X copain T2, T2 login "billy")') |
272 self.assertEquals(sorted(rset.rows), [['guests', 'cochon'], |
272 self.assertEqual(sorted(rset.rows), [['guests', 'cochon'], |
273 ['users', 'cochon'], |
273 ['users', 'cochon'], |
274 ['users', SYT]]) |
274 ['users', SYT]]) |
275 |
275 |
276 def test_cd_restriction(self): |
276 def test_cd_restriction(self): |
277 rset = self.sexecute('CWUser X WHERE X creation_date > "2009-02-01"') |
277 rset = self.sexecute('CWUser X WHERE X creation_date > "2009-02-01"') |
278 # admin/anon but no ldap user since it doesn't support creation_date |
278 # admin/anon but no ldap user since it doesn't support creation_date |
279 self.assertEquals(sorted(e.login for e in rset.entities()), |
279 self.assertEqual(sorted(e.login for e in rset.entities()), |
280 ['admin', 'anon']) |
280 ['admin', 'anon']) |
281 |
281 |
282 def test_union(self): |
282 def test_union(self): |
283 afeids = self.sexecute('State X') |
283 afeids = self.sexecute('State X') |
284 ueids = self.sexecute('CWUser X') |
284 ueids = self.sexecute('CWUser X') |
285 rset = self.sexecute('(Any X WHERE X is State) UNION (Any X WHERE X is CWUser)') |
285 rset = self.sexecute('(Any X WHERE X is State) UNION (Any X WHERE X is CWUser)') |
286 self.assertEquals(sorted(r[0] for r in rset.rows), |
286 self.assertEqual(sorted(r[0] for r in rset.rows), |
287 sorted(r[0] for r in afeids + ueids)) |
287 sorted(r[0] for r in afeids + ueids)) |
288 |
288 |
289 def _init_security_test(self): |
289 def _init_security_test(self): |
290 self.create_user('iaminguestsgrouponly', groups=('guests',)) |
290 self.create_user('iaminguestsgrouponly', groups=('guests',)) |
291 cnx = self.login('iaminguestsgrouponly') |
291 cnx = self.login('iaminguestsgrouponly') |
292 return cnx.cursor() |
292 return cnx.cursor() |
293 |
293 |
294 def test_security1(self): |
294 def test_security1(self): |
295 cu = self._init_security_test() |
295 cu = self._init_security_test() |
296 rset = cu.execute('CWUser X WHERE X login %(login)s', {'login': SYT}) |
296 rset = cu.execute('CWUser X WHERE X login %(login)s', {'login': SYT}) |
297 self.assertEquals(rset.rows, []) |
297 self.assertEqual(rset.rows, []) |
298 rset = cu.execute('Any X WHERE X login "iaminguestsgrouponly"') |
298 rset = cu.execute('Any X WHERE X login "iaminguestsgrouponly"') |
299 self.assertEquals(len(rset.rows), 1) |
299 self.assertEqual(len(rset.rows), 1) |
300 |
300 |
301 def test_security2(self): |
301 def test_security2(self): |
302 cu = self._init_security_test() |
302 cu = self._init_security_test() |
303 rset = cu.execute('Any X WHERE X has_text %(syt)s', {'syt': SYT}) |
303 rset = cu.execute('Any X WHERE X has_text %(syt)s', {'syt': SYT}) |
304 self.assertEquals(rset.rows, []) |
304 self.assertEqual(rset.rows, []) |
305 rset = cu.execute('Any X WHERE X has_text "iaminguestsgrouponly"') |
305 rset = cu.execute('Any X WHERE X has_text "iaminguestsgrouponly"') |
306 self.assertEquals(len(rset.rows), 1) |
306 self.assertEqual(len(rset.rows), 1) |
307 |
307 |
308 def test_security3(self): |
308 def test_security3(self): |
309 cu = self._init_security_test() |
309 cu = self._init_security_test() |
310 rset = cu.execute('Any F WHERE X has_text %(syt)s, X firstname F', {'syt': SYT}) |
310 rset = cu.execute('Any F WHERE X has_text %(syt)s, X firstname F', {'syt': SYT}) |
311 self.assertEquals(rset.rows, []) |
311 self.assertEqual(rset.rows, []) |
312 rset = cu.execute('Any F WHERE X has_text "iaminguestsgrouponly", X firstname F') |
312 rset = cu.execute('Any F WHERE X has_text "iaminguestsgrouponly", X firstname F') |
313 self.assertEquals(rset.rows, [[None]]) |
313 self.assertEqual(rset.rows, [[None]]) |
314 |
314 |
315 def test_nonregr1(self): |
315 def test_nonregr1(self): |
316 self.sexecute('Any X,AA ORDERBY AA DESC WHERE E eid %(x)s, E owned_by X, ' |
316 self.sexecute('Any X,AA ORDERBY AA DESC WHERE E eid %(x)s, E owned_by X, ' |
317 'X modification_date AA', |
317 'X modification_date AA', |
318 {'x': self.session.user.eid}) |
318 {'x': self.session.user.eid}) |
349 class GlobTrFuncTC(TestCase): |
349 class GlobTrFuncTC(TestCase): |
350 |
350 |
351 def test_count(self): |
351 def test_count(self): |
352 trfunc = GlobTrFunc('count', 0) |
352 trfunc = GlobTrFunc('count', 0) |
353 res = trfunc.apply([[1], [2], [3], [4]]) |
353 res = trfunc.apply([[1], [2], [3], [4]]) |
354 self.assertEquals(res, [[4]]) |
354 self.assertEqual(res, [[4]]) |
355 trfunc = GlobTrFunc('count', 1) |
355 trfunc = GlobTrFunc('count', 1) |
356 res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]]) |
356 res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]]) |
357 self.assertEquals(res, [[1, 2], [2, 1], [3, 1]]) |
357 self.assertEqual(res, [[1, 2], [2, 1], [3, 1]]) |
358 |
358 |
359 def test_sum(self): |
359 def test_sum(self): |
360 trfunc = GlobTrFunc('sum', 0) |
360 trfunc = GlobTrFunc('sum', 0) |
361 res = trfunc.apply([[1], [2], [3], [4]]) |
361 res = trfunc.apply([[1], [2], [3], [4]]) |
362 self.assertEquals(res, [[10]]) |
362 self.assertEqual(res, [[10]]) |
363 trfunc = GlobTrFunc('sum', 1) |
363 trfunc = GlobTrFunc('sum', 1) |
364 res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]]) |
364 res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]]) |
365 self.assertEquals(res, [[1, 7], [2, 4], [3, 6]]) |
365 self.assertEqual(res, [[1, 7], [2, 4], [3, 6]]) |
366 |
366 |
367 def test_min(self): |
367 def test_min(self): |
368 trfunc = GlobTrFunc('min', 0) |
368 trfunc = GlobTrFunc('min', 0) |
369 res = trfunc.apply([[1], [2], [3], [4]]) |
369 res = trfunc.apply([[1], [2], [3], [4]]) |
370 self.assertEquals(res, [[1]]) |
370 self.assertEqual(res, [[1]]) |
371 trfunc = GlobTrFunc('min', 1) |
371 trfunc = GlobTrFunc('min', 1) |
372 res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]]) |
372 res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]]) |
373 self.assertEquals(res, [[1, 2], [2, 4], [3, 6]]) |
373 self.assertEqual(res, [[1, 2], [2, 4], [3, 6]]) |
374 |
374 |
375 def test_max(self): |
375 def test_max(self): |
376 trfunc = GlobTrFunc('max', 0) |
376 trfunc = GlobTrFunc('max', 0) |
377 res = trfunc.apply([[1], [2], [3], [4]]) |
377 res = trfunc.apply([[1], [2], [3], [4]]) |
378 self.assertEquals(res, [[4]]) |
378 self.assertEqual(res, [[4]]) |
379 trfunc = GlobTrFunc('max', 1) |
379 trfunc = GlobTrFunc('max', 1) |
380 res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]]) |
380 res = trfunc.apply([[1, 2], [2, 4], [3, 6], [1, 5]]) |
381 self.assertEquals(res, [[1, 5], [2, 4], [3, 6]]) |
381 self.assertEqual(res, [[1, 5], [2, 4], [3, 6]]) |
382 |
382 |
383 # XXX |
383 # XXX |
384 LDAPUserSourceTC._init_repo() |
384 LDAPUserSourceTC._init_repo() |
385 repo = LDAPUserSourceTC.repo |
385 repo = LDAPUserSourceTC.repo |
386 |
386 |
403 repo._free_pool(self.pool) |
403 repo._free_pool(self.pool) |
404 RQLGeneratorTC.tearDown(self) |
404 RQLGeneratorTC.tearDown(self) |
405 |
405 |
406 def test_base(self): |
406 def test_base(self): |
407 rqlst = self._prepare('CWUser X WHERE X login "toto"').children[0] |
407 rqlst = self._prepare('CWUser X WHERE X login "toto"').children[0] |
408 self.assertEquals(self.o.generate(rqlst, 'X')[1], |
408 self.assertEqual(self.o.generate(rqlst, 'X')[1], |
409 '(&(objectClass=top)(objectClass=posixAccount)(uid=toto))') |
409 '(&(objectClass=top)(objectClass=posixAccount)(uid=toto))') |
410 |
410 |
411 def test_kwargs(self): |
411 def test_kwargs(self): |
412 rqlst = self._prepare('CWUser X WHERE X login %(x)s').children[0] |
412 rqlst = self._prepare('CWUser X WHERE X login %(x)s').children[0] |
413 self.o._args = {'x': "toto"} |
413 self.o._args = {'x': "toto"} |
414 self.assertEquals(self.o.generate(rqlst, 'X')[1], |
414 self.assertEqual(self.o.generate(rqlst, 'X')[1], |
415 '(&(objectClass=top)(objectClass=posixAccount)(uid=toto))') |
415 '(&(objectClass=top)(objectClass=posixAccount)(uid=toto))') |
416 |
416 |
417 def test_get_attr(self): |
417 def test_get_attr(self): |
418 rqlst = self._prepare('Any X WHERE E firstname X, E eid 12').children[0] |
418 rqlst = self._prepare('Any X WHERE E firstname X, E eid 12').children[0] |
419 self.assertRaises(UnknownEid, self.o.generate, rqlst, 'E') |
419 self.assertRaises(UnknownEid, self.o.generate, rqlst, 'E') |