Merge branch 'move-shared-runners-remaining-minutes-to-quota' into 'master'

Move shared_runners_remaining_minutes to Ci::Minutes::Quota

See merge request gitlab-org/gitlab!48939

ee/app/models/ci/minutes/context.rb

@@ -5,7 +5,6 @@ module Ci
     class Context
       delegate :shared_runners_minutes_limit_enabled?, to: :level
       delegate :name, to: :namespace, prefix: true
-      delegate :shared_runners_remaining_minutes_percent, to: :namespace
 
       attr_reader :level
 
@@ -15,9 +14,17 @@ module Ci
         @level = project || namespace
       end
 
+      def percent_total_minutes_remaining
+        quota.percent_total_minutes_remaining
+      end
+
       private
 
       attr_reader :project, :namespace
+
+      def quota
+        @quota ||= ::Ci::Minutes::Quota.new(namespace)
+      end
     end
   end
 end

ee/app/models/ci/minutes/notification.rb

@@ -50,7 +50,7 @@ module Ci
       end
 
       def calculate_notification_stage
-        percentage = context.shared_runners_remaining_minutes_percent.to_i
+        percentage = context.percent_total_minutes_remaining.to_i
 
         if percentage <= PERCENTAGES[:exceeded]
           :exceeded

ee/app/models/ci/minutes/quota.rb

@@ -52,8 +52,18 @@ module Ci
         @total_minutes_used ||= namespace.shared_runners_seconds.to_i / 60
       end
 
+      def percent_total_minutes_remaining
+        return 0 if total_minutes == 0
+
+        100 * total_minutes_remaining.to_i / total_minutes
+      end
+
       private
 
+      def total_minutes_remaining
+        [total_minutes.to_i - total_minutes_used, 0].max
+      end
+
       def monthly_minutes_used_up?
         namespace.shared_runners_minutes_limit_enabled? &&
           monthly_minutes_used >= monthly_minutes

ee/app/models/ee/namespace.rb

@@ -265,17 +265,6 @@ module EE
         actual_shared_runners_minutes_limit.nonzero?
     end
 
-    def shared_runners_remaining_minutes_percent
-      return 0 if shared_runners_remaining_minutes.to_f <= 0
-      return 0 if actual_shared_runners_minutes_limit.to_f == 0
-
-      (shared_runners_remaining_minutes.to_f * 100) / actual_shared_runners_minutes_limit.to_f
-    end
-
-    def shared_runners_remaining_minutes_below_threshold?
-      shared_runners_remaining_minutes_percent.to_i <= last_ci_minutes_usage_notification_level.to_i
-    end
-
     def any_project_with_shared_runners_enabled?
       all_projects.with_shared_runners.any?
     end
@@ -445,10 +434,6 @@ module EE
       )
     end
 
-    def shared_runners_remaining_minutes
-      [actual_shared_runners_minutes_limit.to_f - ci_minutes_quota.total_minutes_used.to_f, 0].max
-    end
-
     def total_repository_size_excess_calculation(repository_size_limit, project_level: true)
       total_excess = (total_repository_size_arel - repository_size_limit).sum
       relation = projects_for_repository_size_excess((repository_size_limit unless project_level))

ee/app/models/ee/project.rb

@@ -182,8 +182,7 @@ module EE
       delegate :shared_runners_minutes, :shared_runners_seconds, :shared_runners_seconds_last_reset,
         to: :statistics, allow_nil: true
 
-      delegate :actual_shared_runners_minutes_limit,
-               :shared_runners_remaining_minutes_below_threshold?, to: :shared_runners_limit_namespace
+      delegate :actual_shared_runners_minutes_limit, to: :shared_runners_limit_namespace
 
       delegate :last_update_succeeded?, :last_update_failed?,
         :ever_updated_successfully?, :hard_failed?,

ee/spec/features/ci_shared_runner_warnings_spec.rb

@@ -20,21 +20,21 @@ RSpec.describe 'CI shared runner limits' do
       group.add_developer(user)
     end
 
