(statiqdev#225) fixed ElseIf when Predicate.RequiresDocument is false

nils-a · Dec 19, 2022 · 73347ec · 73347ec
1 parent ca7a49c
commit 73347ec
Show file tree

Hide file tree

Showing 2 changed files with 121 additions and 1 deletion.
diff --git a/src/core/Statiq.Core/Modules/Control/ExecuteIf.cs b/src/core/Statiq.Core/Modules/Control/ExecuteIf.cs
@@ -239,7 +239,7 @@ protected override async Task<IEnumerable<IDocument>> ExecuteContextAsync(IExecu
                         }
                     }
                 }
-                else
+                else if (!hasExecuted)
                 {
                     if (await condition.Predicate.GetValueAsync(null, context))
                     {

diff --git a/tests/core/Statiq.Core.Tests/Modules/Control/ExecuteIfFixture.cs b/tests/core/Statiq.Core.Tests/Modules/Control/ExecuteIfFixture.cs
@@ -252,6 +252,126 @@ public async Task FalseIfWithContextResultsInCorrectCounts()
                 c.OutputCount.ShouldBe(12);
             }
 
+            [Test]
+            public async Task TrueIfWithFollowingElseIfWithContextResultsInCorrectCounts()
+            {
+                // Given
+                CountModule a = new CountModule("A")
+                {
+                    AdditionalOutputs = 2,
+                    EnsureInputDocument = true
+                };
+                CountModule b = new CountModule("B")
+                {
+                    AdditionalOutputs = 2
+                };
+                CountModule c = new CountModule("C")
+                {
+                    AdditionalOutputs = 3
+                };
+
+                // When
+                await ExecuteAsync(
+                    a,
+                    new ExecuteIf(Config.FromContext(_ => true), b)
+                        .ElseIf(Config.FromContext(_ => true), c));
+
+                // Then
+                a.ExecuteCount.ShouldBe(1);
+                b.ExecuteCount.ShouldBe(1);
+                c.ExecuteCount.ShouldBe(0);
+            }
+
+            [Test]
+            public async Task FalseIfWithFollowingElseIfWithContextResultsInCorrectCounts()
+            {
+                // Given
+                CountModule a = new CountModule("A")
+                {
+                    AdditionalOutputs = 2,
+                    EnsureInputDocument = true
+                };
+                CountModule b = new CountModule("B")
+                {
+                    AdditionalOutputs = 2
+                };
+                CountModule c = new CountModule("C")
+                {
+                    AdditionalOutputs = 3
+                };
+
+                // When
+                await ExecuteAsync(
+                    a,
+                    new ExecuteIf(Config.FromContext(_ => false), b)
+                        .ElseIf(Config.FromContext(_ => true), c));
+
+                // Then
+                a.ExecuteCount.ShouldBe(1);
+                b.ExecuteCount.ShouldBe(0);
+                c.ExecuteCount.ShouldBe(1);
+            }
+
+            [Test]
+            public async Task TrueIfWithFollowingElseWithContextResultsInCorrectCounts()
+            {
+                // Given
+                CountModule a = new CountModule("A")
+                {
+                    AdditionalOutputs = 2,
+                    EnsureInputDocument = true
+                };
+                CountModule b = new CountModule("B")
+                {
+                    AdditionalOutputs = 2
+                };
+                CountModule c = new CountModule("C")
+                {
+                    AdditionalOutputs = 3
+                };
+
+                // When
+                await ExecuteAsync(
+                    a,
+                    new ExecuteIf(Config.FromContext(_ => true), b)
+                        .Else(c));
+
+                // Then
+                a.ExecuteCount.ShouldBe(1);
+                b.ExecuteCount.ShouldBe(1);
+                c.ExecuteCount.ShouldBe(0);
+            }
+
+            [Test]
+            public async Task FalseIfWithFollowingElseWithContextResultsInCorrectCounts()
+            {
+                // Given
+                CountModule a = new CountModule("A")
+                {
+                    AdditionalOutputs = 2,
+                    EnsureInputDocument = true
+                };
+                CountModule b = new CountModule("B")
+                {
+                    AdditionalOutputs = 2
+                };
+                CountModule c = new CountModule("C")
+                {
+                    AdditionalOutputs = 3
+                };
+
+                // When
+                await ExecuteAsync(
+                    a,
+                    new ExecuteIf(Config.FromContext(_ => false), b)
+                        .Else(c));
+
+                // Then
+                a.ExecuteCount.ShouldBe(1);
+                b.ExecuteCount.ShouldBe(0);
+                c.ExecuteCount.ShouldBe(1);
+            }
+
             [Test]
             public async Task UnmatchedDocumentsAreAddedToResults()
             {