mirror of https://github.com/pulumi/pulumi.git
87 lines
2.9 KiB
C#
87 lines
2.9 KiB
C#
// Copyright 2016-2021, Pulumi Corporation
|
|
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Collections.Immutable;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
|
|
using Pulumi;
|
|
using Pulumi.Testing;
|
|
using static Pulumi.Utilities.OutputUtilities;
|
|
|
|
namespace Pulumi.Mypkg
|
|
{
|
|
public static class TestHelpers
|
|
{
|
|
public static async Task<(T Result, IEnumerable<string> Deps)> Run<T>(
|
|
IMocks mocks, Func<Output<T>> outputGenerator, TestOptions? options = null)
|
|
{
|
|
options = options ?? new TestOptions();
|
|
options.ProjectName = HelperStack.RegisterBuilderAsProjectName(outputGenerator);
|
|
var resources = await Deployment.TestAsync<HelperStack>(mocks, options);
|
|
var stack = resources.Where(x => x is HelperStack).First() as HelperStack;
|
|
if (stack != null)
|
|
{
|
|
var result = await GetValueAsync(stack.Result);
|
|
if (result is T)
|
|
{
|
|
var deps = await GetDependenciesAsync(stack.Result);
|
|
var urns = new List<string>();
|
|
foreach (var dep in deps)
|
|
{
|
|
var urn = await GetValueAsync(dep.Urn);
|
|
urns.Add(urn);
|
|
}
|
|
return (Result: (T)result, Deps: urns);
|
|
}
|
|
else
|
|
{
|
|
throw new Exception($"The output did not resolve to the correct type: {result}");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
throw new Exception("Did not find stack");
|
|
}
|
|
}
|
|
|
|
public static Output<T> Out<T>(T x)
|
|
{
|
|
return Output.Create<T>(x);
|
|
}
|
|
|
|
|
|
public class HelperStack : Stack
|
|
{
|
|
private static ConcurrentDictionary<string,Func<Output<object?>>> registry =
|
|
new ConcurrentDictionary<string,Func<Output<object?>>>();
|
|
|
|
[Output]
|
|
public Output<object?> Result { get; private set; }
|
|
|
|
public HelperStack()
|
|
{
|
|
Console.WriteLine(Deployment.Instance.ProjectName);
|
|
Func<Output<object?>>? outputBuilder;
|
|
if (!registry.TryGetValue(Deployment.Instance.ProjectName, out outputBuilder))
|
|
{
|
|
throw new Exception("Incorrect use of HelperStack");
|
|
}
|
|
this.Result = outputBuilder.Invoke();
|
|
}
|
|
|
|
public static string RegisterBuilderAsProjectName<T>(Func<Output<T>> builder)
|
|
{
|
|
var projectName = Guid.NewGuid().ToString();
|
|
if (!registry.TryAdd(projectName, () => builder().Apply(x => (object?)x)))
|
|
{
|
|
throw new Exception("Impossible");
|
|
}
|
|
return projectName;
|
|
}
|
|
}
|
|
}
|
|
}
|