diff --git a/lib/tasks/data/truncate_data.rake b/lib/tasks/data/truncate_data.rake index 69f529cc67..12c7866926 100644 --- a/lib/tasks/data/truncate_data.rake +++ b/lib/tasks/data/truncate_data.rake @@ -9,10 +9,11 @@ require 'tasks/data/truncate_data' namespace :ofn do namespace :data do desc 'Truncate data' - task truncate: :environment do + task :truncate, [:months_to_keep] => :environment do |_task, args| guard_and_warn - TruncateData.new.call + months_to_keep = args.months_to_keep.to_i + TruncateData.new(months_to_keep).call end def guard_and_warn diff --git a/lib/tasks/data/truncate_data.rb b/lib/tasks/data/truncate_data.rb index 2add9f6111..5534ec4dfc 100644 --- a/lib/tasks/data/truncate_data.rb +++ b/lib/tasks/data/truncate_data.rb @@ -1,7 +1,7 @@ # frozen_string_literal: true class TruncateData - def initialize(months_to_keep: nil) + def initialize(months_to_keep = nil) @date = (months_to_keep || 3).months.ago end diff --git a/spec/lib/tasks/data/truncate_data_spec.rb b/spec/lib/tasks/data/truncate_data_spec.rb index 9ed8075c72..6d26ae9a29 100644 --- a/spec/lib/tasks/data/truncate_data_spec.rb +++ b/spec/lib/tasks/data/truncate_data_spec.rb @@ -29,7 +29,7 @@ describe TruncateData do ) create(:order, order_cycle: order_cycle) - TruncateData.new(months_to_keep: nil).call + TruncateData.new(nil).call expect(OrderCycle.all).to be_empty end @@ -46,7 +46,7 @@ describe TruncateData do ) create(:order, order_cycle: recent_order_cycle) - TruncateData.new(months_to_keep: 6).call + TruncateData.new(6).call expect(OrderCycle.all).to contain_exactly(recent_order_cycle) end