TestPlatformNugetPackageTests.cs

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.

namespace Microsoft.TestPlatform.AcceptanceTests
{
    using System;
    using System.IO;
    using System.IO.Compression;
    using System.Linq;

    using Microsoft.TestPlatform.TestUtilities;
    using Microsoft.VisualStudio.TestTools.UnitTesting;

    [TestClass]
    public class TestPlatformNugetPackageTests : CodeCoverageAcceptanceTestBase
    {
        private static string nugetPackageFolder;
        private string resultsDirectory;

        [ClassInitialize]
        public static void ClassInit(TestContext testContext)
        {
            var packageLocation = Path.Combine(IntegrationTestEnvironment.TestPlatformRootDirectory, "artifacts", IntegrationTestEnvironment.BuildConfiguration, "packages");
            var nugetPackage = Directory.EnumerateFiles(packageLocation, "Microsoft.TestPlatform.*.nupkg").OrderBy(a => a).FirstOrDefault();
            nugetPackageFolder = Path.Combine(packageLocation, Path.GetFileNameWithoutExtension(nugetPackage));
            ZipFile.ExtractToDirectory(nugetPackage, nugetPackageFolder);

            TryMoveDirectory(
                sourceDirName: Path.Combine(nugetPackageFolder, "tools", "net451", "Team%20Tools"),
                destDirName: Path.Combine(nugetPackageFolder, "tools", "net451", "Team Tools")
            );

            TryMoveDirectory(
                sourceDirName: Path.Combine(nugetPackageFolder, "tools", "net451", "Team Tools", "Dynamic%20Code%20Coverage%20Tools"),
                destDirName: Path.Combine(nugetPackageFolder, "tools", "net451", "Team Tools", "Dynamic Code Coverage Tools")
            );
        }

        [ClassCleanup]
        public static void ClassCleanup()
        {
            Directory.Delete(nugetPackageFolder, true);
        }

        [TestInitialize]
        public void SetUp()
        {
            this.resultsDirectory = GetResultsDirectory();
        }

        [TestCleanup]
        public void CleanUp()
        {
            TryRemoveDirectory(resultsDirectory);
        }

        [TestMethod]
        [TestCategory("Windows-Review")]
        [NetFullTargetFrameworkDataSource(useCoreRunner: false)]
        [NetCoreTargetFrameworkDataSource(useCoreRunner: false)]
        public void RunMultipleTestAssembliesWithCodeCoverage(RunnerInfo runnerInfo)
        {
            AcceptanceTestBase.SetTestEnvironment(this.testEnvironment, runnerInfo);

            var assemblyPaths = this.BuildMultipleAssemblyPath("SimpleTestProject.dll", "SimpleTestProject2.dll").Trim('\"');

            var arguments = CreateCodeCoverageArguments(runnerInfo, assemblyPaths, out var trxFilePath);
            this.InvokeVsTest(arguments);

            this.ExitCodeEquals(1); // failing tests

            var actualCoverageFile = CodeCoverageTests.GetCoverageFileNameFromTrx(trxFilePath, resultsDirectory);
            Console.WriteLine($@"Coverage file: {actualCoverageFile}  Results directory: {resultsDirectory} trxfile: {trxFilePath}");
            Assert.IsTrue(File.Exists(actualCoverageFile), "Coverage file not found: {0}", actualCoverageFile);
        }

        public override string GetConsoleRunnerPath()
        {
            string consoleRunnerPath = string.Empty;

            if (this.IsDesktopRunner())
            {
                consoleRunnerPath = Path.Combine(nugetPackageFolder, "tools", "net451", "Common7", "IDE", "Extensions", "TestPlatform", "vstest.console.exe");
            }

            Assert.IsTrue(File.Exists(consoleRunnerPath), "GetConsoleRunnerPath: Path not found: {0}", consoleRunnerPath);
            return consoleRunnerPath;
        }

        private string CreateCodeCoverageArguments(
            RunnerInfo runnerInfo,
            string assemblyPaths,
            out string trxFilePath)
        {
            string diagFileName = Path.Combine(this.resultsDirectory, "diaglog.txt");

            var arguments = PrepareArguments(assemblyPaths, this.GetTestAdapterPath(), string.Empty,
                this.FrameworkArgValue, runnerInfo.InIsolationValue, resultsDirectory: resultsDirectory);

            arguments = string.Concat(arguments, $" /Diag:{diagFileName}", $" /EnableCodeCoverage");

            trxFilePath = Path.Combine(this.resultsDirectory, Guid.NewGuid() + ".trx");
            arguments = string.Concat(arguments, " /logger:trx;logfilename=" + trxFilePath);

            return arguments;
        }

        private static void TryMoveDirectory(string sourceDirName, string destDirName)
        {
            if (Directory.Exists(sourceDirName))
            {
                Directory.Move(sourceDirName, destDirName);
            }
        }
    }
}