124 lines
3.4 KiB
C#
124 lines
3.4 KiB
C#
// Copyright (c) .NET Foundation and contributors. All rights reserved.
|
|
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using System.Runtime.CompilerServices;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Microsoft.DotNet.Cli.Utils;
|
|
using Microsoft.DotNet.Tools.Common;
|
|
|
|
namespace Microsoft.DotNet.TestFramework
|
|
{
|
|
public class TestAssetInstance
|
|
{
|
|
public TestAssetInstance(TestAssetInfo testAssetInfo, DirectoryInfo root)
|
|
{
|
|
if (testAssetInfo == null)
|
|
{
|
|
throw new ArgumentException(nameof(testAssetInfo));
|
|
}
|
|
|
|
if (root == null)
|
|
{
|
|
throw new ArgumentException(nameof(root));
|
|
}
|
|
|
|
TestAssetInfo = testAssetInfo;
|
|
|
|
Root = root;
|
|
|
|
MigrationBackupRoot = new DirectoryInfo(Path.Combine(root.Parent.FullName, "backup"));
|
|
|
|
if (Root.Exists)
|
|
{
|
|
Root.Delete(recursive: true);
|
|
}
|
|
|
|
Root.Create();
|
|
|
|
if (MigrationBackupRoot.Exists)
|
|
{
|
|
MigrationBackupRoot.Delete(recursive: true);
|
|
}
|
|
}
|
|
|
|
public DirectoryInfo MigrationBackupRoot { get; }
|
|
|
|
public DirectoryInfo Root { get; }
|
|
|
|
public TestAssetInfo TestAssetInfo { get; }
|
|
|
|
|
|
public TestAssetInstance WithSourceFiles()
|
|
{
|
|
var filesToCopy = TestAssetInfo.GetSourceFiles();
|
|
|
|
CopyFiles(filesToCopy);
|
|
|
|
return this;
|
|
}
|
|
|
|
public TestAssetInstance WithRestoreFiles()
|
|
{
|
|
var filesToCopy = TestAssetInfo.GetRestoreFiles();
|
|
|
|
CopyFiles(filesToCopy);
|
|
|
|
return this;
|
|
}
|
|
|
|
public TestAssetInstance WithBuildFiles()
|
|
{
|
|
var filesToCopy = TestAssetInfo.GetBuildFiles();
|
|
|
|
CopyFiles(filesToCopy);
|
|
|
|
return this;
|
|
}
|
|
|
|
public TestAssetInstance WithNuGetConfig(string nugetCache)
|
|
{
|
|
var thisAssembly = typeof(TestAssetInstance).GetTypeInfo().Assembly;
|
|
var newNuGetConfig = Root.GetFile("Nuget.config");
|
|
|
|
var content = @"<?xml version=""1.0"" encoding=""utf-8""?>
|
|
<configuration>
|
|
<packageSources>
|
|
<add key=""test-packages"" value=""$fullpath$"" />
|
|
</packageSources>
|
|
</configuration>";
|
|
content = content.Replace("$fullpath$", nugetCache);
|
|
|
|
using (var newNuGetConfigStream =
|
|
new FileStream(newNuGetConfig.FullName, FileMode.Create, FileAccess.Write))
|
|
{
|
|
var contentBytes = new UTF8Encoding(true).GetBytes(content);
|
|
newNuGetConfigStream.Write(contentBytes, 0, contentBytes.Length);
|
|
}
|
|
|
|
return this;
|
|
}
|
|
|
|
private void CopyFiles(IEnumerable<FileInfo> filesToCopy)
|
|
{
|
|
foreach (var file in filesToCopy)
|
|
{
|
|
var relativePath = file.FullName.Substring(TestAssetInfo.Root.FullName.Length + 1);
|
|
|
|
var newPath = Path.Combine(Root.FullName, relativePath);
|
|
|
|
var newFile = new FileInfo(newPath);
|
|
|
|
PathUtility.EnsureDirectoryExists(newFile.Directory.FullName);
|
|
|
|
file.CopyTo(newPath);
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|