expand traits registration for refund and shipping views

Author: iamgabrielma

WooCommerce/Classes/ViewRelated/Orders/Order Details/Issue Refunds/Cells/RefundShippingDetailsTableViewCell.swift

@@ -52,9 +52,15 @@ final class RefundShippingDetailsTableViewCell: UITableViewCell {
         super.awakeFromNib()
         applyCellStyles()
         applyAccessibilityChanges()
-
-        registerForTraitChanges([UITraitPreferredContentSizeCategory.self]) { [weak self] (_: RefundShippingDetailsTableViewCell, _: UITraitCollection) in
-            self?.applyAccessibilityChanges()
+        
+        let traits: [UITrait] = [
+            UITraitPreferredContentSizeCategory.self,
+            UITraitUserInterfaceIdiom.self,
+            UITraitVerticalSizeClass.self
+        ]
+
+        registerForTraitChanges(traits) { (self: Self, _: UITraitCollection) in
+            self.applyAccessibilityChanges()
         }
     }
 

WooCommerce/Classes/ViewRelated/Orders/Order Details/Issue Refunds/IssueRefundViewController.swift

@@ -48,9 +48,15 @@ final class IssueRefundViewController: UIViewController {
         viewModel.fetch()
         updateWithViewModelContent()
 
-        registerForTraitChanges([UITraitPreferredContentSizeCategory.self]) { [weak self] (_: IssueRefundViewController, _: UITraitCollection) in
-            self?.configureHeaderStackView()
-            self?.tableView.updateHeaderHeight()
+        let traits: [UITrait] = [
+            UITraitPreferredContentSizeCategory.self,
+            UITraitUserInterfaceIdiom.self,
+            UITraitVerticalSizeClass.self
+        ]
+
+        registerForTraitChanges(traits) {(self: Self, _: UITraitCollection) in
+            self.configureHeaderStackView()
+            self.tableView.updateHeaderHeight()
         }
     }