74 lines
2.1 KiB
C#
74 lines
2.1 KiB
C#
// Copyright 2016-2020, Pulumi Corporation
|
|
|
|
using System.Collections.Generic;
|
|
using System.Collections.Immutable;
|
|
using System.Linq;
|
|
using System.Threading.Tasks;
|
|
using Pulumi.Testing;
|
|
using Xunit;
|
|
|
|
namespace Pulumi.Tests.Mocks
|
|
{
|
|
class MyMocks : IMocks
|
|
{
|
|
public Task<object> CallAsync(string token, ImmutableDictionary<string, object> args, string? provider)
|
|
{
|
|
return Task.FromResult<object>(args);
|
|
}
|
|
|
|
public Task<(string id, object state)> NewResourceAsync(string type, string name, ImmutableDictionary<string, object> inputs, string? provider, string? id)
|
|
{
|
|
Assert.Equal("aws:ec2/instance:Instance", type);
|
|
return Task.FromResult<(string, object)>(("i-1234567890abcdef0", new Dictionary<string, object> {
|
|
{ "publicIp", "203.0.113.12" },
|
|
}));
|
|
}
|
|
}
|
|
|
|
public partial class MocksTests
|
|
{
|
|
[Fact]
|
|
public async Task TestCustom()
|
|
{
|
|
var resources = await Testing.RunAsync<MyStack>();
|
|
|
|
var instance = resources.OfType<Instance>().FirstOrDefault();
|
|
Assert.NotNull(instance);
|
|
|
|
var ip = await instance.PublicIp.GetValueAsync();
|
|
Assert.Equal("203.0.113.12", ip);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task TestStack()
|
|
{
|
|
var resources = await Testing.RunAsync<MyStack>();
|
|
|
|
var stack = resources.OfType<MyStack>().FirstOrDefault();
|
|
Assert.NotNull(stack);
|
|
|
|
var ip = await stack.PublicIp.GetValueAsync();
|
|
Assert.Equal("203.0.113.12", ip);
|
|
}
|
|
}
|
|
|
|
public static class Testing
|
|
{
|
|
public static Task<ImmutableArray<Resource>> RunAsync<T>() where T : Stack, new()
|
|
{
|
|
return Deployment.TestAsync<T>(new MyMocks(), new TestOptions { IsPreview = false });
|
|
}
|
|
|
|
public static Task<T> GetValueAsync<T>(this Output<T> output)
|
|
{
|
|
var tcs = new TaskCompletionSource<T>();
|
|
output.Apply(v =>
|
|
{
|
|
tcs.SetResult(v);
|
|
return v;
|
|
});
|
|
return tcs.Task;
|
|
}
|
|
}
|
|
}
|