refactoring tests

eebbesen · Dec 19, 2024 · 8a284ad · 8a284ad
1 parent fd69031
commit 8a284ad
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 9 deletions.
diff --git a/StpFoodBlazor/StpFoodBlazor/Components/Pages/Deals.razor b/StpFoodBlazor/StpFoodBlazor/Components/Pages/Deals.razor
@@ -14,13 +14,14 @@
     <div class="container mb-5">
         <DealFilterBar selectedDayOfWeek="@selectedDayOfWeek" OnSelectedDayChanged="OnSelectedDayChanged"
             happyHour="@happyHour" OnHappyHourChanged="OnHappyHourChanged" />
-        <DealTableHeader selectedDayOfWeek="@selectedDayOfWeek" />
 @if (deals == null)
 {
+        <DealTableHeader selectedDayOfWeek="@TimeServiceImpl.GetDayOfWeek()" />
         <DealTableBodyPlaceholder />
 }
 else
 {
+        <DealTableHeader selectedDayOfWeek="@selectedDayOfWeek" />
         <DealTableBody deals="@deals" selectedDayOfWeek="@selectedDayOfWeek" />
 }
     </div>

diff --git a/StpFoodBlazor/StpFoodBlazorTest/Pages/DealsTest.razor b/StpFoodBlazor/StpFoodBlazorTest/Pages/DealsTest.razor
@@ -27,9 +27,10 @@
         ctx.Services.AddSingleton<IDealService>(testDealService);
 
         var cut = ctx.Render(@<Deals />);
-        cut.WaitForElement("#deals_table_body_placeholder");
-        var elements = cut.Find("#deals_table_body_placeholder");
+        var header = cut.WaitForElement("#deals_table_header");
+        var elements = cut.WaitForElement("#deals_table_body_placeholder");
 
+        Assert.Equal("Place", header.Children[0].InnerHtml);
         Assert.Equal(30, elements.ChildElementCount);
         Assert.Equal("<span class=\"col-12 rounded placeholder bg-white\">&nbsp;</span>",
             elements.Children[0].InnerHtml);
@@ -188,7 +189,6 @@
 
     private IElement getElements(IRenderedFragment cut)
     {
-        cut.WaitForElement("#deals_table_body", TimeSpan.FromSeconds(5));
-        return cut.Find("#deals_table_body");
+        return cut.WaitForElement("#deals_table_body", TimeSpan.FromSeconds(5));
     }
 }
diff --git a/StpFoodBlazor/StpFoodBlazorTest/Pages/GiftCardsTest.razor b/StpFoodBlazor/StpFoodBlazorTest/Pages/GiftCardsTest.razor
@@ -25,9 +25,10 @@
         ctx.Services.AddSingleton<IGiftCardService>(testGiftCardService);
 
         var cut = ctx.Render(@<GiftCards />);
-        cut.WaitForElement("#giftcards_table_body_placeholder");
-        var elements = cut.Find("#giftcards_table_body_placeholder");
+        var header = cut.WaitForElement("#giftcards_table_header");
+        var elements = cut.WaitForElement("#giftcards_table_body_placeholder");
 
+        Assert.Equal("Place", header.Children[0].InnerHtml);
         Assert.Equal(30, elements.ChildElementCount);
         Assert.Equal("<span class=\"col-12 rounded placeholder bg-white\">&nbsp;</span>",
             elements.Children[0].InnerHtml);
@@ -74,7 +75,6 @@
 
     private IElement getElements(IRenderedFragment cut)
     {
-        cut.WaitForElement("#giftcards_table_body", TimeSpan.FromSeconds(5));
-        return cut.Find("#giftcards_table_body");
+        return cut.WaitForElement("#giftcards_table_body", TimeSpan.FromSeconds(5));
     }
 }