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.template;
  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.permission.ws.PermissionDependenciesFinder;
  47 import org.sonar.server.permission.ws.WsPermissionParameters;
  48 import org.sonar.server.tester.UserSessionRule;
  49 import org.sonar.server.ws.TestRequest;
  50 import org.sonar.server.ws.WsActionTester;
  51 import org.sonar.test.DbTests;
  52
  53 import static com.google.common.collect.FluentIterable.from;
  54 import static org.assertj.core.api.Assertions.assertThat;
  55 import static org.sonar.api.web.UserRole.CODEVIEWER;
  56 import static org.sonar.api.web.UserRole.ISSUE_ADMIN;
  57 import static org.sonar.db.permission.PermissionTemplateTesting.newPermissionTemplateDto;
  58 import static org.sonar.db.user.GroupMembershipQuery.IN;
  59 import static org.sonar.db.user.UserTesting.newUserDto;
  60 import static org.sonar.server.permission.ws.WsPermissionParameters.PARAM_PERMISSION;
  61 import static org.sonar.server.permission.ws.WsPermissionParameters.PARAM_TEMPLATE_NAME;
  62 import static org.sonar.server.permission.ws.WsPermissionParameters.PARAM_USER_LOGIN;
  63
  64 @Category(DbTests.class)
  65 public class AddUserToTemplateActionTest {
  66
  67   private static final String USER_LOGIN = "user-login";
  68   @Rule
  69   public DbTester db = DbTester.create(System2.INSTANCE);
  70   @Rule
  71   public ExpectedException expectedException = ExpectedException.none();
  72   @Rule
  73   public UserSessionRule userSession = UserSessionRule.standalone();
  74
  75   WsActionTester ws;
  76   DbClient dbClient;
  77   DbSession dbSession;
  78   UserDto user;
  79   PermissionTemplateDto permissionTemplate;
  80
  81   @Before
  82   public void setUp() {
  83     dbClient = db.getDbClient();
  84     dbSession = db.getSession();
  85     userSession.login().setGlobalPermissions(GlobalPermissions.SYSTEM_ADMIN);
  86
  87     PermissionDependenciesFinder dependenciesFinder = new PermissionDependenciesFinder(dbClient, new ComponentFinder(dbClient));
  88     ws = new WsActionTester(new AddUserToTemplateAction(dbClient, dependenciesFinder, userSession));
  89
  90     user = insertUser(newUserDto().setLogin(USER_LOGIN));
  91     permissionTemplate = insertPermissionTemplate(newPermissionTemplateDto());
  92     commit();
  93   }
  94
  95   @Test
  96   public void add_user_to_template() {
  97     newRequest(USER_LOGIN, permissionTemplate.getUuid(), CODEVIEWER);
  98
  99     assertThat(getLoginsInTemplateAndPermission(permissionTemplate.getId(), CODEVIEWER)).containsExactly(USER_LOGIN);
 100   }
 101
 102   @Test
 103   public void add_user_to_template_by_name() {
 104     ws.newRequest()
 105       .setParam(PARAM_USER_LOGIN, USER_LOGIN)
 106       .setParam(PARAM_PERMISSION, CODEVIEWER)
 107       .setParam(PARAM_TEMPLATE_NAME, permissionTemplate.getName().toUpperCase())
 108       .execute();
 109     commit();
 110
 111     assertThat(getLoginsInTemplateAndPermission(permissionTemplate.getId(), CODEVIEWER)).containsExactly(USER_LOGIN);
 112   }
 113
 114   @Test
 115   public void does_not_add_a_user_twice() {
 116     newRequest(USER_LOGIN, permissionTemplate.getUuid(), ISSUE_ADMIN);
 117     newRequest(USER_LOGIN, permissionTemplate.getUuid(), ISSUE_ADMIN);
 118
 119     assertThat(getLoginsInTemplateAndPermission(permissionTemplate.getId(), ISSUE_ADMIN)).containsExactly(USER_LOGIN);
 120   }
 121
 122   @Test
 123   public void fail_if_not_a_project_permission() {
 124     expectedException.expect(BadRequestException.class);
 125
 126     newRequest(USER_LOGIN, permissionTemplate.getUuid(), GlobalPermissions.PREVIEW_EXECUTION);
 127   }
 128
 129   @Test
 130   public void fail_if_insufficient_privileges() {
 131     expectedException.expect(ForbiddenException.class);
 132     userSession.setGlobalPermissions(GlobalPermissions.QUALITY_PROFILE_ADMIN);
 133
 134     newRequest(USER_LOGIN, permissionTemplate.getUuid(), CODEVIEWER);
 135   }
 136
 137   @Test
 138   public void fail_if_not_logged_in() {
 139     expectedException.expect(UnauthorizedException.class);
 140     userSession.anonymous();
 141
 142     newRequest(USER_LOGIN, permissionTemplate.getUuid(), CODEVIEWER);
 143   }
 144
 145   @Test
 146   public void fail_if_user_missing() {
 147     expectedException.expect(IllegalArgumentException.class);
 148
 149     newRequest(null, permissionTemplate.getUuid(), CODEVIEWER);
 150   }
 151
 152   @Test
 153   public void fail_if_permission_missing() {
 154     expectedException.expect(IllegalArgumentException.class);
 155
 156     newRequest(USER_LOGIN, permissionTemplate.getUuid(), null);
 157   }
 158
 159   @Test
 160   public void fail_if_template_uuid_and_name_are_missing() {
 161     expectedException.expect(BadRequestException.class);
 162
 163     newRequest(USER_LOGIN, null, CODEVIEWER);
 164   }
 165
 166   @Test
 167   public void fail_if_user_does_not_exist() {
 168     expectedException.expect(NotFoundException.class);
 169     expectedException.expectMessage("User with login 'unknown-login' is not found");
 170
 171     newRequest("unknown-login", permissionTemplate.getUuid(), CODEVIEWER);
 172   }
 173
 174   @Test
 175   public void fail_if_template_key_does_not_exist() {
 176     expectedException.expect(NotFoundException.class);
 177     expectedException.expectMessage("Permission template with id 'unknown-key' is not found");
 178
 179     newRequest(USER_LOGIN, "unknown-key", CODEVIEWER);
 180   }
 181
 182   private void newRequest(@Nullable String userLogin, @Nullable String templateKey, @Nullable String permission) {
 183     TestRequest request = ws.newRequest();
 184     if (userLogin != null) {
 185       request.setParam(PARAM_USER_LOGIN, userLogin);
 186     }
 187     if (templateKey != null) {
 188       request.setParam(WsPermissionParameters.PARAM_TEMPLATE_UUID, templateKey);
 189     }
 190     if (permission != null) {
 191       request.setParam(WsPermissionParameters.PARAM_PERMISSION, permission);
 192     }
 193
 194     request.execute();
 195   }
 196
 197   private void commit() {
 198     dbSession.commit();
 199   }
 200
 201   private UserDto insertUser(UserDto userDto) {
 202     return dbClient.userDao().insert(dbSession, userDto.setActive(true));
 203   }
 204
 205   private PermissionTemplateDto insertPermissionTemplate(PermissionTemplateDto permissionTemplate) {
 206     return dbClient.permissionTemplateDao().insert(dbSession, permissionTemplate);
 207   }
 208
 209   private List<String> getLoginsInTemplateAndPermission(long templateId, String permission) {
 210     PermissionQuery permissionQuery = PermissionQuery.builder().permission(permission).membership(IN).build();
 211     return from(dbClient.permissionTemplateDao()
 212       .selectUsers(dbSession, permissionQuery, templateId, 0, Integer.MAX_VALUE))
 213       .transform(UserWithPermissionToUserLogin.INSTANCE)
 214       .toList();
 215   }
 216
 217   private enum UserWithPermissionToUserLogin implements Function<UserWithPermissionDto, String> {
 218     INSTANCE;
 219
 220     @Override
 221     public String apply(@Nonnull UserWithPermissionDto userWithPermission) {
 222       return userWithPermission.getLogin();
 223     }
 224
 225   }
 226 }