/*
* SonarQube, open source software quality management tool.
* Copyright (C) 2008-2014 SonarSource
* mailto:contact AT sonarsource DOT com
*
* SonarQube 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.
*
* SonarQube 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.sonar.server.computation;
import org.junit.After;
import org.junit.Before;
import org.junit.ClassRule;
import org.junit.Test;
import org.sonar.api.issue.Issue;
import org.sonar.api.security.DefaultGroups;
import org.sonar.api.web.UserRole;
import org.sonar.core.component.ComponentDto;
import org.sonar.core.computation.db.AnalysisReportDto;
import org.sonar.core.issue.db.IssueDto;
import org.sonar.core.permission.GlobalPermissions;
import org.sonar.core.permission.PermissionFacade;
import org.sonar.core.persistence.DbSession;
import org.sonar.core.persistence.MyBatis;
import org.sonar.core.rule.RuleDto;
import org.sonar.core.user.UserDto;
import org.sonar.server.component.ComponentTesting;
import org.sonar.server.component.SnapshotTesting;
import org.sonar.server.db.DbClient;
import org.sonar.server.issue.IssueQuery;
import org.sonar.server.issue.IssueTesting;
import org.sonar.server.issue.db.IssueDao;
import org.sonar.server.issue.index.IssueIndex;
import org.sonar.server.platform.BackendCleanup;
import org.sonar.server.rule.RuleTesting;
import org.sonar.server.rule.db.RuleDao;
import org.sonar.server.search.IndexDefinition;
import org.sonar.server.search.QueryContext;
import org.sonar.server.search.Result;
import org.sonar.server.tester.ServerTester;
import org.sonar.server.user.MockUserSession;
import java.util.Date;
import java.util.List;
import static com.google.common.collect.Lists.newArrayList;
import static org.fest.assertions.Assertions.assertThat;
public class IndexProjectIssuesStepTest {
private static final String DEFAULT_PROJECT_KEY = "123456789-987654321";
@ClassRule
public static ServerTester tester = new ServerTester();
private IndexProjectIssuesStep sut;
private AnalysisReportQueue queue;
private DbClient db;
private DbSession session;
private MockUserSession userSession;
@Before
public void setUp() throws Exception {
tester.clearDbAndIndexes();
db = tester.get(DbClient.class);
session = db.openSession(false);
queue = tester.get(AnalysisReportQueue.class);
sut = tester.get(IndexProjectIssuesStep.class);
UserDto connectedUser = new UserDto().setLogin("gandalf").setName("Gandalf");
db.userDao().insert(session, connectedUser);
userSession = MockUserSession.set()
.setLogin(connectedUser.getLogin())
.setUserId(connectedUser.getId().intValue())
.setGlobalPermissions(GlobalPermissions.SCAN_EXECUTION);
session.commit();
}
@After
public void after() {
MyBatis.closeQuietly(session);
}
@Test
public void add_issues_in_index() {
ComponentDto project = insertPermissionsForProject(DEFAULT_PROJECT_KEY);
db.issueAuthorizationDao().synchronizeAfter(session, new Date(0));
ComponentDto file = ComponentTesting.newFileDto(project);
db.componentDao().insert(session, file);
db.snapshotDao().insert(session, SnapshotTesting.createForComponent(file, project));
RuleDto rule = RuleTesting.newXooX1();
tester.get(RuleDao.class).insert(session, rule);
IssueDto issue = IssueTesting.newDto(rule, file, project);
db.issueDao().insert(session, issue);
session.commit();
clearIssueIndexToSimulateBatchInsertWithoutIndexing();
queue.add(DEFAULT_PROJECT_KEY, 123L);
AnalysisReportDto report = queue.bookNextAvailable();
sut.execute(session, report, project);
// Check that the issue has well be indexed in E/S
assertThat(tester.get(IssueIndex.class).getNullableByKey(issue.getKey())).isNotNull();
}
@Test
public void index_a_lot_of_issues() throws Exception {
// ARRANGE
ComponentDto project = insertPermissionsForProject(DEFAULT_PROJECT_KEY);
db.issueAuthorizationDao().synchronizeAfter(session, new Date(0));
ComponentDto file = ComponentTesting.newFileDto(project);
db.componentDao().insert(session, file);
db.snapshotDao().insert(session, SnapshotTesting.createForComponent(file, project));
RuleDto rule = RuleTesting.newXooX1();
tester.get(RuleDao.class).insert(session, rule);
List<String> issueKeys = newArrayList();
for (int i = 0; i < 2001; i++) {
IssueDto issue = IssueTesting.newDto(rule, file, project);
tester.get(IssueDao.class).insert(session, issue);
issueKeys.add(issue.getKey());
}
session.commit();
session.clearCache();
clearIssueIndexToSimulateBatchInsertWithoutIndexing();
queue.add(DEFAULT_PROJECT_KEY, 123L);
List<AnalysisReportDto> reports = queue.findByProjectKey(DEFAULT_PROJECT_KEY);
AnalysisReportDto reportDto = reports.get(0);
// ACT
sut.execute(session, reportDto, project);
session.commit();
session.clearCache();
// ASSERT
Result<Issue> issueIndex = tester.get(IssueIndex.class).search(IssueQuery.builder().build(), new QueryContext());
assertThat(issueIndex.getTotal()).isEqualTo(2001);
}
private ComponentDto insertPermissionsForProject(String projectKey) {
ComponentDto project = ComponentTesting.newProjectDto().setKey(projectKey);
db.componentDao().insert(session, project);
tester.get(PermissionFacade.class).insertGroupPermission(project.getId(), DefaultGroups.ANYONE, UserRole.USER, session);
userSession.addProjectPermissions(UserRole.USER, project.key());
session.commit();
return project;
}
private void clearIssueIndexToSimulateBatchInsertWithoutIndexing() {
tester.get(BackendCleanup.class).clearIndexType(IndexDefinition.ISSUES);
}
}