428 rset.related_entity(0, 2) |
428 rset.related_entity(0, 2) |
429 |
429 |
430 def test_entities(self): |
430 def test_entities(self): |
431 rset = self.execute('Any U,G WHERE U in_group G') |
431 rset = self.execute('Any U,G WHERE U in_group G') |
432 # make sure we have at least one element |
432 # make sure we have at least one element |
433 self.failUnless(rset) |
433 self.assertTrue(rset) |
434 self.assertEqual(set(e.e_schema.type for e in rset.entities(0)), |
434 self.assertEqual(set(e.e_schema.type for e in rset.entities(0)), |
435 set(['CWUser',])) |
435 set(['CWUser',])) |
436 self.assertEqual(set(e.e_schema.type for e in rset.entities(1)), |
436 self.assertEqual(set(e.e_schema.type for e in rset.entities(1)), |
437 set(['CWGroup',])) |
437 set(['CWGroup',])) |
438 |
438 |
439 def test_iter_rows_with_entities(self): |
439 def test_iter_rows_with_entities(self): |
440 rset = self.execute('Any U,UN,G,GN WHERE U in_group G, U login UN, G name GN') |
440 rset = self.execute('Any U,UN,G,GN WHERE U in_group G, U login UN, G name GN') |
441 # make sure we have at least one element |
441 # make sure we have at least one element |
442 self.failUnless(rset) |
442 self.assertTrue(rset) |
443 out = list(rset.iter_rows_with_entities())[0] |
443 out = list(rset.iter_rows_with_entities())[0] |
444 self.assertEqual( out[0].login, out[1] ) |
444 self.assertEqual( out[0].login, out[1] ) |
445 self.assertEqual( out[2].name, out[3] ) |
445 self.assertEqual( out[2].name, out[3] ) |
446 |
446 |
447 def test_printable_rql(self): |
447 def test_printable_rql(self): |