Fix time tracking parsing of months59729-estimate-quick-action-does-not-produce-correct-time-for-1mo

Patches ChronicDuration to use our custom conversions
when parsing months

5 files changed, 79 insertions, 0 deletions

diff --git a/changelogs/unreleased/59729-estimate-quick-action-does-not-produce-correct-time-for-1mo.yml b/changelogs/unreleased/59729-estimate-quick-action-does-not-produce-correct-time-for-1mo.yml
new file mode 100644
index 00000000000..1962a662179
--- /dev/null
+++ b/changelogs/unreleased/59729-estimate-quick-action-does-not-produce-correct-time-for-1mo.yml
@@ -0,0 +1,5 @@
+---
+title: Fix parsing of months in time tracking commands
+merge_request: 32165
+author:
+type: fixed
diff --git a/config/initializers/chronic_duration.rb b/config/initializers/chronic_duration.rb
index b65b06c813a..aa43eef434c 100644
--- a/config/initializers/chronic_duration.rb
+++ b/config/initializers/chronic_duration.rb
@@ -1 +1,5 @@
+# frozen_string_literal: true
+
 ChronicDuration.raise_exceptions = true
+
+ChronicDuration.prepend Gitlab::Patch::ChronicDuration
diff --git a/lib/gitlab/patch/chronic_duration.rb b/lib/gitlab/patch/chronic_duration.rb
new file mode 100644
index 00000000000..ab3cba3657f
--- /dev/null
+++ b/lib/gitlab/patch/chronic_duration.rb
@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+# Fixes a bug where parsing months doesn't take into account
+# the ChronicDuration.days_per_week setting
+#
+# We can remove this when we do a refactor and push upstream in
+# https://gitlab.com/gitlab-org/gitlab-ce/issues/66637
+
+module Gitlab
+  module Patch
+    module ChronicDuration
+      extend ActiveSupport::Concern
+
+      class_methods do
+        def duration_units_seconds_multiplier(unit)
+          return 0 unless duration_units_list.include?(unit)
+
+          case unit
+          when 'months'
+            3600 * ::ChronicDuration.hours_per_day * ::ChronicDuration.days_per_month
+          else
+            super
+          end
+        end
+
+        # ChronicDuration#output uses 1mo = 4w as the conversion so we do the same here.
+        # We do need to add a special case for the default days_per_week value because
+        # we want to retain existing behavior for the default case
+        def days_per_month
+          ::ChronicDuration.days_per_week == 7 ? 30 : ::ChronicDuration.days_per_week * 4
+        end
+      end
+    end
+  end
+end
diff --git a/spec/lib/gitlab/patch/chronic_duration_spec.rb b/spec/lib/gitlab/patch/chronic_duration_spec.rb
new file mode 100644
index 00000000000..541037ec1a2
--- /dev/null
+++ b/spec/lib/gitlab/patch/chronic_duration_spec.rb
@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe Gitlab::Patch::ChronicDuration do
+  subject { ChronicDuration.parse('1mo') }
+
+  it 'uses default conversions' do
+    expect(subject).to eq(2_592_000)
+  end
+
+  context 'with custom conversions' do
+    before do
+      ChronicDuration.hours_per_day = 8
+      ChronicDuration.days_per_week = 5
+    end
+
+    after do
+      ChronicDuration.hours_per_day = 24
+      ChronicDuration.days_per_week = 7
+    end
+
+    it 'uses custom conversions' do
+      expect(subject).to eq(576_000)
+    end
+  end
+end
diff --git a/spec/lib/gitlab/time_tracking_formatter_spec.rb b/spec/lib/gitlab/time_tracking_formatter_spec.rb
index a85d418777f..cfc804c13a7 100644
--- a/spec/lib/gitlab/time_tracking_formatter_spec.rb
+++ b/spec/lib/gitlab/time_tracking_formatter_spec.rb
@@ -17,6 +17,14 @@ describe Gitlab::TimeTrackingFormatter do
 
       it { expect(subject).to eq(-12_000) }
     end
+
+    context 'durations with months' do
+      let(:duration_string) { '1mo' }
+
+      it 'uses our custom conversions' do
+        expect(subject).to eq(576_000)
+      end
+    end
   end
 
   describe '#output' do