source.dussan.org Git - sonarqube.git/blob

   1 /*
   2  * SonarQube, open source software quality management tool.
   3  * Copyright (C) 2008-2014 SonarSource
   4  * mailto:contact AT sonarsource DOT com
   5  *
   6  * SonarQube is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * SonarQube is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  */
  20
  21 package org.sonar.server.permission.ws;
  22
  23 import com.google.common.base.Function;
  24 import java.util.List;
  25 import javax.annotation.Nonnull;
  26 import javax.annotation.Nullable;
  27 import org.junit.Before;
  28 import org.junit.Rule;
  29 import org.junit.Test;
  30 import org.junit.experimental.categories.Category;
  31 import org.junit.rules.ExpectedException;
  32 import org.sonar.api.utils.System2;
  33 import org.sonar.core.permission.GlobalPermissions;
  34 import org.sonar.db.DbClient;
  35 import org.sonar.db.DbSession;
  36 import org.sonar.db.DbTester;
  37 import org.sonar.db.permission.PermissionQuery;
  38 import org.sonar.db.permission.PermissionTemplateDto;
  39 import org.sonar.db.permission.UserWithPermissionDto;
  40 import org.sonar.db.user.UserDto;
  41 import org.sonar.server.component.ComponentFinder;
  42 import org.sonar.server.exceptions.BadRequestException;
  43 import org.sonar.server.exceptions.ForbiddenException;
  44 import org.sonar.server.exceptions.NotFoundException;
  45 import org.sonar.server.exceptions.UnauthorizedException;
  46 import org.sonar.server.tester.UserSessionRule;
  47 import org.sonar.server.ws.TestRequest;
  48 import org.sonar.server.ws.WsActionTester;
  49 import org.sonar.test.DbTests;
  50
  51 import static com.google.common.collect.FluentIterable.from;
  52 import static org.assertj.core.api.Assertions.assertThat;
  53 import static org.sonar.api.web.UserRole.CODEVIEWER;
  54 import static org.sonar.api.web.UserRole.ISSUE_ADMIN;
  55 import static org.sonar.db.permission.PermissionTemplateTesting.newPermissionTemplateDto;
  56 import static org.sonar.db.user.GroupMembershipQuery.IN;
  57 import static org.sonar.db.user.UserTesting.newUserDto;
  58
  59 @Category(DbTests.class)
  60 public class RemoveUserFromTemplateActionTest {
  61
  62   private static final String USER_LOGIN = "user-login";
  63   private static final String DEFAULT_PERMISSION = CODEVIEWER;
  64   @Rule
  65   public DbTester db = DbTester.create(System2.INSTANCE);
  66   @Rule
  67   public ExpectedException expectedException = ExpectedException.none();
  68   @Rule
  69   public UserSessionRule userSession = UserSessionRule.standalone();
  70
  71   WsActionTester ws;
  72   DbClient dbClient;
  73   DbSession dbSession;
  74   UserDto user;
  75   PermissionTemplateDto permissionTemplate;
  76
  77   @Before
  78   public void setUp() {
  79     dbClient = db.getDbClient();
  80     dbSession = db.getSession();
  81     userSession.login().setGlobalPermissions(GlobalPermissions.SYSTEM_ADMIN);
  82
  83     PermissionDependenciesFinder dependenciesFinder = new PermissionDependenciesFinder(dbClient, new ComponentFinder(dbClient));
  84     ws = new WsActionTester(new RemoveUserFromTemplateAction(dbClient, dependenciesFinder, userSession));
  85
  86     user = insertUser(newUserDto().setLogin(USER_LOGIN));
  87     permissionTemplate = insertPermissionTemplate(newPermissionTemplateDto());
  88     addUserToTemplate(user, permissionTemplate, DEFAULT_PERMISSION);
  89     commit();
  90   }
  91
  92   @Test
  93   public void remove_user_from_template() {
  94     newRequest(USER_LOGIN, permissionTemplate.getKee(), DEFAULT_PERMISSION);
  95
  96     assertThat(getLoginsInTemplateAndPermission(permissionTemplate.getId(), DEFAULT_PERMISSION)).isEmpty();
  97   }
  98
  99   @Test
 100   public void remove_user_from_template_twice_without_failing() {
 101     newRequest(USER_LOGIN, permissionTemplate.getKee(), DEFAULT_PERMISSION);
 102     newRequest(USER_LOGIN, permissionTemplate.getKee(), DEFAULT_PERMISSION);
 103
 104     assertThat(getLoginsInTemplateAndPermission(permissionTemplate.getId(), DEFAULT_PERMISSION)).isEmpty();
 105   }
 106
 107   @Test
 108   public void keep_user_permission_not_removed() {
 109     addUserToTemplate(user, permissionTemplate, ISSUE_ADMIN);
 110     commit();
 111
 112     newRequest(USER_LOGIN, permissionTemplate.getKee(), DEFAULT_PERMISSION);
 113
 114     assertThat(getLoginsInTemplateAndPermission(permissionTemplate.getId(), DEFAULT_PERMISSION)).isEmpty();
 115     assertThat(getLoginsInTemplateAndPermission(permissionTemplate.getId(), ISSUE_ADMIN)).containsExactly(user.getLogin());
 116   }
 117
 118   @Test
 119   public void keep_other_users_when_one_user_removed() {
 120     UserDto newUser = insertUser(newUserDto().setLogin("new-login"));
 121     addUserToTemplate(newUser, permissionTemplate, DEFAULT_PERMISSION);
 122     commit();
 123
 124     newRequest(USER_LOGIN, permissionTemplate.getKee(), DEFAULT_PERMISSION);
 125
 126     assertThat(getLoginsInTemplateAndPermission(permissionTemplate.getId(), DEFAULT_PERMISSION)).containsExactly("new-login");
 127   }
 128
 129   @Test
 130   public void fail_if_not_a_project_permission() {
 131     expectedException.expect(BadRequestException.class);
 132
 133     newRequest(USER_LOGIN, permissionTemplate.getKee(), GlobalPermissions.PREVIEW_EXECUTION);
 134   }
 135
 136   @Test
 137   public void fail_if_insufficient_privileges() {
 138     expectedException.expect(ForbiddenException.class);
 139     userSession.setGlobalPermissions(GlobalPermissions.QUALITY_PROFILE_ADMIN);
 140
 141     newRequest(USER_LOGIN, permissionTemplate.getKee(), DEFAULT_PERMISSION);
 142   }
 143
 144   @Test
 145   public void fail_if_not_logged_in() {
 146     expectedException.expect(UnauthorizedException.class);
 147     userSession.anonymous();
 148
 149     newRequest(USER_LOGIN, permissionTemplate.getKee(), DEFAULT_PERMISSION);
 150   }
 151
 152   @Test
 153   public void fail_if_user_missing() {
 154     expectedException.expect(IllegalArgumentException.class);
 155
 156     newRequest(null, permissionTemplate.getKee(), DEFAULT_PERMISSION);
 157   }
 158
 159   @Test
 160   public void fail_if_permission_missing() {
 161     expectedException.expect(IllegalArgumentException.class);
 162
 163     newRequest(USER_LOGIN, permissionTemplate.getKee(), null);
 164   }
 165
 166   @Test
 167   public void fail_if_template_key_missing() {
 168     expectedException.expect(IllegalArgumentException.class);
 169
 170     newRequest(USER_LOGIN, null, DEFAULT_PERMISSION);
 171   }
 172
 173   @Test
 174   public void fail_if_user_does_not_exist() {
 175     expectedException.expect(NotFoundException.class);
 176     expectedException.expectMessage("User with login 'unknown-login' is not found");
 177
 178     newRequest("unknown-login", permissionTemplate.getKee(), DEFAULT_PERMISSION);
 179   }
 180
 181   @Test
 182   public void fail_if_template_key_does_not_exist() {
 183     expectedException.expect(NotFoundException.class);
 184     expectedException.expectMessage("Permission template with id 'unknown-key' is not found");
 185
 186     newRequest(USER_LOGIN, "unknown-key", DEFAULT_PERMISSION);
 187   }
 188
 189   private void newRequest(@Nullable String userLogin, @Nullable String templateKey, @Nullable String permission) {
 190     TestRequest request = ws.newRequest();
 191     if (userLogin != null) {
 192       request.setParam(Parameters.PARAM_USER_LOGIN, userLogin);
 193     }
 194     if (templateKey != null) {
 195       request.setParam(Parameters.PARAM_TEMPLATE_ID_EXPLICIT, templateKey);
 196     }
 197     if (permission != null) {
 198       request.setParam(Parameters.PARAM_PERMISSION, permission);
 199     }
 200
 201     request.execute();
 202   }
 203
 204   private void commit() {
 205     dbSession.commit();
 206   }
 207
 208   private UserDto insertUser(UserDto userDto) {
 209     return dbClient.userDao().insert(dbSession, userDto.setActive(true));
 210   }
 211
 212   private PermissionTemplateDto insertPermissionTemplate(PermissionTemplateDto permissionTemplate) {
 213     return dbClient.permissionTemplateDao().insert(dbSession, permissionTemplate);
 214   }
 215
 216   private List<String> getLoginsInTemplateAndPermission(long templateId, String permission) {
 217     PermissionQuery permissionQuery = PermissionQuery.builder().permission(permission).membership(IN).build();
 218     return from(dbClient.permissionTemplateDao()
 219       .selectUsers(dbSession, permissionQuery, templateId, 0, Integer.MAX_VALUE))
 220       .transform(UserWithPermissionToUserLogin.INSTANCE)
 221       .toList();
 222   }
 223
 224   private enum UserWithPermissionToUserLogin implements Function<UserWithPermissionDto, String> {
 225     INSTANCE;
 226
 227     @Override
 228     public String apply(@Nonnull UserWithPermissionDto userWithPermission) {
 229       return userWithPermission.getLogin();
 230     }
 231
 232   }
 233
 234   private void addUserToTemplate(UserDto user, PermissionTemplateDto permissionTemplate, String permission) {
 235     dbClient.permissionTemplateDao().insertUserPermission(dbSession, permissionTemplate.getId(), user.getId(), permission);
 236   }
 237 }