WooCommerce: Exclude the Test product category from shop page except for admins and testers.

woocommerce-exclude-product-category-from-shop-page.php

<?php

/**
 * Exclude test products from a particular category on the shop page
 * for non admins and non testers.
 *
 * I.e., only let admins and testers see test products.
 *
 * Reference: https://docs.woocommerce.com/document/exclude-a-category-from-the-shop-page/
 */
function custom_pre_get_posts_query( $q ) {
  if ( is_user_logged_in() ) {
    $user = wp_get_current_user();
    $user_role = $user->roles[0];
    if ( ( $user_role === 'administrator') || ( $user_role === 'tester' ) ) {
      return;
    }
  }

  $tax_query = (array) $q->get( 'tax_query' );

  $tax_query[] = array(
    'taxonomy' => 'product_cat',
    'field' => 'slug',
    'terms' => array( 'test' ), // Don't display products in the test category on the shop page.
    'operator' => 'NOT IN'
  );

  $q->set( 'tax_query', $tax_query );
}
add_action( 'woocommerce_product_query', 'custom_pre_get_posts_query' );