Defensive wait in test

Author: linkdotnet

tests/LinkDotNet.Blog.IntegrationTests/Web/Features/Archive/ArchivePageTests.cs

@@ -21,7 +21,7 @@ public async Task ShouldDisplayAllBlogPosts()
 
         var cut = ctx.RenderComponent<ArchivePage>();
 
-        cut.WaitForElements("h2");
+        cut.WaitForState(() => cut.FindAll("h2").Count == 2);
         var yearHeader = cut.FindAll("h2");
         yearHeader.Should().HaveCount(2);
         yearHeader[0].TextContent.Should().Be("2022");