From d4d244fa0e94c20f33d3753bf5c9a9cb369a465b Mon Sep 17 00:00:00 2001
From: John Oleksowicz <john.oleksowicz@automattic.com>
Date: Fri, 4 Aug 2023 13:52:16 -0500
Subject: [PATCH] Ensure coupon stats are updated in tests

MAILPOET-5007
---
 mailpoet/tests/_support/IntegrationTester.php | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/mailpoet/tests/_support/IntegrationTester.php b/mailpoet/tests/_support/IntegrationTester.php
index cfb725bc18..c972ab0f3c 100644
--- a/mailpoet/tests/_support/IntegrationTester.php
+++ b/mailpoet/tests/_support/IntegrationTester.php
@@ -1,6 +1,5 @@
 <?php declare(strict_types = 1);
 
-use Automattic\WooCommerce\Admin\API\Reports\Orders\Stats\DataStore;
 use Codeception\Scenario;
 use MailPoet\Automation\Engine\Data\Automation;
 use MailPoet\Automation\Engine\Data\AutomationRun;
@@ -214,10 +213,12 @@ class IntegrationTester extends \Codeception\Actor {
   }
 
   public function updateWooOrderStats(int $orderId): void {
-    if (!class_exists('Automattic\WooCommerce\Admin\API\Reports\Orders\Stats\DataStore')) {
-      return;
+    if (class_exists('Automattic\WooCommerce\Admin\API\Reports\Orders\Stats\DataStore')) {
+      \Automattic\WooCommerce\Admin\API\Reports\Orders\Stats\DataStore::sync_order($orderId);
+    }
+    if (class_exists('Automattic\WooCommerce\Admin\API\Reports\Coupons\DataStore')) {
+      \Automattic\WooCommerce\Admin\API\Reports\Coupons\DataStore::sync_order_coupons($orderId);
     }
-    DataStore::sync_order($orderId);
   }
 
   public function deleteWordPressTerms(): void {