Fixed deposit tests class

Automattic · Jan 4, 2024 · a9b2903 · a9b2903
1 parent 18af61d
commit a9b2903
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/tests/unit/helpers/class-wc-helper-deposit-product-manager.php b/tests/unit/helpers/class-wc-helper-deposit-product-manager.php
@@ -11,10 +11,25 @@
  * This helper class should ONLY be used for unit tests!.
  */
 class WC_Deposits_Product_Manager {
-	public static function get_deposit_type( WC_Product_Simple $product ) {
+	/**
+	 * @param WC_Product_Simple|int $product
+	 * @return mixed
+	 */
+	public static function get_deposit_type( $product ) {
+		if ( is_int( $product ) ) {
+			$product = wc_get_product( $product );
+		}
 		return $product->get_meta( '_wc_deposit_type' );
 	}
-	public static function deposits_enabled( WC_Product_Simple $product ) {
+
+	/**
+	 * @param WC_Product_Simple|int $product
+	 * @return bool
+	 */
+	public static function deposits_enabled( $product ) {
+		if ( is_int( $product ) ) {
+			$product = wc_get_product( $product );
+		}
 		return true === $product->get_meta( '_wc_deposits_enabled' );
 	}
 }