def test_delete_client_role(self):
client_def1 = 'def1'
client_def2 = 'def2'
role_id1, role_name1 = 1, 'name1'
role_id2, role_name2 = 2, 'name2'
role_id3, role_name3 = 3, 'name3'
rbac = RBAC()
rbac.create_role(role_id1, role_name1, None)
rbac.create_role(role_id2, role_name2, None)
rbac.create_role(role_id3, role_name3, None)
rbac.create_client_role(client_def1, role_id1)
rbac.create_client_role(client_def1, role_id2)
rbac.create_client_role(client_def2, role_id2)
rbac.create_client_role(client_def2, role_id3)
rbac.delete_client_role(client_def1, role_id1)
rbac.delete_client_role(client_def2, role_id3)
self.assertEquals(rbac.client_def_to_role_id[client_def1], set([role_id2]))
self.assertEquals(rbac.client_def_to_role_id[client_def2], set([role_id2]))
rbac.delete_client_role(client_def1, role_id2)
rbac.delete_client_role(client_def2, role_id2)
self.assertEquals(rbac.client_def_to_role_id[client_def1], set([]))
self.assertEquals(rbac.client_def_to_role_id[client_def2], set([]))
def test_delete_client_role(self):
client_def1 = 'def1'
client_def2 = 'def2'
role_id1, role_name1 = 1, 'name1'
role_id2, role_name2 = 2, 'name2'
role_id3, role_name3 = 3, 'name3'
rbac = RBAC()
rbac.create_role(role_id1, role_name1, None)
rbac.create_role(role_id2, role_name2, None)
rbac.create_role(role_id3, role_name3, None)
rbac.create_client_role(client_def1, role_id1)
rbac.create_client_role(client_def1, role_id2)
rbac.create_client_role(client_def2, role_id2)
rbac.create_client_role(client_def2, role_id3)
rbac.delete_client_role(client_def1, role_id1)
rbac.delete_client_role(client_def2, role_id3)
self.assertEquals(rbac.client_def_to_role_id[client_def1],
set([role_id2]))
self.assertEquals(rbac.client_def_to_role_id[client_def2],
set([role_id2]))
rbac.delete_client_role(client_def1, role_id2)
rbac.delete_client_role(client_def2, role_id2)
self.assertEquals(rbac.client_def_to_role_id[client_def1], set([]))
self.assertEquals(rbac.client_def_to_role_id[client_def2], set([]))