-    where(:case_name, :percent, :remaining_minutes) do
-      'warning level' | 30 | 4
-      'danger level' | 5 | 1
+    where(:case_name, :percent_threshold, :minutes_limit, :minutes_used) do
+      'warning level' | 30 | 1000 | 800
+      'danger level'  | 5  | 1000 | 980
     end
 
     with_them do
       context "when there is a notification and minutes still exist", :js do
         let(:message) do
-          "Group #{group.name} has #{percent}% or less Shared Runner Pipeline minutes remaining. " \
+          "Group #{group.name} has #{percent_threshold}% or less Shared Runner Pipeline minutes remaining. " \
             "Once it runs out, no new jobs or pipelines in its projects will run."
         end
 
         before do
-          group.update!(shared_runners_minutes_limit: 20)
-          allow_any_instance_of(EE::Namespace).to receive(:shared_runners_remaining_minutes).and_return(remaining_minutes)
+          group.update!(shared_runners_minutes_limit: minutes_limit)
+          allow_any_instance_of(EE::Namespace).to receive(:shared_runners_seconds).and_return(minutes_used.minutes)
         end
 
         it 'displays a warning message on pipelines page' do
@@ -111,21 +111,21 @@ RSpec.describe 'CI shared runner limits' do
       group.add_owner(user)
     end
 
-    where(:case_name, :percent, :remaining_minutes) do
-      'warning level' | 30 | 4
-      'danger level' | 5 | 1
+    where(:case_name, :percent_threshold, :minutes_limit, :minutes_used) do
+      'warning level' | 30 | 1000 | 800
+      'danger level'  | 5  | 1000 | 980
     end
 
     with_them do
       context "when there is a notification and minutes still exist", :js do
         let(:message) do
-          "Group #{group.name} has #{percent}% or less Shared Runner Pipeline minutes remaining. " \
+          "Group #{group.name} has #{percent_threshold}% or less Shared Runner Pipeline minutes remaining. " \
             "Once it runs out, no new jobs or pipelines in its projects will run."
         end
 
         before do
-          group.update!(shared_runners_minutes_limit: 20)
-          allow_any_instance_of(EE::Namespace).to receive(:shared_runners_remaining_minutes).and_return(remaining_minutes)
+          group.update!(shared_runners_minutes_limit: minutes_limit)
+          allow_any_instance_of(EE::Namespace).to receive(:shared_runners_seconds).and_return(minutes_used.minutes)
         end
 
         it 'displays a warning message on group overview page' do

ee/spec/models/ci/minutes/notification_spec.rb

@@ -38,7 +38,7 @@ RSpec.describe Ci::Minutes::Notification do
 
       context 'when at the warning level' do
         before do
-          allow(group).to receive(:shared_runners_remaining_minutes).and_return(4)
+          allow(group).to receive(:shared_runners_seconds).and_return(16.minutes)
         end
 
         describe '#show?' do
@@ -70,7 +70,7 @@ RSpec.describe Ci::Minutes::Notification do
 
       context 'when at the danger level' do
         before do
-          allow(group).to receive(:shared_runners_remaining_minutes).and_return(1)
+          allow(group).to receive(:shared_runners_seconds).and_return(19.minutes)
         end
 
         describe '#show?' do
@@ -102,7 +102,7 @@ RSpec.describe Ci::Minutes::Notification do
 
       context 'when right at the limit for notification' do
         before do
-          allow(group).to receive(:shared_runners_remaining_minutes).and_return(6)
+          allow(group).to receive(:shared_runners_seconds).and_return(14.minutes)
         end
 
         describe '#show?' do
@@ -167,7 +167,7 @@ RSpec.describe Ci::Minutes::Notification do
   shared_examples 'not eligible to see notifications' do
     before do
       group.shared_runners_minutes_limit = 10
-      allow(group).to receive(:shared_runners_remaining_minutes).and_return(2)
+      allow(group).to receive(:shared_runners_seconds).and_return(8.minutes)
     end
 
     context 'when not permitted to see notifications' do

ee/spec/models/ci/minutes/quota_spec.rb

@@ -275,4 +275,27 @@ RSpec.describe Ci::Minutes::Quota do
       it { is_expected.to eq(expected_minutes) }
     end
   end
