Remove obsolete License classes

ee/config/metrics/license/20210204124831_license_md5.yml

@@ -11,8 +11,7 @@ time_frame: none
 data_source: license
 instrumentation_class: LicenseMetric
 options:
-  attribute:
-    - md5
+  attribute: md5
 data_category: standard
 distribution:
   - ee

ee/config/metrics/license/20210204124833_license_id.yml

@@ -12,8 +12,7 @@ data_source: license
 data_category: subscription
 instrumentation_class: LicenseMetric
 options:
-  attribute:
-    - id
+  attribute: id
 distribution:
 - ee
 tier:

ee/config/metrics/license/20210204124843_license_user_count.yml

@@ -12,8 +12,7 @@ data_source: license
 data_category: subscription
 instrumentation_class: LicenseMetric
 options:
-  attribute:
-    user_count
+  attribute: user_count
 distribution:
 - ee
 tier:

ee/config/metrics/license/20210204124845_license_starts_at.yml

@@ -12,8 +12,7 @@ data_source: license
 data_category: subscription
 instrumentation_class: LicenseMetric
 options:
-  attribute:
-    - starts_at
+  attribute: starts_at
 distribution:
 - ee
 tier:

ee/config/metrics/license/20210204124847_license_expires_at.yml

@@ -12,8 +12,7 @@ time_frame: none
 data_source: license
 instrumentation_class: LicenseMetric
 options:
-  attribute:
-    - expires_at
+  attribute: expires_at
 distribution:
 - ee
 tier:

ee/config/metrics/license/20210204124849_license_plan.yml

@@ -12,8 +12,7 @@ data_source: license
 data_category: subscription
 instrumentation_class: LicenseMetric
 options:
-  attribute:
-    - plan
+  attribute: plan
 distribution:
 - ee
 tier:

ee/config/metrics/license/20210204124851_license_trial.yml

@@ -12,8 +12,7 @@ data_source: license
 data_category: subscription
 instrumentation_class: LicenseMetric
 options:
-  attribute:
-    - trial
+  attribute: trial
 distribution:
 - ee
 tier:

ee/config/metrics/license/20210204124852_license_subscription_id.yml

@@ -11,8 +11,7 @@ time_frame: none
 data_source: license
 instrumentation_class: LicenseMetric
 options:
-  attribute:
-    - subsription_id
+  attribute: subscription_id
 data_category: standard
 distribution:
 - ee

ee/config/metrics/license/20210204124926_license_trial_ends_on.yml

@@ -12,8 +12,7 @@ data_source: database
 data_category: subscription
 instrumentation_class: LicenseMetric
 options:
-  attribute:
-    - trial_ends_on
+  attribute: trial_ends_on
 distribution:
 - ee
 tier:

ee/lib/ee/gitlab/usage_data.rb

@@ -72,7 +72,7 @@ module EE
         def features_usage_data_ee
           {
             elasticsearch_enabled: alt_usage_data(fallback: nil) { ::Gitlab::CurrentSettings.elasticsearch_search? },
-            license_trial_ends_on: alt_usage_data(fallback: nil) { add_metric("LicenseMetric", options: { attribute: :trial_ends_on }) },
+            license_trial_ends_on: add_metric("LicenseMetric", options: { attribute: "trial_ends_on" }),
             geo_enabled: alt_usage_data(fallback: nil) { ::Gitlab::Geo.enabled? },
             user_cap_feature_enabled: add_metric('UserCapSettingEnabledMetric', time_frame: 'none')
           }
@@ -90,7 +90,7 @@ module EE
             end
 
           if license
-            usage_data[:license_md5] = add_metric("LicenseMdFiveMetric")
+            usage_data[:license_md5] = add_metric("LicenseMetric", options: { attribute: 'md5' })
             usage_data[:license_id] = license.license_id
             # rubocop: disable UsageData/LargeTable
             usage_data[:historical_max_users] = add_metric("HistoricalMaxUsersMetric")
@@ -103,7 +103,7 @@ module EE
             usage_data[:license_plan] = license.plan
             usage_data[:license_add_ons] = license.add_ons
             usage_data[:license_trial] = license.trial?
-            usage_data[:license_subscription_id] = alt_usage_data(fallback: nil) { license.subscription_id }
+            usage_data[:license_subscription_id] = license.subscription_id
           end
 
           usage_data

ee/lib/gitlab/usage/metrics/instrumentations/license_md_five_metric.rb

-# frozen_string_literal: true
-
-module Gitlab
-  module Usage
-    module Metrics
-      module Instrumentations
-        class LicenseMdFiveMetric < ::Gitlab::Usage::Metrics::Instrumentations::GenericMetric
-          value do
-            ::License.current.md5
-          end
-        end
-      end
-    end
-  end
-end

ee/lib/gitlab/usage/metrics/instrumentations/license_metric.rb

@@ -13,7 +13,7 @@ module Gitlab
           #   attribute: md5
           # end
 
