From 1175d99cc709f77c12cfae3a23bf6e9977c840c8 Mon Sep 17 00:00:00 2001
From: Rostislav Wolny <rosta@mailpoet.com>
Date: Tue, 26 Mar 2019 10:55:46 +0100
Subject: [PATCH] Add cleanup methods to wooCommerce data factories

[MAILPOET-1936]
---
 tests/DataFactories/WooCommerceCustomer.php | 14 ++++++++++++++
 tests/DataFactories/WooCommerceOrder.php    | 14 ++++++++++++++
 tests/DataFactories/WooCommerceProduct.php  | 14 ++++++++++++++
 3 files changed, 42 insertions(+)

diff --git a/tests/DataFactories/WooCommerceCustomer.php b/tests/DataFactories/WooCommerceCustomer.php
index 2cc691984d..7e8dbfcd12 100644
--- a/tests/DataFactories/WooCommerceCustomer.php
+++ b/tests/DataFactories/WooCommerceCustomer.php
@@ -60,6 +60,20 @@ class WooCommerceCustomer {
     return json_decode($customer_out[0], true);
   }
 
+  /**
+   * @param int $id
+   */
+  function delete($id) {
+    $this->tester->cliToArray("wc customer delete $id --force=1 --allow-root --user=admin");
+  }
+
+  function deleteAll() {
+    $list = $this->tester->cliToArray("wc customer list --format=json --allow-root --user=admin --fields=id");
+    foreach (json_decode($list[0], true) as $item) {
+      $this->delete($item['id']);
+    }
+  }
+
   private function update($item, $value) {
     $data = $this->data;
     $data[$item] = $value;
diff --git a/tests/DataFactories/WooCommerceOrder.php b/tests/DataFactories/WooCommerceOrder.php
index aaede0f1d8..c3f25b065e 100644
--- a/tests/DataFactories/WooCommerceOrder.php
+++ b/tests/DataFactories/WooCommerceOrder.php
@@ -85,6 +85,20 @@ class WooCommerceOrder {
     return json_decode($order_out[0], true);
   }
 
+  /**
+   * @param int $id
+   */
+  function delete($id) {
+    $this->tester->cliToArray("wc shop_order delete $id --force=1 --allow-root --user=admin");
+  }
+
+  function deleteAll() {
+    $list = $this->tester->cliToArray("wc shop_order list --format=json --allow-root --user=admin --fields=id");
+    foreach (json_decode($list[0], true) as $item) {
+      $this->delete($item['id']);
+    }
+  }
+
   private function update($update_data) {
     $data = $this->data;
     foreach ($update_data as $item => $value) {
diff --git a/tests/DataFactories/WooCommerceProduct.php b/tests/DataFactories/WooCommerceProduct.php
index 013aaaf130..c1b4ece394 100644
--- a/tests/DataFactories/WooCommerceProduct.php
+++ b/tests/DataFactories/WooCommerceProduct.php
@@ -64,6 +64,20 @@ class WooCommerceProduct {
     return json_decode($product_out[0], true);
   }
 
+  /**
+   * @param int $id
+   */
+  function delete($id) {
+    $this->tester->cliToArray("wc product delete $id --force=1 --allow-root --user=admin");
+  }
+
+  function deleteAll() {
+    $list = $this->tester->cliToArray("wc product list --format=json --allow-root --user=admin --fields=id");
+    foreach (json_decode($list[0], true) as $item) {
+      $this->delete($item['id']);
+    }
+  }
+
   private function update($item, $value) {
     $data = $this->data;
     $data[$item] = $value;