def setUp(self):
super(SimpleUserGroupTests, self).setUp()
if RestAuthUser.get_all(self.conn):
raise RuntimeError("Found leftover users.")
if RestAuthGroup.get_all(self.conn):
raise RuntimeError("Found leftover groups.")
self.user = RestAuthUser.create(self.conn, username, password)
self.group = RestAuthGroup.create(self.conn, groupname)
def test_removeGroupClass(self):
self.assertFalse(self.user.in_group(self.group))
self.assertEqual([], self.user.get_groups())
self.user.add_group(self.group)
self.assertTrue(self.user.in_group(self.group))
self.assertEqual([self.group], RestAuthGroup.get_all(self.conn, self.user))
self.user.remove_group(self.group)
self.assertFalse(self.user.in_group(self.group))
self.assertEqual([], self.user.get_groups())
def test_addGroupClass(self):
self.user.add_group(self.group)
self.assertEqual([self.group], self.user.get_groups())
self.assertEqual([self.group], RestAuthGroup.get_all(self.conn, self.user))
def tearDown(self):
for user in RestAuthUser.get_all(self.conn):
user.remove()
for grp in RestAuthGroup.get_all(self.conn):
grp.remove()