-          ALLOWED_ATTRIBUTES = %i(md5
+          ALLOWED_ATTRIBUTES = %w(md5
                                   id
                                   plan
                                   trial
@@ -27,18 +27,23 @@ module Gitlab
             super
 
             raise ArgumentError, "License options attribute are required" unless license_attribute.present?
-            raise ArgumentError, "attribute should one allowed" unless license_attribute.in?(ALLOWED_ATTRIBUTES)
-          end
-
-          def license_attribute
-            options[:attribute]
+            raise ArgumentError, "Attribute: #{license_attribute} it not allowed" unless license_attribute.in?(ALLOWED_ATTRIBUTES)
           end
 
           def value
-            alt_usage_data(fallback: -1) do
-              ::Licese.current.send(license_attribute)
+            return ::License.trial_ends_on if license_attribute == "trial_ends_on"
+
+            alt_usage_data(fallback: nil) do
+              # license_attribute is checked in the constructor, so it's safe
+              ::License.current.send(license_attribute)  # rubocop: disable GitlabSecurity/PublicSend
             end
           end
+
+          private
+
+          def license_attribute
+            options[:attribute]
+          end
         end
       end
     end

ee/lib/gitlab/usage/metrics/instrumentations/zuora_subscription_id_metric.rb

-# frozen_string_literal: true
-
-module Gitlab
-  module Usage
-    module Metrics
-      module Instrumentations
-        class ZuoraSubscriptionIdMetric < ::Gitlab::Usage::Metrics::Instrumentations::GenericMetric
-          value do
-            ::License.current.subscription_id
-          end
-        end
-      end
-    end
-  end
-end

ee/spec/lib/ee/gitlab/usage_data_spec.rb

@@ -187,6 +187,10 @@ RSpec.describe Gitlab::UsageData do
   describe '.features_usage_data_ee' do
     subject { described_class.features_usage_data_ee }
 
+    before do
+      stub_feature_flags(usage_data_instrumentation: false)
+    end
+
     it 'gathers feature usage data of EE' do
       expect(subject[:elasticsearch_enabled]).to eq(Gitlab::CurrentSettings.elasticsearch_search?)
       expect(subject[:geo_enabled]).to eq(Gitlab::Geo.enabled?)

ee/spec/lib/gitlab/usage/metrics/instrumentations/license_md_five_metric_spec.rb

-# frozen_string_literal: true
-
-require 'spec_helper'
-
-RSpec.describe Gitlab::Usage::Metrics::Instrumentations::LicenseMdFiveMetric do
-  it_behaves_like 'a correct instrumented metric value', { time_frame: 'none', data_source: 'ruby' } do
-    let(:expected_value) { Digest::MD5.hexdigest(::License.current.data) }
-  end
-end

ee/spec/lib/gitlab/usage/metrics/instrumentations/zuora_subscription_id_metric_spec.rb

-# frozen_string_literal: true
-
-require 'spec_helper'
-
-RSpec.describe Gitlab::Usage::Metrics::Instrumentations::ZuoraSubscriptionIdMetric do
-  it_behaves_like 'a correct instrumented metric value', { time_frame: 'none', data_source: 'ruby' } do
-    let(:expected_value) { ::License.current.subscription_id }
-  end
-end

lib/gitlab/usage/metrics/names_suggestions/generator.rb

@@ -10,10 +10,10 @@ module Gitlab
               uncached_data.deep_stringify_keys.dig(*key_path.split('.'))
             end
 
-            def add_metric(metric, time_frame: 'none')
+            def add_metric(metric, time_frame: 'none', options: {})
               metric_class = "Gitlab::Usage::Metrics::Instrumentations::#{metric}".constantize
 
-              metric_class.new(time_frame: time_frame).suggested_name
+              metric_class.new(time_frame: time_frame, options: options).suggested_name
             end
 
             private

lib/gitlab/usage_data_non_sql_metrics.rb

@@ -12,10 +12,10 @@ module Gitlab
         super.with_indifferent_access.deep_merge(instrumentation_metrics.with_indifferent_access)
       end
 
-      def add_metric(metric, time_frame: 'none')
+      def add_metric(metric, time_frame: 'none', options: {})
         metric_class = "Gitlab::Usage::Metrics::Instrumentations::#{metric}".constantize
 
-        metric_class.new(time_frame: time_frame).instrumentation
+        metric_class.new(time_frame: time_frame, options: options).instrumentation
       end
 
       def count(relation, column = nil, batch: true, batch_size: nil, start: nil, finish: nil)

lib/gitlab/usage_data_queries.rb

@@ -12,10 +12,10 @@ module Gitlab
         super.with_indifferent_access.deep_merge(instrumentation_metrics.with_indifferent_access)
       end
 
-      def add_metric(metric, time_frame: 'none')
+      def add_metric(metric, time_frame: 'none', options: {})
         metric_class = "Gitlab::Usage::Metrics::Instrumentations::#{metric}".constantize
 
-        metric_class.new(time_frame: time_frame).instrumentation
+        metric_class.new(time_frame: time_frame, options: options).instrumentation
       end
 
       def count(relation, column = nil, *args, **kwargs)

lib/gitlab/utils/usage_data.rb

@@ -49,6 +49,7 @@ module Gitlab
         # Results of this method should be overwritten by instrumentation class values
         # -100 indicates the metric was not properly merged.
         return INSTRUMENTATION_CLASS_FALLBACK if Feature.enabled?(:usage_data_instrumentation)
+
         metric_class = "Gitlab::Usage::Metrics::Instrumentations::#{metric}".constantize
 
         metric_class.new(time_frame: time_frame, options: options).value