Merge branch 'rs-early-access-program-group-toggle' into 'master'

Early access program tracking by namespace settings

See merge request https://gitlab.com/gitlab-org/gitlab/-/merge_requests/153170



Merged-by: Pavel Shutsin <pshutsin@gitlab.com>
Approved-by: Aishwarya Subramanian <asubramanian@gitlab.com>
Approved-by: Abdul Wadood <awadood@gitlab.com>
Approved-by: Pavel Shutsin <pshutsin@gitlab.com>
Reviewed-by: Pavel Shutsin <pshutsin@gitlab.com>
Reviewed-by: Huzaifa Iftikhar <hiftikhar@gitlab.com>
Reviewed-by: Abdul Wadood <awadood@gitlab.com>
Reviewed-by: Aishwarya Subramanian <asubramanian@gitlab.com>
Co-authored-by: Rostyslav Safonov <elhowm@gmail.com>

db/migrate/20240515181541_add_early_access_program_participant_to_namespace_settings.rb

+# frozen_string_literal: true
+
+class AddEarlyAccessProgramParticipantToNamespaceSettings < Gitlab::Database::Migration[2.2]
+  milestone '17.1'
+
+  def change
+    add_column :namespace_settings, :early_access_program_participant, :boolean, null: false, default: false
+  end
+end

db/schema_migrations/20240515181541

+209ef8ffed9a59acdd813b3feaa259231ec812733e58341dedd0a8b941d8a658
\ No newline at end of file

db/structure.sql

@@ -12281,6 +12281,7 @@ CREATE TABLE namespace_settings (
     lock_duo_features_enabled boolean DEFAULT false NOT NULL,
     disable_personal_access_tokens boolean DEFAULT false NOT NULL,
     enable_auto_assign_gitlab_duo_pro_seats boolean DEFAULT false NOT NULL,
+    early_access_program_participant boolean DEFAULT false NOT NULL,
     remove_dormant_members boolean DEFAULT false NOT NULL,
     remove_dormant_members_period integer DEFAULT 90 NOT NULL,
     CONSTRAINT check_0ba93c78c7 CHECK ((char_length(default_branch_name) <= 255)),

lib/gitlab/internal_events.rb

@@ -40,7 +40,7 @@ def track_event(
         send_application_instrumentation_event(event_name, additional_properties, kwargs) if send_snowplow_event
 
         if Feature.enabled?(:early_access_program, kwargs[:user], type: :wip)
-          create_early_access_program_event(event_name, category, additional_properties[:label], kwargs[:user])
+          create_early_access_program_event(event_name, category, additional_properties[:label], kwargs)
         end
       rescue StandardError => e
         extra = {}
@@ -189,8 +189,9 @@ def send_application_instrumentation_event(event_name, additional_properties, kw
         gitlab_sdk_client.track(event_name, tracked_attributes)
       end
 
-      def create_early_access_program_event(event_name, category, event_label, user)
-        return if user.nil? || !user.user_preference.early_access_event_tracking?
+      def create_early_access_program_event(event_name, category, event_label, kwargs)
+        user, namespace = kwargs.values_at(:user, :namespace)
+        return if user.nil? || !namespace&.namespace_settings&.early_access_program_participant?
 
         ::EarlyAccessProgram::TrackingEvent.create(
           user: user, event_name: event_name.to_s, event_label: event_label, category: category

spec/lib/gitlab/internal_events_spec.rb

@@ -591,6 +591,16 @@ def validate_service_ping_context(service_ping_context)
     end
 
     context 'with early access program tracking' do
+      let(:namespace_participating) { false }
+      let(:namespace) do
+        settings = create(:namespace_settings, early_access_program_participant: namespace_participating)
+        create(:namespace, namespace_settings: settings)
+      end
+
+      let(:event_kwargs) do
+        { user: user, project: project, send_snowplow_event: send_snowplow_event, namespace: namespace }
+      end
+
       shared_examples 'does not create early access program tracking event' do
         it do
           track_event
@@ -601,7 +611,7 @@ def validate_service_ping_context(service_ping_context)
 
       before do
         allow(sdk_client).to receive(:track)
-          .with(event_name, { project_id: project.id, namespace_id: project.namespace.id })
+          .with(event_name, { project_id: project&.id, namespace_id: namespace&.id })
       end
 
       context 'when early_access_program FF is enabled' do
@@ -615,25 +625,31 @@ def validate_service_ping_context(service_ping_context)
           it_behaves_like 'does not create early access program tracking event'
         end
 
+        context 'without namespace' do
+          let(:project) { nil }
+          let(:namespace) { nil }
+
+          it_behaves_like 'does not create early access program tracking event'
+        end
+
         context 'with user' do
-          context 'when user is not early access program participant' do
+          context 'when namespace is not early access program participant' do
             it_behaves_like 'does not create early access program tracking event'
           end
 
-          context 'when user is early access program participant' do
+          context 'when namespace is early access program participant' do
+            let(:namespace_participating) { true }
             let(:event_name) { 'g_edit_by_snippet_ide' }
             let(:additional_properties) { { label: 'label_name' } }
             let(:user) { create(:user) }
 
             before do
-              preference_stub = instance_double(UserPreference, early_access_event_tracking?: true)
-              allow(user).to receive(:user_preference).and_return(preference_stub)
               allow(sdk_client).to receive(:track)
                 .with(
                   event_name,
                   {
                     project_id: project.id,
-                    namespace_id: project.namespace.id,
+                    namespace_id: namespace.id,
                     additional_properties: additional_properties
                   }
                 )