add additional removal indicator

This commit is contained in:
Paul Dorsch 2024-09-13 09:45:43 -04:00
Родитель 1d1e13bfa7
Коммит 541b6468a2
7 изменённых файлов: 21 добавлений и 14 удалений

Просмотреть файл

@ -2,6 +2,7 @@ namespace Microsoft.ComponentDetection.Common;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
using Microsoft.ComponentDetection.Contracts;
@ -39,14 +40,14 @@ public class FileUtilityService : IFileUtilityService
}
/// <inheritdoc />
public (string DuplicateFilePath, bool CreatedDuplicate) DuplicateFileWithoutLines(string fileName, string removalIndicator)
public (string DuplicateFilePath, bool CreatedDuplicate) DuplicateFileWithoutLines(string fileName, params string[] removalIndicators)
{
// Read all lines from the file and filter out the lines that start with the removal indicator.
var removedAnyLines = false;
var linesToKeep = new List<string>();
foreach (var line in File.ReadLines(fileName))
{
if (line == null || line.Trim().StartsWith(removalIndicator, System.StringComparison.OrdinalIgnoreCase))
if (line == null || removalIndicators.Any(removalIndicator => line.Trim().StartsWith(removalIndicator, System.StringComparison.OrdinalIgnoreCase)))
{
removedAnyLines = true;
}

Просмотреть файл

@ -47,7 +47,7 @@ public interface IFileUtilityService
/// Duplicates a file, removing any lines that starts with the given string.
/// </summary>
/// <param name="fileName">Path to the file.</param>
/// <param name="removalIndicator">The string that indicates a line should be removed.</param>
/// <param name="removalIndicators">The strings that indicates a line should be removed.</param>
/// <returns>Returns the path of the new file, and whether or not one was created.</returns>
(string DuplicateFilePath, bool CreatedDuplicate) DuplicateFileWithoutLines(string fileName, string removalIndicator);
(string DuplicateFilePath, bool CreatedDuplicate) DuplicateFileWithoutLines(string fileName, params string[] removalIndicators);
}

Просмотреть файл

@ -149,7 +149,7 @@ public class PipCommandService : IPipCommandService
if (this.environmentService.IsEnvironmentVariableValueTrue(PipReportIgnoreFileLevelIndexUrlEnvVar))
{
// check for --index-url in requirements.txt and remove it from the file, since we want to use PIP_INDEX_URL from the environment.
var (duplicateFilePath, createdDuplicate) = this.fileUtilityService.DuplicateFileWithoutLines(formattedPath, "--index-url");
var (duplicateFilePath, createdDuplicate) = this.fileUtilityService.DuplicateFileWithoutLines(formattedPath, "--index-url", "-i");
if (createdDuplicate)
{
var duplicateFileName = Path.GetFileName(duplicateFilePath);
@ -169,7 +169,7 @@ public class PipCommandService : IPipCommandService
// When PIP_INDEX_URL is set, we need to pass it as a parameter to pip install command.
// This should be done before running detection by the build system, otherwise the detection
// will default to the public PyPI index if not configured in pip defaults. Note this index URL may have credentials, we need to remove it when logging.
pipReportCommand += $" --dry-run --ignore-installed --quiet --no-input --report {reportName}";
pipReportCommand += $" --dry-run --ignore-installed --quiet --no-cache-dir --no-input --report {reportName}";
if (this.environmentService.DoesEnvironmentVariableExist("PIP_INDEX_URL"))
{
pipReportCommand += $" --index-url {this.environmentService.GetEnvironmentVariable("PIP_INDEX_URL")}";

Просмотреть файл

@ -23,16 +23,21 @@ public class FileUtilityServiceTests
// Arrange
var fileName = $"{directory}/Resources/test-file-duplicate.txt";
var removalIndicator = "//REMOVE";
var expectedDuplicateFilePath = Path.Combine(directory, "Resources", "temp.test-file-duplicate.txt");
// Act
var (duplicateFilePath, createdDuplicate) = this.fileUtilityService.DuplicateFileWithoutLines(fileName, removalIndicator);
var (duplicateFilePath, createdDuplicate) = this.fileUtilityService.DuplicateFileWithoutLines(fileName, "//REMOVE", "//ME");
// Assert
createdDuplicate.Should().BeTrue();
duplicateFilePath.Should().Be(expectedDuplicateFilePath);
File.Exists(expectedDuplicateFilePath).Should().BeTrue();
var contents = File.ReadAllText(expectedDuplicateFilePath);
contents.Should().NotContain("//REMOVE");
contents.Should().NotContain("//ME");
contents.Should().Contain("hello");
contents.Should().Contain("world");
}
[TestMethod]

Просмотреть файл

@ -1,4 +1,4 @@
<Project Sdk="Microsoft.NET.Sdk">
<Project Sdk="Microsoft.NET.Sdk">
<ItemGroup Label="Project References">
<ProjectReference Include="..\Microsoft.ComponentDetection.TestsUtilities\Microsoft.ComponentDetection.TestsUtilities.csproj" />
@ -9,8 +9,8 @@
<PackageReference Include="FluentAssertions.Analyzers" PrivateAssets="all" />
<PackageReference Include="Microsoft.Extensions.Logging" />
<PackageReference Include="System.Threading.Tasks.Dataflow" />
<PackageReference Include="MSTest.TestAdapter "/>
<PackageReference Include="Microsoft.NET.Test.Sdk "/>
<PackageReference Include="MSTest.TestAdapter " />
<PackageReference Include="Microsoft.NET.Test.Sdk " />
</ItemGroup>
<ItemGroup>

Просмотреть файл

@ -1,3 +1,4 @@
hello
//REMOVE
world
//ME

Просмотреть файл

@ -500,7 +500,7 @@ public class PipCommandServiceTests
this.envVarService.Setup(x => x.IsEnvironmentVariableValueTrue("PipReportIgnoreFileLevelIndexUrl")).Returns(true);
this.commandLineInvokationService.Setup(x => x.CanCommandBeLocatedAsync("pip", It.IsAny<IEnumerable<string>>(), "--version")).ReturnsAsync(true);
this.fileUtilityService.Setup(x => x.DuplicateFileWithoutLines(It.IsAny<string>(), "--index-url"))
this.fileUtilityService.Setup(x => x.DuplicateFileWithoutLines(It.IsAny<string>(), "--index-url", "-i"))
.Returns(("C:/asdf/temp.requirements.txt", true))
.Verifiable();
this.fileUtilityService.Setup(x => x.ReadAllTextAsync(It.IsAny<FileInfo>()))
@ -535,7 +535,7 @@ public class PipCommandServiceTests
this.envVarService.Setup(x => x.IsEnvironmentVariableValueTrue("PipReportIgnoreFileLevelIndexUrl")).Returns(true);
this.commandLineInvokationService.Setup(x => x.CanCommandBeLocatedAsync("pip", It.IsAny<IEnumerable<string>>(), "--version")).ReturnsAsync(true);
this.fileUtilityService.Setup(x => x.DuplicateFileWithoutLines(It.IsAny<string>(), "--index-url"))
this.fileUtilityService.Setup(x => x.DuplicateFileWithoutLines(It.IsAny<string>(), "--index-url", "-i"))
.Returns((null, false))
.Verifiable();
this.fileUtilityService.Setup(x => x.ReadAllTextAsync(It.IsAny<FileInfo>()))
@ -590,7 +590,7 @@ public class PipCommandServiceTests
this.logger.Object);
var (report, reportFile) = await service.GenerateInstallationReportAsync(testPath);
this.fileUtilityService.Verify(x => x.DuplicateFileWithoutLines(It.IsAny<string>(), "--index-url"), Times.Never);
this.fileUtilityService.Verify(x => x.DuplicateFileWithoutLines(It.IsAny<string>(), "--index-url", "-i"), Times.Never);
this.commandLineInvokationService.Verify();
}