123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455 |
- /*
- * SonarQube
- * Copyright (C) 2009-2021 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.sonar.ce.task.projectanalysis.source;
-
- import java.util.Arrays;
- import java.util.Collections;
- import java.util.List;
- import java.util.function.Consumer;
- import org.apache.commons.codec.digest.DigestUtils;
- import org.junit.Before;
- import org.junit.Rule;
- import org.junit.Test;
- import org.mockito.Mockito;
- import org.sonar.api.utils.System2;
- import org.sonar.ce.task.projectanalysis.component.Component;
- import org.sonar.ce.task.projectanalysis.component.FileAttributes;
- import org.sonar.ce.task.projectanalysis.component.ReportComponent;
- import org.sonar.ce.task.projectanalysis.component.TreeRootHolderRule;
- import org.sonar.ce.task.projectanalysis.scm.Changeset;
- import org.sonar.ce.task.projectanalysis.step.BaseStepTest;
- import org.sonar.ce.task.step.ComputationStep;
- import org.sonar.ce.task.step.TestComputationStepContext;
- import org.sonar.core.util.SequenceUuidFactory;
- import org.sonar.core.util.Uuids;
- import org.sonar.db.DbClient;
- import org.sonar.db.DbSession;
- import org.sonar.db.DbTester;
- import org.sonar.db.protobuf.DbFileSources;
- import org.sonar.db.source.FileSourceDto;
- import org.sonar.db.source.LineHashVersion;
-
- import static org.assertj.core.api.Assertions.assertThat;
- import static org.mockito.Mockito.mock;
- import static org.mockito.Mockito.verify;
- import static org.mockito.Mockito.when;
-
- public class PersistFileSourcesStepTest extends BaseStepTest {
-
- private static final int FILE1_REF = 3;
- private static final String PROJECT_UUID = "PROJECT";
- private static final String PROJECT_KEY = "PROJECT_KEY";
- private static final String FILE1_UUID = "FILE1";
- private static final long NOW = 123456789L;
- private static final long PAST = 15000L;
-
- private System2 system2 = mock(System2.class);
-
- @Rule
- public DbTester dbTester = DbTester.create(system2);
- @Rule
- public TreeRootHolderRule treeRootHolder = new TreeRootHolderRule();
-
- private SourceLinesHashRepository sourceLinesHashRepository = mock(SourceLinesHashRepository.class);
- private SourceLinesHashRepositoryImpl.LineHashesComputer lineHashesComputer = mock(SourceLinesHashRepositoryImpl.LineHashesComputer.class);
- private FileSourceDataComputer fileSourceDataComputer = mock(FileSourceDataComputer.class);
- private FileSourceDataWarnings fileSourceDataWarnings = mock(FileSourceDataWarnings.class);
-
- private DbClient dbClient = dbTester.getDbClient();
- private DbSession session = dbTester.getSession();
-
- private PersistFileSourcesStep underTest;
-
- @Before
- public void setup() {
- when(system2.now()).thenReturn(NOW);
- when(sourceLinesHashRepository.getLineHashesComputerToPersist(Mockito.any(Component.class))).thenReturn(lineHashesComputer);
- underTest = new PersistFileSourcesStep(dbClient, system2, treeRootHolder, sourceLinesHashRepository, fileSourceDataComputer, fileSourceDataWarnings, new SequenceUuidFactory());
- initBasicReport(1);
- }
-
- @Override
- protected ComputationStep step() {
- return underTest;
- }
-
- @Test
- public void persist_sources() {
- List<String> lineHashes = Arrays.asList("137f72c3708c6bd0de00a0e5a69c699b", "e6251bcf1a7dc3ba5e7933e325bbe605");
- String sourceHash = "ee5a58024a155466b43bc559d953e018";
- DbFileSources.Data fileSourceData = DbFileSources.Data.newBuilder()
- .addAllLines(Arrays.asList(
- DbFileSources.Line.newBuilder().setSource("line1").setLine(1).build(),
- DbFileSources.Line.newBuilder().setSource("line2").setLine(2).build()))
- .build();
- when(fileSourceDataComputer.compute(fileComponent().build(), fileSourceDataWarnings))
- .thenReturn(new FileSourceDataComputer.Data(fileSourceData, lineHashes, sourceHash, null));
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getProjectUuid()).isEqualTo(PROJECT_UUID);
- assertThat(fileSourceDto.getFileUuid()).isEqualTo(FILE1_UUID);
- assertThat(fileSourceDto.getBinaryData()).isNotEmpty();
- assertThat(fileSourceDto.getDataHash()).isNotEmpty();
- assertThat(fileSourceDto.getLineHashesVersion()).isEqualTo(LineHashVersion.WITHOUT_SIGNIFICANT_CODE.getDbValue());
- assertThat(fileSourceDto.getLineHashes()).isNotEmpty();
- assertThat(fileSourceDto.getCreatedAt()).isEqualTo(NOW);
- assertThat(fileSourceDto.getUpdatedAt()).isEqualTo(NOW);
-
- DbFileSources.Data data = fileSourceDto.getSourceData();
- assertThat(data.getLinesCount()).isEqualTo(2);
- assertThat(data.getLines(0).getLine()).isOne();
- assertThat(data.getLines(0).getSource()).isEqualTo("line1");
- assertThat(data.getLines(1).getLine()).isEqualTo(2);
- assertThat(data.getLines(1).getSource()).isEqualTo("line2");
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void persist_source_hashes() {
- List<String> lineHashes = Arrays.asList("137f72c3708c6bd0de00a0e5a69c699b", "e6251bcf1a7dc3ba5e7933e325bbe605");
- String sourceHash = "ee5a58024a155466b43bc559d953e018";
- setComputedData(DbFileSources.Data.newBuilder().build(), lineHashes, sourceHash, null);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getLineHashes()).containsExactly("137f72c3708c6bd0de00a0e5a69c699b", "e6251bcf1a7dc3ba5e7933e325bbe605");
- assertThat(fileSourceDto.getSrcHash()).isEqualTo("ee5a58024a155466b43bc559d953e018");
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void persist_coverage() {
- DbFileSources.Data dbData = DbFileSources.Data.newBuilder().addLines(
- DbFileSources.Line.newBuilder()
- .setConditions(10)
- .setCoveredConditions(2)
- .setLineHits(1)
- .setLine(1)
- .build())
- .build();
- setComputedData(dbData);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getSourceData()).isEqualTo(dbData);
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- private ReportComponent.Builder fileComponent() {
- return ReportComponent.builder(Component.Type.FILE, FILE1_REF).setUuid(FILE1_UUID).setKey("PROJECT_KEY" + ":src/Foo.java");
- }
-
- @Test
- public void persist_scm() {
- DbFileSources.Data dbData = DbFileSources.Data.newBuilder().addLines(
- DbFileSources.Line.newBuilder()
- .setScmAuthor("john")
- .setScmDate(123456789L)
- .setScmRevision("rev-1")
- .build())
- .build();
- setComputedData(dbData);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getSourceData()).isEqualTo(dbData);
- assertThat(fileSourceDto.getRevision()).isNull();
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void persist_scm_some_lines() {
- DbFileSources.Data dbData = DbFileSources.Data.newBuilder().addAllLines(Arrays.asList(
- DbFileSources.Line.newBuilder()
- .setScmAuthor("john")
- .setScmDate(123456789L)
- .setScmRevision("rev-1")
- .build(),
- DbFileSources.Line.newBuilder()
- .setScmDate(223456789L)
- .build(),
- DbFileSources.Line.newBuilder()
- .build()))
- .build();
- setComputedData(dbData);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
-
- DbFileSources.Data data = fileSourceDto.getSourceData();
-
- assertThat(data.getLinesList()).hasSize(3);
-
- assertThat(data.getLines(0).getScmAuthor()).isEqualTo("john");
- assertThat(data.getLines(0).getScmDate()).isEqualTo(123456789L);
- assertThat(data.getLines(0).getScmRevision()).isEqualTo("rev-1");
-
- assertThat(data.getLines(1).getScmAuthor()).isEmpty();
- assertThat(data.getLines(1).getScmDate()).isEqualTo(223456789L);
- assertThat(data.getLines(1).getScmRevision()).isEmpty();
-
- assertThat(data.getLines(2).getScmAuthor()).isEmpty();
- assertThat(data.getLines(2).getScmDate()).isZero();
- assertThat(data.getLines(2).getScmRevision()).isEmpty();
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void persist_highlighting() {
- DbFileSources.Data dbData = DbFileSources.Data.newBuilder().addLines(
- DbFileSources.Line.newBuilder()
- .setHighlighting("2,4,a")
- .build())
- .build();
- setComputedData(dbData);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- DbFileSources.Data data = fileSourceDto.getSourceData();
- assertThat(data).isEqualTo(dbData);
- assertThat(data.getLinesList()).hasSize(1);
- assertThat(data.getLines(0).getHighlighting()).isEqualTo("2,4,a");
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void persist_symbols() {
- DbFileSources.Data dbData = DbFileSources.Data.newBuilder().addAllLines(Arrays.asList(
- DbFileSources.Line.newBuilder()
- .setSymbols("2,4,1")
- .build(),
- DbFileSources.Line.newBuilder().build(),
- DbFileSources.Line.newBuilder()
- .setSymbols("1,3,1")
- .build()))
- .build();
- setComputedData(dbData);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getSourceData()).isEqualTo(dbData);
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void persist_duplication() {
- DbFileSources.Data dbData = DbFileSources.Data.newBuilder().addLines(
- DbFileSources.Line.newBuilder()
- .addDuplication(2)
- .build())
- .build();
- setComputedData(dbData);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getSourceData()).isEqualTo(dbData);
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void save_revision() {
- Changeset latest = Changeset.newChangesetBuilder().setDate(0L).setRevision("rev-1").build();
- setComputedData(DbFileSources.Data.newBuilder().build(), Collections.singletonList("lineHashes"), "srcHash", latest);
-
- underTest.execute(new TestComputationStepContext());
-
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getRevision()).isEqualTo("rev-1");
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void not_save_revision() {
- setComputedData(DbFileSources.Data.newBuilder().build());
-
- underTest.execute(new TestComputationStepContext());
-
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getRevision()).isNull();
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void not_update_sources_when_nothing_has_changed() {
- dbClient.fileSourceDao().insert(dbTester.getSession(), createDto());
- dbTester.getSession().commit();
-
- Changeset changeset = Changeset.newChangesetBuilder().setDate(1L).setRevision("rev-1").build();
- setComputedData(DbFileSources.Data.newBuilder().build(), Collections.singletonList("lineHash"), "sourceHash", changeset);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getSrcHash()).isEqualTo("sourceHash");
- assertThat(fileSourceDto.getLineHashes()).isEqualTo(Collections.singletonList("lineHash"));
- assertThat(fileSourceDto.getCreatedAt()).isEqualTo(PAST);
- assertThat(fileSourceDto.getUpdatedAt()).isEqualTo(PAST);
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void update_sources_when_source_updated() {
- // Existing sources
- long past = 150000L;
- dbClient.fileSourceDao().insert(dbTester.getSession(), new FileSourceDto()
- .setUuid(Uuids.createFast())
- .setProjectUuid(PROJECT_UUID)
- .setFileUuid(FILE1_UUID)
- .setSrcHash("5b4bd9815cdb17b8ceae19eb1810c34c")
- .setLineHashes(Collections.singletonList("6438c669e0d0de98e6929c2cc0fac474"))
- .setDataHash("6cad150e3d065976c230cddc5a09efaa")
- .setSourceData(DbFileSources.Data.newBuilder()
- .addLines(DbFileSources.Line.newBuilder()
- .setLine(1)
- .setSource("old line")
- .build())
- .build())
- .setCreatedAt(past)
- .setUpdatedAt(past)
- .setRevision("rev-0"));
- dbTester.getSession().commit();
-
- DbFileSources.Data newSourceData = DbFileSources.Data.newBuilder()
- .addLines(DbFileSources.Line.newBuilder()
- .setLine(1)
- .setSource("old line")
- .setScmDate(123456789L)
- .setScmRevision("rev-1")
- .setScmAuthor("john")
- .build())
- .build();
-
- Changeset changeset = Changeset.newChangesetBuilder().setDate(1L).setRevision("rev-1").build();
- setComputedData(newSourceData, Collections.singletonList("6438c669e0d0de98e6929c2cc0fac474"), "5b4bd9815cdb17b8ceae19eb1810c34c", changeset);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getCreatedAt()).isEqualTo(past);
- assertThat(fileSourceDto.getUpdatedAt()).isEqualTo(NOW);
- assertThat(fileSourceDto.getRevision()).isEqualTo("rev-1");
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void update_sources_when_src_hash_is_missing() {
- dbClient.fileSourceDao().insert(dbTester.getSession(), createDto(dto -> dto.setSrcHash(null)));
- dbTester.getSession().commit();
-
- DbFileSources.Data sourceData = DbFileSources.Data.newBuilder().build();
- setComputedData(sourceData, Collections.singletonList("lineHash"), "newSourceHash", null);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getCreatedAt()).isEqualTo(PAST);
- assertThat(fileSourceDto.getUpdatedAt()).isEqualTo(NOW);
- assertThat(fileSourceDto.getSrcHash()).isEqualTo("newSourceHash");
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- @Test
- public void update_sources_when_revision_is_missing() {
- DbFileSources.Data sourceData = DbFileSources.Data.newBuilder()
- .addLines(DbFileSources.Line.newBuilder()
- .setLine(1)
- .setSource("line")
- .build())
- .build();
-
- dbClient.fileSourceDao().insert(dbTester.getSession(), createDto(dto -> dto.setRevision(null)));
- dbTester.getSession().commit();
-
- Changeset changeset = Changeset.newChangesetBuilder().setDate(1L).setRevision("revision").build();
- setComputedData(sourceData, Collections.singletonList("137f72c3708c6bd0de00a0e5a69c699b"), "29f25900140c94db38035128cb6de6a2", changeset);
-
- underTest.execute(new TestComputationStepContext());
-
- assertThat(dbTester.countRowsOfTable("file_sources")).isOne();
- FileSourceDto fileSourceDto = dbClient.fileSourceDao().selectByFileUuid(session, FILE1_UUID);
- assertThat(fileSourceDto.getCreatedAt()).isEqualTo(PAST);
- assertThat(fileSourceDto.getUpdatedAt()).isEqualTo(NOW);
- assertThat(fileSourceDto.getRevision()).isEqualTo("revision");
- verify(fileSourceDataWarnings).commitWarnings();
- }
-
- private FileSourceDto createDto() {
- return createDto(dto -> {
- });
- }
-
- private FileSourceDto createDto(Consumer<FileSourceDto> modifier) {
- DbFileSources.Data sourceData = DbFileSources.Data.newBuilder().build();
- byte[] data = FileSourceDto.encodeSourceData(sourceData);
- String dataHash = DigestUtils.md5Hex(data);
-
- FileSourceDto dto = new FileSourceDto()
- .setUuid(Uuids.createFast())
- .setProjectUuid(PROJECT_UUID)
- .setFileUuid(FILE1_UUID)
- .setSrcHash("sourceHash")
- .setLineHashes(Collections.singletonList("lineHash"))
- .setDataHash(dataHash)
- .setRevision("rev-1")
- .setSourceData(sourceData)
- .setCreatedAt(PAST)
- .setUpdatedAt(PAST);
-
- modifier.accept(dto);
- return dto;
- }
-
- private void setComputedData(DbFileSources.Data data, List<String> lineHashes, String sourceHash, Changeset latestChangeWithRevision) {
- FileSourceDataComputer.Data computedData = new FileSourceDataComputer.Data(data, lineHashes, sourceHash, latestChangeWithRevision);
- when(fileSourceDataComputer.compute(fileComponent().build(), fileSourceDataWarnings)).thenReturn(computedData);
- }
-
- private void setComputedData(DbFileSources.Data data) {
- FileSourceDataComputer.Data computedData = new FileSourceDataComputer.Data(data, Collections.emptyList(), "", null);
- when(fileSourceDataComputer.compute(fileComponent().build(), fileSourceDataWarnings)).thenReturn(computedData);
- }
-
- private void initBasicReport(int numberOfLines) {
- ReportComponent root = ReportComponent.builder(Component.Type.PROJECT, 1).setUuid(PROJECT_UUID).setKey(PROJECT_KEY).addChildren(
- fileComponent().setFileAttributes(new FileAttributes(false, null, numberOfLines)).build())
- .build();
- treeRootHolder.setRoots(root, root);
- }
- }
|