Move additional checks for adding default rows into the extension

wordpress-mobile · Jun 20, 2024 · 7684a35 · 7684a35
1 parent 25bf46d
commit 7684a35
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/WordPress/Classes/ViewRelated/Stats/Extensions/UITableViewCell+Stats.swift b/WordPress/Classes/ViewRelated/Stats/Extensions/UITableViewCell+Stats.swift
@@ -98,6 +98,9 @@ extension UITableViewCell {
                              inStackView rowsStackView: UIStackView,
                              forType statType: StatType,
                              configuration: StatsTotalRowConfiguration) {
+         if rowsStackView.arrangedSubviews.isEmpty {
+             addDefaultTotalRows(toStackView: rowsStackView)
+         }
 
         guard !dataRows.isEmpty else {
             configureForNoData(inStackView: rowsStackView, forType: statType)

diff --git a/WordPress/Classes/ViewRelated/Stats/Period Stats/Countries/CountriesCell.swift b/WordPress/Classes/ViewRelated/Stats/Period Stats/Countries/CountriesCell.swift
@@ -44,9 +44,6 @@ class CountriesCell: StatsBaseCell, NibLoadable {
         bottomSeparatorLine.isHidden = forDetails
 
         if !forDetails {
-            if rowsStackView.arrangedSubviews.isEmpty {
-                addDefaultTotalRows(toStackView: rowsStackView)
-            }
             configureTotalRows(
                 dataRows,
                 inStackView: rowsStackView,

diff --git a/WordPress/Classes/ViewRelated/Stats/Shared Views/TopTotalsCell.swift b/WordPress/Classes/ViewRelated/Stats/Shared Views/TopTotalsCell.swift
@@ -83,9 +83,6 @@ class TopTotalsCell: StatsBaseCell, NibLoadable {
         self.forDetails = forDetails
 
         if !forDetails {
-            if rowsStackView.arrangedSubviews.isEmpty {
-                addDefaultTotalRows(toStackView: rowsStackView)
-            }
             configureTotalRows(
                 dataRows,
                 inStackView: rowsStackView,