dotnet-installer/test/Microsoft.DotNet.ProjectJsonMigration.Tests/ProjectJsonBuilder.cs
Bryan Thornbury b0554d3ff3 PR Feedback
2016-08-30 12:00:56 -07:00

110 lines
2.9 KiB
C#

using Microsoft.DotNet.TestFramework;
using Newtonsoft.Json.Linq;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading.Tasks;
namespace Microsoft.DotNet.ProjectJsonMigration.Tests
{
/// <summary>
/// Used to build up test scenario project.jsons without needing to add a new test asset.
/// </summary>
public class ProjectJsonBuilder
{
private readonly TestAssetsManager _testAssetsManager;
private JObject _projectJson;
private bool _baseDefined = false;
private bool _baseProjectDirectory;
public ProjectJsonBuilder(TestAssetsManager testAssetsManager)
{
_testAssetsManager = testAssetsManager;
}
public string SaveToDisk(string outputDirectory)
{
EnsureBaseIsSet();
var projectPath = Path.Combine(outputDirectory, "project.json");
File.WriteAllText(projectPath, _projectJson.ToString());
return projectPath;
}
public JObject Build()
{
EnsureBaseIsSet();
return _projectJson;
}
public ProjectJsonBuilder FromTestAssetBase(string testAssetName)
{
var testProjectDirectory = _testAssetsManager.CreateTestInstance(testAssetName).Path;
var testProject = Path.Combine(testProjectDirectory, "project.json");
SetBase(JObject.Parse(File.ReadAllText(testProject)));
return this;
}
public ProjectJsonBuilder FromStringBase(string jsonString)
{
SetBase(JObject.Parse(jsonString));
return this;
}
public ProjectJsonBuilder FromEmptyBase()
{
SetBase(new JObject());
return this;
}
public ProjectJsonBuilder WithCustomProperty(string propertyName, Dictionary<string, string> value)
{
EnsureBaseIsSet();
_projectJson[propertyName] = JObject.FromObject(value);
return this;
}
public ProjectJsonBuilder WithCustomProperty(string propertyName, string value)
{
EnsureBaseIsSet();
_projectJson[propertyName] = value;
return this;
}
public ProjectJsonBuilder WithCustomProperty(string propertyName, string[] value)
{
EnsureBaseIsSet();
_projectJson[propertyName] = JArray.FromObject(value);
return this;
}
private void SetBase(JObject project)
{
if (_baseDefined)
{
throw new Exception("Base was already defined.");
}
_baseDefined = true;
_projectJson = project;
}
private void EnsureBaseIsSet()
{
if (!_baseDefined)
{
throw new Exception("Cannot build without base set");
}
}
}
}