+
+  describe '#percent_total_minutes_remaining' do
+    subject { quota.percent_total_minutes_remaining }
+
+    where(:total_minutes_used, :monthly_minutes, :purchased_minutes, :result) do
+      0   | 0   | 0 | 0
+      10  | 0   | 0 | 0
+      0   | 70  | 30 | 100
+      60  | 70  | 30 | 40
+      100 | 70  | 30 | 0
+      120 | 70  | 30 | 0
+    end
+
+    with_them do
+      before do
+        allow(namespace).to receive(:shared_runners_seconds).and_return(total_minutes_used * 60)
+        allow(namespace).to receive(:shared_runners_minutes_limit).and_return(monthly_minutes)
+        allow(namespace).to receive(:extra_shared_runners_minutes_limit).and_return(purchased_minutes)
+      end
+
+      it { is_expected.to eq(result) }
+    end
+  end
 end

ee/spec/models/ee/namespace_spec.rb

@@ -770,73 +770,6 @@ RSpec.describe Namespace do
     end
   end
 
-  describe '#shared_runners_remaining_minutes_percent' do
-    let(:namespace) { build(:namespace) }
-
-    subject { namespace.shared_runners_remaining_minutes_percent }
-
-    it 'returns the minutes left as a percent of the limit' do
-      stub_minutes_used_and_limit(8, 10)
-
-      expect(subject).to eq(20)
-    end
-
-    it 'returns 100 when minutes used are 0' do
-      stub_minutes_used_and_limit(0, 10)
-
-      expect(subject).to eq(100)
-    end
-
-    it 'returns 0 when the limit is 0' do
-      stub_minutes_used_and_limit(0, 0)
-
-      expect(subject).to eq(0)
-    end
-
-    it 'returns 0 when the limit is nil' do
-      stub_minutes_used_and_limit(nil, nil)
-
-      expect(subject).to eq(0)
-    end
-
-    it 'returns 0 when minutes used are over the limit' do
-      stub_minutes_used_and_limit(11, 10)
-
-      expect(subject).to eq(0)
-    end
-
-    it 'returns 0 when minutes used are equal to the limit' do
-      stub_minutes_used_and_limit(10, 10)
-
-      expect(subject).to eq(0)
-    end
-
-    def stub_minutes_used_and_limit(minutes_used, limit)
-      seconds_used = minutes_used.present? ? minutes_used * 60 : minutes_used
-      allow(namespace).to receive(:shared_runners_seconds).and_return(seconds_used)
-
-      allow(namespace).to receive(:actual_shared_runners_minutes_limit).and_return(limit)
-    end
-  end
-
-  describe '#shared_runners_remaining_minutes_below_threshold?' do
-    let(:namespace) { build(:namespace, last_ci_minutes_usage_notification_level: 30) }
-
-    subject { namespace.shared_runners_remaining_minutes_below_threshold? }
-
-    it 'is true when minutes left is below the notification level' do
-      allow(namespace).to receive(:shared_runners_remaining_minutes_percent).and_return(10)
-
-      expect(subject).to be_truthy
-    end
-
-    it 'is false when minutes left is not below the notification level' do
-      allow(namespace).to receive(:shared_runners_remaining_minutes_percent).and_return(80)
-
-      expect(subject).to be_falsey
-    end
-  end
-
   describe '#actual_plan' do
     context 'when namespace does not have a subscription associated' do
       it 'generates a subscription and returns default plan' do

ee/spec/models/project_spec.rb

@@ -16,7 +16,6 @@ RSpec.describe Project do
 
     it { is_expected.to delegate_method(:actual_shared_runners_minutes_limit).to(:shared_runners_limit_namespace) }
     it { is_expected.to delegate_method(:shared_runners_minutes_limit_enabled?).to(:shared_runners_limit_namespace) }
-    it { is_expected.to delegate_method(:shared_runners_remaining_minutes_below_threshold?).to(:shared_runners_limit_namespace) }
 
     it { is_expected.to delegate_method(:closest_gitlab_subscription).to(:namespace) }