[C#] feat: add SequenceAugmentation unit tests

microsoft · Jan 10, 2024 · 78acb8d · 78acb8d
1 parent e3125dd
commit 78acb8d
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/...rosoft.TeamsAI/Microsoft.TeamsAI.Tests/AITests/Augmentations/SequenceAugmentationTests.cs b/...rosoft.TeamsAI/Microsoft.TeamsAI.Tests/AITests/Augmentations/SequenceAugmentationTests.cs
@@ -12,6 +12,19 @@ namespace Microsoft.Teams.AI.Tests.AITests.Augmentations
 {
     public class SequenceAugmentationTests
     {
+        [Fact]
+        public void Test_CreatePromptSection_NotNull()
+        {
+            // Arrange
+            SequenceAugmentation augmentation = new(new());
+
+            // Act
+            var section = augmentation.CreatePromptSection();
+
+            // Assert
+            Assert.NotNull(section);
+        }
+
         [Fact]
         public async Task Test_CreatePlanFromResponseAsync_ValidPlan_ShouldSucceed()
         {
@@ -52,6 +65,25 @@ public async Task Test_CreatePlanFromResponseAsync_ValidPlan_ShouldSucceed()
             Assert.Equal("hello", (plan.Commands[1] as PredictedSayCommand)?.Response);
         }
 
+        [Fact]
+        public async Task Test_CreatePlanFromResponseAsync_EmptyResponse_ReturnsNull()
+        {
+            // Arrange
+            Mock<ITurnContext> context = new();
+            MemoryFork memory = new();
+            SequenceAugmentation augmentation = new(new());
+            PromptResponse promptResponse = new()
+            {
+                Status = PromptResponseStatus.Success
+            };
+
+            // Act
+            var plan = await augmentation.CreatePlanFromResponseAsync(context.Object, memory, promptResponse);
+
+            // Assert
+            Assert.Null(plan);
+        }
+
         [Fact]
         public async Task Test_CreatePlanFromResponseAsync_InvalidContent_ReturnsNull()
         {