Add more tests

aspirant-project · May 26, 2024 · 05a6539 · 05a6539
1 parent 31ee7ba
commit 05a6539
Show file tree

Hide file tree

Showing 3 changed files with 45 additions and 10 deletions.
diff --git a/...g/ResourceBuilderPublishWithExtensions.cs → .../ResourceBuilderExtensions.PublishWith.cs b/...g/ResourceBuilderPublishWithExtensions.cs → .../ResourceBuilderExtensions.PublishWith.cs
@@ -4,9 +4,9 @@
 namespace Aspirant.Hosting;
 
 /// <summary>
-/// <c>PublishWith</c> extension methods for <see cref="IResourceBuilder{TResource}"/>.
+/// Extension methods for <see cref="IResourceBuilder{TResource}"/>.
 /// </summary>
-public static class ResourceBuilderPublishWithExtensions
+public static partial class ResourceBuilderExtensions
 {
 /// <summary>
     /// Adds an annotation to the resource being built when <see cref="DistributedApplicationExecutionContext.IsPublishMode"/> is <c>true</c>.

diff --git a/...sting/ResourceBuilderRunWithExtensions.cs → ...ting/ResourceBuilderExtensions.RunWith.cs b/...sting/ResourceBuilderRunWithExtensions.cs → ...ting/ResourceBuilderExtensions.RunWith.cs
@@ -3,10 +3,7 @@
 
 namespace Aspirant.Hosting;
 
-/// <summary>
-/// <c>RunWith</c> extension methods for <see cref="IResourceBuilder{TResource}"/>.
-/// </summary>
-public static class ResourceBuilderRunWithExtensions
+public static partial class ResourceBuilderExtensions
 {
     /// <summary>
     /// Adds an annotation to the resource being built when <see cref="DistributedApplicationExecutionContext.IsRunMode"/> is <c>true</c>.

diff --git a/tests/Aspirant.Hosting.UnitTests/RunWithTests.cs b/tests/Aspirant.Hosting.UnitTests/RunWithTests.cs
@@ -1,4 +1,3 @@
-using System.Diagnostics.CodeAnalysis;
 using Aspire.Hosting;
 using Aspire.Hosting.ApplicationModel;
 using Microsoft.Extensions.DependencyInjection;
@@ -7,11 +6,45 @@ namespace Aspirant.Hosting.UnitTests;
 
 public class RunWithTests
 {
+    [Fact]
+    public void AnnotationAddedWhenRunning()
+    {
+        var builder = DistributedApplication.CreateBuilder();
+        var runWithAnnotation = new TestAnnotation();
+        var resource = builder.AddResource(new TestResource("resourceA"))
+            .WithAnnotation<TestAnnotation>()
+            .RunWithAnnotation(runWithAnnotation);
+
+        using var app = builder.Build();
+        var appModel = app.Services.GetRequiredService<DistributedApplicationModel>();
+        var resourceA = appModel.Resources.First(p => p.Name == "resourceA");
+
+        var annotations = resourceA.Annotations.OfType<TestAnnotation>();
+        Assert.Contains(runWithAnnotation, annotations);
+    }
+
+    [Fact]
+    public void AnnotationNotAddedWhenPublishing()
+    {
+        var builder = DistributedApplication.CreateBuilder(["Publishing:Publisher=manifest"]);
+        var runWithAnnotation = new TestAnnotation();
+        var resource = builder.AddResource(new TestResource("resourceA"))
+            .WithAnnotation<TestAnnotation>()
+            .RunWithAnnotation(runWithAnnotation);
+
+        using var app = builder.Build();
+        var appModel = app.Services.GetRequiredService<DistributedApplicationModel>();
+        var resourceA = appModel.Resources.First(p => p.Name == "resourceA");
+
+        var annotations = resourceA.Annotations.OfType<TestAnnotation>();
+        Assert.DoesNotContain(runWithAnnotation, annotations);
+    }
+
     [Fact]
     public void SimpleEnvironmentWithNameAndValueAddedWhenRunning()
     {
         var builder = DistributedApplication.CreateBuilder();
-        var resource = builder.AddResource(new ResourceA("resourceA"))
+        var resource = builder.AddResource(new TestResource("resourceA"))
             .WithEnvironment("first", "firstValue")
             .RunWithEnvironment("myName", "value");
 
@@ -27,7 +60,7 @@ public void SimpleEnvironmentWithNameAndValueAddedWhenRunning()
     public void SimpleEnvironmentWithNameAndValueNotAddedWhenPublishing()
     {
         var builder = DistributedApplication.CreateBuilder(["Publishing:Publisher=manifest"]);
-        var resource = builder.AddResource(new ResourceA("resourceA"))
+        var resource = builder.AddResource(new TestResource("resourceA"))
             .WithEnvironment("first", "firstValue")
             .RunWithEnvironment("myName", "value");
 
@@ -56,10 +89,15 @@ private static Dictionary<string, object> MaterializeEnvironmentVariables(Distri
         return environmentVariables;
     }
 
-    private sealed class ResourceA(string name) : IResourceWithEnvironment
+    private sealed class TestResource(string name) : IResourceWithEnvironment
     {
         public string Name => name;
 
         public ResourceAnnotationCollection Annotations { get; } = [];
     }
+
+    private sealed class TestAnnotation : IResourceAnnotation
+    {
+
+    }
 }