test : GetValue

StefH · May 7, 2020 · b2e6ad7 · b2e6ad7
1 parent edd4c2c
commit b2e6ad7
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 7 deletions.
diff --git a/tests/Pulumi.Azure.Extensions.Tests/Storage/BlobCollectionTests.cs b/tests/Pulumi.Azure.Extensions.Tests/Storage/BlobCollectionTests.cs
@@ -129,6 +129,8 @@ public async Task Blob()
 
  Assert.NotNull(blob);
  blob.GetResourceName().Should().Be("test");
+
+ (await blob.Name.GetValueAsync()).Should().Be("test");
  }
 
  [Fact]
@@ -162,6 +164,11 @@ public async Task Folder_WithFiles()
  Assert.NotNull(blobs);
 
  blobs.Count.Should().Be(4);
+
+ blobs[0].Name.GetValue().Should().Be("files.zip");
+ blobs[1].Name.GetValue().Should().Be("HtmlFile1.html");
+ blobs[2].Name.GetValue().Should().Be("TextFile1.txt");
+ blobs[3].Name.GetValue().Should().Be("x\\TextFile3.txt");
  }
 
  [Fact]

diff --git a/tests/Pulumi.Azure.Extensions.Tests/TestingExtensions.cs b/tests/Pulumi.Azure.Extensions.Tests/TestingExtensions.cs
@@ -1,17 +1,24 @@
-using System;
-using System.Collections.Generic;
-using System.Text;
-using System.Threading.Tasks;
+using System.Threading.Tasks;
 
 namespace Pulumi.Azure.Extensions.Tests
 {
  internal static class TestingExtensions
  {
- public static T GetValue<T>(this Output<T> output)
+ public static Task<T> GetValueAsync<T>(this Output<T> output)
  {
- var field = output.GetType().GetField("DataTask");
+ var tcs = new TaskCompletionSource<T>();
+ output.Apply(v =>
+ {
+ tcs.SetResult(v);
+ return v;
+ });
+
+ return tcs.Task;
+ }
 
- return default;
+ public static T GetValue<T>(this Output<T> output)
+ {
+ return output.GetValueAsync().GetAwaiter().GetResult();
  }
  }
 }