def test_add_and_remove_user_role(self):
context = self.context
tenant_id = context["tenant"]["id"]
user_id = context["user"]["id"]
fake_role = mock.MagicMock()
self.mock_identity.return_value.create_role.return_value = fake_role
scenario = basic.AddAndRemoveUserRole(context)
scenario.run()
self.mock_identity.return_value.create_role.assert_called_once_with()
self.mock_identity.return_value.add_role.assert_called_once_with(
role_id=fake_role.id, user_id=user_id, project_id=tenant_id)
self.mock_identity.return_value.revoke_role.assert_called_once_with(
fake_role.id, user_id=user_id, project_id=tenant_id)
def test_assign_and_remove_user_role(self):
context = self._get_context()
scenario = basic.AddAndRemoveUserRole(context)
fake_tenant = context["tenant"]["id"]
fake_user = context["user"]["id"]
fake_role = mock.MagicMock()
scenario._tenant_create = mock.MagicMock(return_value=fake_tenant)
scenario._user_create = mock.MagicMock(return_value=fake_user)
scenario._role_create = mock.MagicMock(return_value=fake_role)
scenario._role_add = mock.MagicMock()
scenario._role_remove = mock.MagicMock()
scenario.run()
scenario._role_create.assert_called_once_with()
scenario._role_add.assert_called_once_with(fake_user, fake_role,
fake_tenant)
scenario._role_remove.assert_called_once_with(fake_user, fake_role,
fake_tenant)