From 6ceeda7d9e4e3c79586f3d511e2caeae26544e0c Mon Sep 17 00:00:00 2001
From: Pau Perez <saulopefa@gmail.com>
Date: Mon, 16 Mar 2020 18:57:25 +0100
Subject: [PATCH] Instrument TruncateData logging start and end

---
 lib/tasks/data/truncate_data.rb           | 26 +++++++++++++++--------
 spec/lib/tasks/data/truncate_data_spec.rb |  1 +
 2 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/lib/tasks/data/truncate_data.rb b/lib/tasks/data/truncate_data.rb
index 0c5e86c538..2e9a703535 100644
--- a/lib/tasks/data/truncate_data.rb
+++ b/lib/tasks/data/truncate_data.rb
@@ -12,26 +12,34 @@ class TruncateData
   end
 
   def call
-    truncate_inventory
-    truncate_adjustments
-    truncate_order_associations
-    truncate_order_cycle_data
+    logging do
+      truncate_inventory
+      truncate_adjustments
+      truncate_order_associations
+      truncate_order_cycle_data
 
-    sql_delete_from "spree_orders #{where_oc_id_in_ocs_to_delete}"
+      sql_delete_from "spree_orders #{where_oc_id_in_ocs_to_delete}"
 
-    truncate_subscriptions
+      truncate_subscriptions
 
-    sql_delete_from "order_cycles #{where_ocs_to_delete}"
+      sql_delete_from "order_cycles #{where_ocs_to_delete}"
 
-    Spree::TokenizedPermission.where("created_at < '#{date}'").delete_all
+      Spree::TokenizedPermission.where("created_at < '#{date}'").delete_all
 
-    remove_transient_data
+      remove_transient_data
+    end
   end
 
   private
 
   attr_reader :date
 
+  def logging
+    Rails.logger.info("TruncateData started with truncation date #{date}")
+    yield
+    Rails.logger.info("TruncateData finished")
+  end
+
   def truncate_order_associations
     sql_delete_from "spree_line_items #{where_order_id_in_orders_to_delete}"
     sql_delete_from "spree_payments #{where_order_id_in_orders_to_delete}"
diff --git a/spec/lib/tasks/data/truncate_data_spec.rb b/spec/lib/tasks/data/truncate_data_spec.rb
index d5972fa2ea..3362450c48 100644
--- a/spec/lib/tasks/data/truncate_data_spec.rb
+++ b/spec/lib/tasks/data/truncate_data_spec.rb
@@ -8,6 +8,7 @@ describe TruncateData do
       allow(Spree::StateChange).to receive(:delete_all)
       allow(Spree::LogEntry).to receive(:delete_all)
       allow(TruncateData::Session).to receive(:delete_all)
+      allow(Rails.logger).to receive(:info)
     end
 
     context 'when months_to_keep is not specified' do