Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Updating merge api, #3053

Merged
10 commits merged into from
Sep 22, 2021
Merged

Updating merge api, #3053

Changes from 2 commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
3b33b7a
Updating merge api,
fhnaseer Sep 13, 2021
b7d40f5
Merge remote-tracking branch 'origin/main' into dev/fhnaseer/MergeRep…
fhnaseer Sep 13, 2021
f1f1742
Storing reflection data for multiple usage,
fhnaseer Sep 14, 2021
1f9a3a3
Merge remote-tracking branch 'origin/main' into dev/fhnaseer/MergeRep…
fhnaseer Sep 16, 2021
8986e82
Merge remote-tracking branch 'origin/main' into dev/fhnaseer/MergeRep…
fhnaseer Sep 17, 2021
c959b6e
Adding tests for merging,
fhnaseer Sep 17, 2021
40b7d0d
fixing tabs and spaces,
fhnaseer Sep 17, 2021
db810fa
Adding integration tests,
fhnaseer Sep 17, 2021
5600202
Fixing tests for framework,
fhnaseer Sep 22, 2021
2713db5
fixing test,
fhnaseer Sep 22, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
88 changes: 47 additions & 41 deletions src/Microsoft.TestPlatform.Utilities/CodeCoverageDataAttachmentsHandler.cs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,12 +21,13 @@ public class CodeCoverageDataAttachmentsHandler : IDataCollectorAttachmentProces
{
private const string CoverageUri = "datacollector://microsoft/CodeCoverage/2.0";
private const string CoverageFileExtension = ".coverage";
private const string XmlFileExtension = ".xml";
private const string CoverageFriendlyName = "Code Coverage";

private const string CodeCoverageIOAssemblyName = "Microsoft.VisualStudio.Coverage.IO";
private const string CoverageFileUtilityTypeName = "CoverageFileUtility";
private const string MergeMethodName = "MergeCoverageFilesAsync";
private const string WriteMethodName = "WriteCoverageFile";
private const string MergeMethodName = "MergeCoverageReportsAsync";
private const string CoverageMergeOperationName = "CoverageMergeOperation";

private static readonly Uri CodeCoverageDataCollectorUri = new Uri(CoverageUri);

Expand All @@ -39,50 +40,50 @@ public IEnumerable<Uri> GetExtensionUris()

public async Task<ICollection<AttachmentSet>> ProcessAttachmentSetsAsync(ICollection<AttachmentSet> attachments, IProgress<int> progressReporter, IMessageLogger logger, CancellationToken cancellationToken)
{
if (attachments != null && attachments.Any())
{
var coverageReportFilePaths = new List<string>();
var coverageOtherFilePaths = new List<string>();
if ((attachments?.Any()) != true)
return new Collection<AttachmentSet>();

var coverageReportFilePaths = new List<string>();
var coverageOtherFilePaths = new List<string>();

foreach (var attachmentSet in attachments)
foreach (var attachmentSet in attachments)
{
foreach (var attachment in attachmentSet.Attachments)
{
foreach (var attachment in attachmentSet.Attachments)
if (attachment.Uri.LocalPath.EndsWith(CoverageFileExtension, StringComparison.OrdinalIgnoreCase) ||
attachment.Uri.LocalPath.EndsWith(XmlFileExtension, StringComparison.OrdinalIgnoreCase))
{
coverageReportFilePaths.Add(attachment.Uri.LocalPath);
}
else
{
if (attachment.Uri.LocalPath.EndsWith(CoverageFileExtension, StringComparison.OrdinalIgnoreCase))
{
coverageReportFilePaths.Add(attachment.Uri.LocalPath);
}
else
{
coverageOtherFilePaths.Add(attachment.Uri.LocalPath);
}
coverageOtherFilePaths.Add(attachment.Uri.LocalPath);
}
}
}

if (coverageReportFilePaths.Count > 1)
{
var mergedCoverageReportFilePath = await this.MergeCodeCoverageFilesAsync(coverageReportFilePaths, progressReporter, cancellationToken).ConfigureAwait(false);
if (!string.IsNullOrEmpty(mergedCoverageReportFilePath))
{
var resultAttachmentSet = new AttachmentSet(CodeCoverageDataCollectorUri, CoverageFriendlyName);
resultAttachmentSet.Attachments.Add(UriDataAttachment.CreateFrom(mergedCoverageReportFilePath, CoverageFriendlyName));
if (coverageReportFilePaths.Count > 1)
{
var mergedCoverageReports = await this.MergeCodeCoverageFilesAsync(coverageReportFilePaths, progressReporter, cancellationToken).ConfigureAwait(false);
var resultAttachmentSet = new AttachmentSet(CodeCoverageDataCollectorUri, CoverageFriendlyName);

foreach (var coverageOtherFilePath in coverageOtherFilePaths)
{
resultAttachmentSet.Attachments.Add(UriDataAttachment.CreateFrom(coverageOtherFilePath, string.Empty));
}
foreach (var coverageReport in mergedCoverageReports)
{
resultAttachmentSet.Attachments.Add(UriDataAttachment.CreateFrom(coverageReport, CoverageFriendlyName));
}

return new Collection<AttachmentSet> { resultAttachmentSet };
}
foreach (var coverageOtherFilePath in coverageOtherFilePaths)
{
resultAttachmentSet.Attachments.Add(UriDataAttachment.CreateFrom(coverageOtherFilePath, string.Empty));
}

return attachments;
return new Collection<AttachmentSet> { resultAttachmentSet };
}

return new Collection<AttachmentSet>();
return attachments;
}

private async Task<string> MergeCodeCoverageFilesAsync(IList<string> files, IProgress<int> progressReporter, CancellationToken cancellationToken)
private async Task<IList<string>> MergeCodeCoverageFilesAsync(IList<string> files, IProgress<int> progressReporter, CancellationToken cancellationToken)
{
try
{
Expand Down Expand Up @@ -110,28 +111,33 @@ private async Task<string> MergeCodeCoverageFilesAsync(IList<string> files, IPro
return null;
}

private async Task<string> MergeCodeCoverageFilesAsync(IList<string> files, CancellationToken cancellationToken)
private async Task<IList<string>> MergeCodeCoverageFilesAsync(IList<string> files, CancellationToken cancellationToken)
{
cancellationToken.ThrowIfCancellationRequested();

var assemblyPath = Path.Combine(Path.GetDirectoryName(typeof(CodeCoverageDataAttachmentsHandler).GetTypeInfo().Assembly.GetAssemblyLocation()), CodeCoverageIOAssemblyName + ".dll");

// Get assembly, type and methods
Assembly assembly = new PlatformAssemblyLoadContext().LoadAssemblyFromPath(assemblyPath);
var classType = assembly.GetType($"{CodeCoverageIOAssemblyName}.{CoverageFileUtilityTypeName}");
var classInstance = Activator.CreateInstance(classType);
var mergeMethodInfo = classType?.GetMethod(MergeMethodName, new[] { typeof(IList<string>), typeof(CancellationToken) });
var writeMethodInfo = classType?.GetMethod(WriteMethodName);

var types = assembly.GetTypes();
var mergeOperationEnum = Array.Find(types, d => d.Name == CoverageMergeOperationName);
var mergeOperationValues = Enum.GetValues(mergeOperationEnum);
var mergeMethodInfo = classType?.GetMethod(MergeMethodName, new[] { typeof(string), typeof(IList<string>), mergeOperationEnum, typeof(bool), typeof(CancellationToken) });

// Invoke methods
var task = (Task)mergeMethodInfo.Invoke(classInstance, new object[] { files, cancellationToken });
var task = (Task)mergeMethodInfo.Invoke(classInstance, new object[] { files[0], files, mergeOperationValues.GetValue(0), true, cancellationToken });
await task.ConfigureAwait(false);
var coverageData = task.GetType().GetProperty("Result").GetValue(task, null);
writeMethodInfo.Invoke(classInstance, new object[] { files[0], coverageData });
var mergedResults = coverageData as IList<string>;

// Delete original files and keep merged file only
foreach (var file in files.Skip(1))
foreach (var file in files)
{
if (mergedResults.Contains(file))
continue;

try
{
File.Delete(file);
Expand All @@ -142,7 +148,7 @@ private async Task<string> MergeCodeCoverageFilesAsync(IList<string> files, Canc
}
}

return files[0];
return mergedResults;
}
}
}