You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

ServerUserSessionTest.java 25KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604
  1. /*
  2. * SonarQube
  3. * Copyright (C) 2009-2020 SonarSource SA
  4. * mailto:info AT sonarsource DOT com
  5. *
  6. * This program 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. * This program 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. package org.sonar.server.user;
  21. import java.util.Arrays;
  22. import javax.annotation.Nullable;
  23. import org.junit.Rule;
  24. import org.junit.Test;
  25. import org.junit.rules.ExpectedException;
  26. import org.sonar.api.utils.System2;
  27. import org.sonar.api.web.UserRole;
  28. import org.sonar.db.DbClient;
  29. import org.sonar.db.DbTester;
  30. import org.sonar.db.component.ComponentDto;
  31. import org.sonar.db.user.GroupDto;
  32. import org.sonar.db.user.UserDto;
  33. import org.sonar.server.exceptions.ForbiddenException;
  34. import static com.google.common.base.Preconditions.checkState;
  35. import static java.util.Arrays.asList;
  36. import static org.assertj.core.api.Assertions.assertThat;
  37. import static org.sonar.core.permission.GlobalPermissions.PROVISIONING;
  38. import static org.sonar.core.permission.GlobalPermissions.SYSTEM_ADMIN;
  39. import static org.sonar.db.component.ComponentTesting.newChildComponent;
  40. import static org.sonar.db.component.ComponentTesting.newFileDto;
  41. import static org.sonar.db.permission.OrganizationPermission.ADMINISTER;
  42. import static org.sonar.db.permission.OrganizationPermission.PROVISION_PROJECTS;
  43. import static org.sonar.db.permission.OrganizationPermission.SCAN;
  44. public class ServerUserSessionTest {
  45. @Rule
  46. public DbTester db = DbTester.create(System2.INSTANCE);
  47. @Rule
  48. public ExpectedException expectedException = ExpectedException.none();
  49. private DbClient dbClient = db.getDbClient();
  50. @Test
  51. public void anonymous_is_not_logged_in_and_does_not_have_login() {
  52. UserSession session = newAnonymousSession();
  53. assertThat(session.getLogin()).isNull();
  54. assertThat(session.getUuid()).isNull();
  55. assertThat(session.isLoggedIn()).isFalse();
  56. }
  57. @Test
  58. public void getGroups_is_empty_on_anonymous() {
  59. assertThat(newAnonymousSession().getGroups()).isEmpty();
  60. }
  61. @Test
  62. public void getGroups_is_empty_if_user_is_not_member_of_any_group() {
  63. UserDto user = db.users().insertUser();
  64. assertThat(newUserSession(user).getGroups()).isEmpty();
  65. }
  66. @Test
  67. public void getGroups_returns_the_groups_of_logged_in_user() {
  68. UserDto user = db.users().insertUser();
  69. GroupDto group1 = db.users().insertGroup();
  70. GroupDto group2 = db.users().insertGroup();
  71. db.users().insertMember(group1, user);
  72. db.users().insertMember(group2, user);
  73. assertThat(newUserSession(user).getGroups()).extracting(GroupDto::getUuid).containsOnly(group1.getUuid(), group2.getUuid());
  74. }
  75. @Test
  76. public void getGroups_keeps_groups_in_cache() {
  77. UserDto user = db.users().insertUser();
  78. GroupDto group1 = db.users().insertGroup();
  79. GroupDto group2 = db.users().insertGroup();
  80. db.users().insertMember(group1, user);
  81. ServerUserSession session = newUserSession(user);
  82. assertThat(session.getGroups()).extracting(GroupDto::getUuid).containsOnly(group1.getUuid());
  83. // membership updated but not cache
  84. db.users().insertMember(group2, user);
  85. assertThat(session.getGroups()).extracting(GroupDto::getUuid).containsOnly(group1.getUuid());
  86. }
  87. @Test
  88. public void isRoot_is_false_is_flag_root_is_false_on_UserDto() {
  89. UserDto root = db.users().insertUser();
  90. root = db.users().makeRoot(root);
  91. assertThat(newUserSession(root).isRoot()).isTrue();
  92. UserDto notRoot = db.users().insertUser();
  93. assertThat(newUserSession(notRoot).isRoot()).isFalse();
  94. }
  95. @Test
  96. public void checkIsRoot_throws_IPFE_if_flag_root_is_false_on_UserDto() {
  97. UserDto user = db.users().insertUser();
  98. UserSession underTest = newUserSession(user);
  99. expectInsufficientPrivilegesForbiddenException();
  100. underTest.checkIsRoot();
  101. }
  102. @Test
  103. public void checkIsRoot_does_not_fail_if_flag_root_is_true_on_UserDto() {
  104. UserDto root = db.users().insertUser();
  105. root = db.users().makeRoot(root);
  106. UserSession underTest = newUserSession(root);
  107. assertThat(underTest.checkIsRoot()).isSameAs(underTest);
  108. }
  109. @Test
  110. public void hasComponentUuidPermission_returns_true_when_flag_root_is_true_on_UserDto_no_matter_if_user_has_project_permission_for_given_uuid() {
  111. UserDto root = db.users().insertUser();
  112. root = db.users().makeRoot(root);
  113. ComponentDto project = db.components().insertPrivateProject();
  114. ComponentDto file = db.components().insertComponent(newFileDto(project));
  115. UserSession underTest = newUserSession(root);
  116. assertThat(underTest.hasComponentUuidPermission(UserRole.USER, file.uuid())).isTrue();
  117. assertThat(underTest.hasComponentUuidPermission(UserRole.CODEVIEWER, file.uuid())).isTrue();
  118. assertThat(underTest.hasComponentUuidPermission(UserRole.ADMIN, file.uuid())).isTrue();
  119. assertThat(underTest.hasComponentUuidPermission("whatever", "who cares?")).isTrue();
  120. }
  121. @Test
  122. public void checkComponentUuidPermission_succeeds_if_user_has_permission_for_specified_uuid_in_db() {
  123. UserDto root = db.users().insertUser();
  124. root = db.users().makeRoot(root);
  125. ComponentDto project = db.components().insertPrivateProject();
  126. ComponentDto file = db.components().insertComponent(newFileDto(project));
  127. UserSession underTest = newUserSession(root);
  128. assertThat(underTest.checkComponentUuidPermission(UserRole.USER, file.uuid())).isSameAs(underTest);
  129. assertThat(underTest.checkComponentUuidPermission("whatever", "who cares?")).isSameAs(underTest);
  130. }
  131. @Test
  132. public void checkComponentUuidPermission_fails_with_FE_when_user_has_not_permission_for_specified_uuid_in_db() {
  133. UserDto user = db.users().insertUser();
  134. ComponentDto project = db.components().insertPrivateProject();
  135. db.users().insertProjectPermissionOnUser(user, UserRole.USER, project);
  136. UserSession session = newUserSession(user);
  137. expectInsufficientPrivilegesForbiddenException();
  138. session.checkComponentUuidPermission(UserRole.USER, "another-uuid");
  139. }
  140. @Test
  141. public void checkPermission_throws_ForbiddenException_when_user_doesnt_have_the_specified_permission() {
  142. UserDto user = db.users().insertUser();
  143. expectInsufficientPrivilegesForbiddenException();
  144. newUserSession(user).checkPermission(PROVISION_PROJECTS);
  145. }
  146. @Test
  147. public void checkPermission_succeeds_when_user_has_the_specified_permission() {
  148. UserDto root = db.users().insertUser();
  149. root = db.users().makeRoot(root);
  150. db.users().insertPermissionOnUser(root, PROVISIONING);
  151. newUserSession(root).checkPermission(PROVISION_PROJECTS);
  152. }
  153. @Test
  154. public void checkPermission_succeeds_when_user_is_root() {
  155. UserDto root = db.users().insertUser();
  156. root = db.users().makeRoot(root);
  157. newUserSession(root).checkPermission(PROVISION_PROJECTS);
  158. }
  159. @Test
  160. public void test_hasPermission_for_logged_in_user() {
  161. ComponentDto project = db.components().insertPrivateProject();
  162. UserDto user = db.users().insertUser();
  163. db.users().insertPermissionOnUser(user, PROVISION_PROJECTS);
  164. db.users().insertProjectPermissionOnUser(user, UserRole.ADMIN, project);
  165. UserSession session = newUserSession(user);
  166. assertThat(session.hasPermission(PROVISION_PROJECTS)).isTrue();
  167. assertThat(session.hasPermission(ADMINISTER)).isFalse();
  168. }
  169. @Test
  170. public void test_hasPermission_for_anonymous_user() {
  171. db.users().insertPermissionOnAnyone(PROVISION_PROJECTS);
  172. UserSession session = newAnonymousSession();
  173. assertThat(session.hasPermission(PROVISION_PROJECTS)).isTrue();
  174. assertThat(session.hasPermission(ADMINISTER)).isFalse();
  175. }
  176. @Test
  177. public void hasPermission_keeps_cache_of_permissions_of_logged_in_user() {
  178. UserDto user = db.users().insertUser();
  179. db.users().insertPermissionOnUser(user, PROVISIONING);
  180. UserSession session = newUserSession(user);
  181. // feed the cache
  182. assertThat(session.hasPermission(PROVISION_PROJECTS)).isTrue();
  183. // change permissions without updating the cache
  184. db.users().deletePermissionFromUser(user, PROVISION_PROJECTS);
  185. db.users().insertPermissionOnUser(user, SCAN);
  186. assertThat(session.hasPermission(PROVISION_PROJECTS)).isTrue();
  187. assertThat(session.hasPermission(ADMINISTER)).isFalse();
  188. assertThat(session.hasPermission(SCAN)).isFalse();
  189. }
  190. @Test
  191. public void hasPermission_keeps_cache_of_permissions_of_anonymous_user() {
  192. db.users().insertPermissionOnAnyone(PROVISION_PROJECTS);
  193. UserSession session = newAnonymousSession();
  194. // feed the cache
  195. assertThat(session.hasPermission(PROVISION_PROJECTS)).isTrue();
  196. // change permissions without updating the cache
  197. db.users().insertPermissionOnAnyone(SCAN);
  198. assertThat(session.hasPermission(PROVISION_PROJECTS)).isTrue();
  199. assertThat(session.hasPermission(SCAN)).isFalse();
  200. }
  201. @Test
  202. public void hasComponentPermissionByDtoOrUuid_returns_true_for_anonymous_user_for_permissions_USER_and_CODEVIEWER_on_public_projects_without_permissions() {
  203. ComponentDto publicProject = db.components().insertPublicProject();
  204. ServerUserSession underTest = newAnonymousSession();
  205. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.USER, publicProject)).isTrue();
  206. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.CODEVIEWER, publicProject)).isTrue();
  207. }
  208. @Test
  209. public void hasComponentPermissionByDtoOrUuid_returns_true_for_anonymous_user_for_permissions_USER_and_CODEVIEWER_on_public_projects_with_global_permissions() {
  210. ComponentDto publicProject = db.components().insertPublicProject();
  211. db.users().insertProjectPermissionOnAnyone("p1", publicProject);
  212. ServerUserSession underTest = newAnonymousSession();
  213. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.USER, publicProject)).isTrue();
  214. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.CODEVIEWER, publicProject)).isTrue();
  215. }
  216. @Test
  217. public void hasComponentPermissionByDtoOrUuid_returns_true_for_anonymous_user_for_permissions_USER_and_CODEVIEWER_on_public_projects_with_group_permissions() {
  218. ComponentDto publicProject = db.components().insertPublicProject();
  219. db.users().insertProjectPermissionOnGroup(db.users().insertGroup(), "p1", publicProject);
  220. ServerUserSession underTest = newAnonymousSession();
  221. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.USER, publicProject)).isTrue();
  222. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.CODEVIEWER, publicProject)).isTrue();
  223. }
  224. @Test
  225. public void hasComponentPermissionByDtoOrUuid_returns_true_for_anonymous_user_for_permissions_USER_and_CODEVIEWER_on_public_projects_with_user_permissions() {
  226. ComponentDto publicProject = db.components().insertPublicProject();
  227. db.users().insertProjectPermissionOnUser(db.users().insertUser(), "p1", publicProject);
  228. ServerUserSession underTest = newAnonymousSession();
  229. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.USER, publicProject)).isTrue();
  230. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.CODEVIEWER, publicProject)).isTrue();
  231. }
  232. @Test
  233. public void hasComponentPermissionByDtoOrUuid_returns_false_for_authenticated_user_for_permissions_USER_and_CODEVIEWER_on_private_projects_without_permissions() {
  234. UserDto user = db.users().insertUser();
  235. ComponentDto privateProject = db.components().insertPrivateProject();
  236. ServerUserSession underTest = newUserSession(user);
  237. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.USER, privateProject)).isFalse();
  238. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.CODEVIEWER, privateProject)).isFalse();
  239. }
  240. @Test
  241. public void hasComponentPermissionByDtoOrUuid_returns_false_for_authenticated_user_for_permissions_USER_and_CODEVIEWER_on_private_projects_with_group_permissions() {
  242. UserDto user = db.users().insertUser();
  243. ComponentDto privateProject = db.components().insertPrivateProject();
  244. db.users().insertProjectPermissionOnGroup(db.users().insertGroup(), "p1", privateProject);
  245. ServerUserSession underTest = newUserSession(user);
  246. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.USER, privateProject)).isFalse();
  247. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.CODEVIEWER, privateProject)).isFalse();
  248. }
  249. @Test
  250. public void hasComponentPermissionByDtoOrUuid_returns_false_for_authenticated_user_for_permissions_USER_and_CODEVIEWER_on_private_projects_with_user_permissions() {
  251. UserDto user = db.users().insertUser();
  252. ComponentDto privateProject = db.components().insertPrivateProject();
  253. db.users().insertProjectPermissionOnUser(db.users().insertUser(), "p1", privateProject);
  254. ServerUserSession underTest = newUserSession(user);
  255. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.USER, privateProject)).isFalse();
  256. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.CODEVIEWER, privateProject)).isFalse();
  257. }
  258. @Test
  259. public void hasComponentPermissionByDtoOrUuid_returns_true_for_anonymous_user_for_inserted_permissions_on_group_AnyOne_on_public_projects() {
  260. ComponentDto publicProject = db.components().insertPublicProject();
  261. db.users().insertProjectPermissionOnAnyone("p1", publicProject);
  262. ServerUserSession underTest = newAnonymousSession();
  263. assertThat(hasComponentPermissionByDtoOrUuid(underTest, "p1", publicProject)).isTrue();
  264. }
  265. @Test
  266. public void hasComponentPermissionByDtoOrUuid_returns_false_for_anonymous_user_for_inserted_permissions_on_group_on_public_projects() {
  267. ComponentDto publicProject = db.components().insertPublicProject();
  268. GroupDto group = db.users().insertGroup();
  269. db.users().insertProjectPermissionOnGroup(group, "p1", publicProject);
  270. ServerUserSession underTest = newAnonymousSession();
  271. assertThat(hasComponentPermissionByDtoOrUuid(underTest, "p1", publicProject)).isFalse();
  272. }
  273. @Test
  274. public void hasComponentPermissionByDtoOrUuid_returns_false_for_anonymous_user_for_inserted_permissions_on_group_on_private_projects() {
  275. ComponentDto privateProject = db.components().insertPrivateProject();
  276. GroupDto group = db.users().insertGroup();
  277. db.users().insertProjectPermissionOnGroup(group, "p1", privateProject);
  278. ServerUserSession underTest = newAnonymousSession();
  279. assertThat(hasComponentPermissionByDtoOrUuid(underTest, "p1", privateProject)).isFalse();
  280. }
  281. @Test
  282. public void hasComponentPermissionByDtoOrUuid_returns_false_for_anonymous_user_for_inserted_permissions_on_user_on_public_projects() {
  283. UserDto user = db.users().insertUser();
  284. ComponentDto publicProject = db.components().insertPublicProject();
  285. db.users().insertProjectPermissionOnUser(user, "p1", publicProject);
  286. ServerUserSession underTest = newAnonymousSession();
  287. assertThat(hasComponentPermissionByDtoOrUuid(underTest, "p1", publicProject)).isFalse();
  288. }
  289. @Test
  290. public void hasComponentPermissionByDtoOrUuid_returns_false_for_anonymous_user_for_inserted_permissions_on_user_on_private_projects() {
  291. UserDto user = db.users().insertUser();
  292. ComponentDto project = db.components().insertPrivateProject();
  293. db.users().insertProjectPermissionOnUser(user, "p1", project);
  294. ServerUserSession underTest = newAnonymousSession();
  295. assertThat(hasComponentPermissionByDtoOrUuid(underTest, "p1", project)).isFalse();
  296. }
  297. @Test
  298. public void hasComponentPermissionByDtoOrUuid_returns_true_for_any_project_or_permission_for_root_user() {
  299. UserDto root = db.users().insertUser();
  300. root = db.users().makeRoot(root);
  301. ComponentDto publicProject = db.components().insertPublicProject();
  302. ServerUserSession underTest = newUserSession(root);
  303. assertThat(hasComponentPermissionByDtoOrUuid(underTest, "does not matter", publicProject)).isTrue();
  304. }
  305. @Test
  306. public void hasComponentPermissionByDtoOrUuid_keeps_cache_of_permissions_of_logged_in_user() {
  307. UserDto user = db.users().insertUser();
  308. ComponentDto publicProject = db.components().insertPublicProject();
  309. db.users().insertProjectPermissionOnUser(user, UserRole.ADMIN, publicProject);
  310. UserSession underTest = newUserSession(user);
  311. // feed the cache
  312. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.ADMIN, publicProject)).isTrue();
  313. // change permissions without updating the cache
  314. db.users().deletePermissionFromUser(publicProject, user, UserRole.ADMIN);
  315. db.users().insertProjectPermissionOnUser(user, UserRole.ISSUE_ADMIN, publicProject);
  316. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.ADMIN, publicProject)).isTrue();
  317. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.ISSUE_ADMIN, publicProject)).isFalse();
  318. }
  319. @Test
  320. public void hasComponentPermissionByDtoOrUuid_keeps_cache_of_permissions_of_anonymous_user() {
  321. ComponentDto publicProject = db.components().insertPublicProject();
  322. db.users().insertProjectPermissionOnAnyone(UserRole.ADMIN, publicProject);
  323. UserSession underTest = newAnonymousSession();
  324. // feed the cache
  325. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.ADMIN, publicProject)).isTrue();
  326. // change permissions without updating the cache
  327. db.users().deleteProjectPermissionFromAnyone(publicProject, UserRole.ADMIN);
  328. db.users().insertProjectPermissionOnAnyone(UserRole.ISSUE_ADMIN, publicProject);
  329. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.ADMIN, publicProject)).isTrue();
  330. assertThat(hasComponentPermissionByDtoOrUuid(underTest, UserRole.ISSUE_ADMIN, publicProject)).isFalse();
  331. }
  332. private boolean hasComponentPermissionByDtoOrUuid(UserSession underTest, String permission, ComponentDto component) {
  333. boolean b1 = underTest.hasComponentPermission(permission, component);
  334. boolean b2 = underTest.hasComponentUuidPermission(permission, component.uuid());
  335. checkState(b1 == b2, "Different behaviors");
  336. return b1;
  337. }
  338. @Test
  339. public void keepAuthorizedComponents_returns_empty_list_if_no_permissions_are_granted() {
  340. ComponentDto publicProject = db.components().insertPublicProject();
  341. ComponentDto privateProject = db.components().insertPrivateProject();
  342. UserSession underTest = newAnonymousSession();
  343. assertThat(underTest.keepAuthorizedComponents(UserRole.ADMIN, Arrays.asList(privateProject, publicProject))).isEmpty();
  344. }
  345. @Test
  346. public void keepAuthorizedComponents_filters_components_with_granted_permissions_for_logged_in_user() {
  347. UserDto user = db.users().insertUser();
  348. ComponentDto publicProject = db.components().insertPublicProject();
  349. ComponentDto privateProject = db.components().insertPrivateProject();
  350. db.users().insertProjectPermissionOnUser(user, UserRole.ADMIN, privateProject);
  351. UserSession underTest = newUserSession(user);
  352. assertThat(underTest.keepAuthorizedComponents(UserRole.ISSUE_ADMIN, Arrays.asList(privateProject, publicProject))).isEmpty();
  353. assertThat(underTest.keepAuthorizedComponents(UserRole.ADMIN, Arrays.asList(privateProject, publicProject))).containsExactly(privateProject);
  354. }
  355. @Test
  356. public void keepAuthorizedComponents_filters_components_with_granted_permissions_for_anonymous() {
  357. ComponentDto publicProject = db.components().insertPublicProject();
  358. ComponentDto privateProject = db.components().insertPrivateProject();
  359. db.users().insertProjectPermissionOnAnyone(UserRole.ISSUE_ADMIN, publicProject);
  360. UserSession underTest = newAnonymousSession();
  361. assertThat(underTest.keepAuthorizedComponents(UserRole.ADMIN, Arrays.asList(privateProject, publicProject))).isEmpty();
  362. assertThat(underTest.keepAuthorizedComponents(UserRole.ISSUE_ADMIN, Arrays.asList(privateProject, publicProject))).containsExactly(publicProject);
  363. }
  364. @Test
  365. public void keepAuthorizedComponents_returns_all_specified_components_if_root() {
  366. UserDto root = db.users().insertUser();
  367. root = db.users().makeRoot(root);
  368. ComponentDto publicProject = db.components().insertPublicProject();
  369. ComponentDto privateProject = db.components().insertPrivateProject();
  370. UserSession underTest = newUserSession(root);
  371. assertThat(underTest.keepAuthorizedComponents(UserRole.ADMIN, Arrays.asList(privateProject, publicProject)))
  372. .containsExactly(privateProject, publicProject);
  373. }
  374. @Test
  375. public void keepAuthorizedComponents_on_branches() {
  376. UserDto user = db.users().insertUser();
  377. ComponentDto privateProject = db.components().insertPrivateProject();
  378. db.users().insertProjectPermissionOnUser(user, UserRole.ADMIN, privateProject);
  379. ComponentDto privateBranchProject = db.components().insertProjectBranch(privateProject);
  380. UserSession underTest = newUserSession(user);
  381. assertThat(underTest.keepAuthorizedComponents(UserRole.ADMIN, asList(privateProject, privateBranchProject)))
  382. .containsExactlyInAnyOrder(privateProject, privateBranchProject);
  383. }
  384. @Test
  385. public void isSystemAdministrator_returns_true_if_org_feature_is_enabled_and_user_is_root() {
  386. UserDto root = db.users().insertUser();
  387. root = db.users().makeRoot(root);
  388. UserSession session = newUserSession(root);
  389. assertThat(session.isSystemAdministrator()).isTrue();
  390. }
  391. @Test
  392. public void isSystemAdministrator_returns_false_if_org_feature_is_enabled_and_user_is_not_root() {
  393. UserDto user = db.users().insertUser();
  394. UserSession session = newUserSession(user);
  395. assertThat(session.isSystemAdministrator()).isFalse();
  396. }
  397. @Test
  398. public void isSystemAdministrator_returns_true_if_user_is_administrator() {
  399. UserDto user = db.users().insertUser();
  400. db.users().insertPermissionOnUser(user, SYSTEM_ADMIN);
  401. UserSession session = newUserSession(user);
  402. assertThat(session.isSystemAdministrator()).isTrue();
  403. }
  404. @Test
  405. public void isSystemAdministrator_returns_false_if_user_is_not_administrator() {
  406. UserDto user = db.users().insertUser();
  407. db.users().insertPermissionOnUser(user, PROVISIONING);
  408. UserSession session = newUserSession(user);
  409. assertThat(session.isSystemAdministrator()).isFalse();
  410. }
  411. @Test
  412. public void keep_isSystemAdministrator_flag_in_cache() {
  413. UserDto user = db.users().insertUser();
  414. db.users().insertPermissionOnUser(user, SYSTEM_ADMIN);
  415. UserSession session = newUserSession(user);
  416. session.checkIsSystemAdministrator();
  417. db.getDbClient().userDao().deactivateUser(db.getSession(), user);
  418. db.commit();
  419. // should fail but succeeds because flag is kept in cache
  420. session.checkIsSystemAdministrator();
  421. }
  422. @Test
  423. public void checkIsSystemAdministrator_succeeds_if_system_administrator() {
  424. UserDto root = db.users().insertUser();
  425. root = db.users().makeRoot(root);
  426. UserSession session = newUserSession(root);
  427. session.checkIsSystemAdministrator();
  428. }
  429. @Test
  430. public void checkIsSystemAdministrator_throws_ForbiddenException_if_not_system_administrator() {
  431. UserDto user = db.users().insertUser();
  432. UserSession session = newUserSession(user);
  433. expectedException.expect(ForbiddenException.class);
  434. expectedException.expectMessage("Insufficient privileges");
  435. session.checkIsSystemAdministrator();
  436. }
  437. @Test
  438. public void hasComponentPermission_on_branch_checks_permissions_of_its_project() {
  439. UserDto user = db.users().insertUser();
  440. ComponentDto privateProject = db.components().insertPrivateProject();
  441. ComponentDto branch = db.components().insertProjectBranch(privateProject, b -> b.setKey("feature/foo"));
  442. ComponentDto fileInBranch = db.components().insertComponent(newChildComponent("fileUuid", branch, branch));
  443. // permissions are defined on the project, not on the branch
  444. db.users().insertProjectPermissionOnUser(user, "p1", privateProject);
  445. UserSession underTest = newUserSession(user);
  446. assertThat(hasComponentPermissionByDtoOrUuid(underTest, "p1", privateProject)).isTrue();
  447. assertThat(hasComponentPermissionByDtoOrUuid(underTest, "p1", branch)).isTrue();
  448. assertThat(hasComponentPermissionByDtoOrUuid(underTest, "p1", fileInBranch)).isTrue();
  449. }
  450. private ServerUserSession newUserSession(@Nullable UserDto userDto) {
  451. return new ServerUserSession(dbClient, userDto);
  452. }
  453. private ServerUserSession newAnonymousSession() {
  454. return newUserSession(null);
  455. }
  456. private void expectInsufficientPrivilegesForbiddenException() {
  457. expectedException.expect(ForbiddenException.class);
  458. expectedException.expectMessage("Insufficient privileges");
  459. }
  460. }