1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* SonarQube
* Copyright (C) 2009-2024 SonarSource SA
* mailto:info AT sonarsource DOT com
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 3 of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program; if not, write to the Free Software Foundation,
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
package org.sonarqube.ws.tester;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Consumer;
import org.sonarqube.ws.Permissions.PermissionTemplate;
import org.sonarqube.ws.Permissions.SearchTemplatesWsResponse.TemplateIdQualifier;
import org.sonarqube.ws.Projects.CreateWsResponse.Project;
import org.sonarqube.ws.Users;
import org.sonarqube.ws.client.permissions.AddGroupRequest;
import org.sonarqube.ws.client.permissions.AddGroupToTemplateRequest;
import org.sonarqube.ws.client.permissions.AddProjectCreatorToTemplateRequest;
import org.sonarqube.ws.client.permissions.AddUserRequest;
import org.sonarqube.ws.client.permissions.AddUserToTemplateRequest;
import org.sonarqube.ws.client.permissions.ApplyTemplateRequest;
import org.sonarqube.ws.client.permissions.CreateTemplateRequest;
import org.sonarqube.ws.client.permissions.PermissionsService;
import org.sonarqube.ws.client.permissions.SearchTemplatesRequest;
import org.sonarqube.ws.client.permissions.SetDefaultTemplateRequest;
import static com.sonar.orchestrator.container.Server.ADMIN_LOGIN;
import static java.util.Arrays.stream;
public class PermissionTester {
private static final AtomicInteger ID_GENERATOR = new AtomicInteger();
private final TesterSession session;
PermissionTester(TesterSession session) {
this.session = session;
}
@SafeVarargs
public final PermissionTemplate generateTemplate(Consumer<CreateTemplateRequest>... populators) {
int id = ID_GENERATOR.getAndIncrement();
String name = "template" + id;
CreateTemplateRequest request = new CreateTemplateRequest()
.setName(name);
stream(populators).forEach(p -> p.accept(request));
PermissionTemplate template = service().createTemplate(request).getPermissionTemplate();
// Give browse and admin permissions to admin in order to allow admin wsclient to perform any operation on created projects
addUserToTemplate(ADMIN_LOGIN, template, "user");
addUserToTemplate(ADMIN_LOGIN, template, "admin");
return template;
}
public void addUserToTemplate(Users.CreateWsResponse.User user, PermissionTemplate template, String permission) {
addUserToTemplate(user.getLogin(), template, permission);
}
public void addUserToTemplate(String login, PermissionTemplate template, String permission) {
service().addUserToTemplate(new AddUserToTemplateRequest()
.setLogin(login)
.setTemplateName(template.getName())
.setPermission(permission));
}
public void addGroup(String groupName, String permission) {
service().addGroup(new AddGroupRequest().setGroupName(groupName).setPermission(permission));
}
public void addUser(String login, String permission) {
service().addUser(new AddUserRequest().setLogin(login).setPermission(permission));
}
public void addGroupToTemplate(String groupName, PermissionTemplate template, String permission) {
service().addGroupToTemplate(new AddGroupToTemplateRequest()
.setGroupName(groupName)
.setTemplateName(template.getName())
.setPermission(permission));
}
public void addCreatorToTemplate(PermissionTemplate template, String permission) {
this.service().addProjectCreatorToTemplate(
new AddProjectCreatorToTemplateRequest()
.setPermission(permission)
.setTemplateId(template.getId()));
}
public void applyTemplate(PermissionTemplate template, Project project) {
service().applyTemplate(
new ApplyTemplateRequest()
.setTemplateName(template.getName())
.setProjectKey(project.getKey()));
}
public TemplateIdQualifier getDefaultTemplateForProject() {
return service().searchTemplates(new SearchTemplatesRequest()).getDefaultTemplatesList()
.stream()
.filter(t -> t.getQualifier().equals("TRK"))
.findFirst()
.orElseThrow(() -> {
throw new IllegalStateException("Cannot find default template for project");
});
}
public void setDefaultTemplate(TemplateIdQualifier template) {
service().setDefaultTemplate(new SetDefaultTemplateRequest().setTemplateId(template.getTemplateId()));
}
public void setDefaultTemplate(PermissionTemplate template) {
service().setDefaultTemplate(new SetDefaultTemplateRequest().setTemplateId(template.getId()));
}
public PermissionsService service() {
return session.wsClient().permissions();
}
}
|
