woocommerce-paypal-payments/tests/integration/PHPUnit/Factories/ProductFactory.php

<?php
declare(strict_types=1);

namespace WooCommerce\PayPalCommerce\Tests\Integration\Factories;

use WC_Product_Simple;
use WC_Product_Variable;
use WC_Product_Variation;
use WooCommerce\PayPalCommerce\Tests\Integration\Fixtures\ProductPresets;

class ProductFactory
{
    private array $created_product_ids = [];

    /**
     * @param string $preset_name
     * @return \WC_Product
     * @throws \WC_Data_Exception
     */
    public function createFromPreset(string $preset_name): \WC_Product
    {
        $presets = ProductPresets::get();

        if (!isset($presets[$preset_name])) {
            throw new \WC_Data_Exception('invalid_preset', "Product preset '{$preset_name}' not found");
        }

        $preset = $presets[$preset_name];

        if (isset($preset['sku'])) {
            $existing_product_id = wc_get_product_id_by_sku($preset['sku']);
            if ($existing_product_id) {
                $existing_product = wc_get_product($existing_product_id);
                if ($existing_product) {
                    return $existing_product;
                }
            }
        }

        if (isset($preset['product_id'])) {
            $existing_product = wc_get_product($preset['product_id']);
            if ($existing_product) {
                return $existing_product;
            }
        }

        if ($preset['type'] === 'subscription' && !class_exists('\WC_Product_Subscription')) {
            return $this->createSimpleProduct($preset);
        }
        switch ($preset['type']) {
            case 'variable':
                return $this->createVariableProduct($preset);
            case 'subscription':
                return $this->createSubscriptionProduct($preset);
            case 'simple':
            default:
                return $this->createSimpleProduct($preset);
        }
    }

    /**
     * @param array $preset
     * @return WC_Product_Simple
     */
    private function createSimpleProduct(array $preset): WC_Product_Simple
    {
        $product = new WC_Product_Simple();
        $product_id = wc_get_product_id_by_sku($preset['sku']);
        $product->set_sku($preset['sku']);
        $product->set_id($product_id);
        $product->set_name($preset['name']);
        $product->set_regular_price($preset['price']);
        $product->set_status('publish');
        $product->save();

        $this->created_product_ids[] = $product_id;

        return $product;
    }

    /**
     * @param array $preset
     * @return WC_Product_Variation
     */
    private function createVariableProduct(array $preset): WC_Product_Variation
    {
        // Create parent variable product
        $parent = new WC_Product_Variable();
        $product_id = wc_get_product_id_by_sku($preset['sku']);
        $parent->set_sku($preset['sku']);
        $parent->set_id($product_id);
        $parent->set_name($preset['name']);
        $parent->set_status('publish');
        $parent->save();

        // Create variation
        $variation = new WC_Product_Variation();
        $variation->set_id($preset['variation_id']);
        $variation->set_parent_id($preset['product_id']);
        $variation->set_regular_price($preset['price']);
        $variation->set_attributes(['color' => 'red']);
        $variation->set_status('publish');
        $variation->save();

        $this->created_product_ids[] = $product_id;
        $this->created_product_ids[] = $preset['variation_id'];

        return $variation;
    }

    /**
     * @param array $preset
     * @return \WC_Product_Subscription
     */
    private function createSubscriptionProduct(array $preset): \WC_Product_Subscription
    {
        $product = new \WC_Product_Subscription();
        $product->set_props([
            'name' => $preset['name'],
            'regular_price' => $preset['price'],
            'price' => $preset['price'],
            'sku' => $preset['sku'],
            'manage_stock' => false,
            'tax_status' => 'taxable',
            'downloadable' => false,
            'virtual' => false,
            'stock_status' => 'instock',
            'weight' => '1.1',
            'subscription_period' => $preset['subscription_period'],
            'subscription_period_interval' => $preset['subscription_period_interval'],
            'subscription_length' => $preset['subscription_length'],
            'subscription_trial_period' => $preset['subscription_trial_period'],
            'subscription_trial_length' => $preset['subscription_trial_length'],
            'subscription_price' => $preset['subscription_price'],
            'subscription_sign_up_fee' => $preset['subscription_sign_up_fee'],
        ]);

        $product->set_status('publish');
        $product->save();

        $this->created_product_ids[] = $product->get_id();
        return $product;
    }

    /**
     * @param string $sku
     * @return bool
     */
    public function exists(string $sku): bool
    {
        $existing_product_id = wc_get_product_id_by_sku($sku);
        return (bool)$existing_product_id;
    }

    /**
     * Delete all created products
     */
    public function cleanup(): void
    {
        foreach ($this->created_product_ids as $product_id) {
            wp_delete_post($product_id, true);
        }

        $this->created_product_ids = [];
    }
}