Enable trial onboarding flow for other sources

Ensure that the trial onboarding flow is enabled when a user is referred
from docs.gitlab.com as well as learn.gitlab.com

part of: https://gitlab.com/gitlab-org/gitlab/-/issues/349716

Changelog: changed
EE: true

ee/app/controllers/trials_controller.rb

@@ -34,7 +34,7 @@ class TrialsController < ApplicationController
 
     render(:new) && return unless @result[:success]
 
-    if params[:glm_source] == 'about.gitlab.com'
+    if EE::TrialHelper::TRIAL_ONBOARDING_SOURCE_URLS.include?(params[:glm_source])
       redirect_to(new_users_sign_up_group_path(url_params.merge(trial_onboarding_flow: true)))
     elsif @namespace = helpers.only_trialable_group_namespace
       params[:namespace_id] = @namespace.id

ee/app/helpers/ee/trial_helper.rb

@@ -2,6 +2,8 @@
 
 module EE
   module TrialHelper
+    TRIAL_ONBOARDING_SOURCE_URLS = %w(about.gitlab.com docs.gitlab.com learn.gitlab.com).freeze
+
     def company_size_options_for_select(selected = '')
       options_for_select([
         [_('Please select'), ''],
@@ -14,7 +16,7 @@ module EE
     end
 
     def should_ask_company_question?
-      glm_params[:glm_source] != 'about.gitlab.com'
+      TRIAL_ONBOARDING_SOURCE_URLS.exclude?(glm_params[:glm_source])
     end
 
     def glm_params

ee/spec/controllers/trials_controller_spec.rb

@@ -102,6 +102,22 @@ RSpec.describe TrialsController, :saas do
         end
       end
 
+      context 'coming from docs.gitlab.com' do
+        let(:post_params) { { glm_source: 'docs.gitlab.com' } }
+
+        it 'redirects to trial onboarding' do
+          is_expected.to redirect_to(new_users_sign_up_group_path(glm_source: 'docs.gitlab.com', trial_onboarding_flow: true))
+        end
+      end
+
+      context 'coming from learn.gitlab.com' do
+        let(:post_params) { { glm_source: 'learn.gitlab.com' } }
+
+        it 'redirects to trial onboarding' do
+          is_expected.to redirect_to(new_users_sign_up_group_path(glm_source: 'learn.gitlab.com', trial_onboarding_flow: true))
+        end
+      end
+
       context 'when user has 1 trial eligible namespace', :experiment do
         let_it_be(:namespace) { create(:group, path: 'namespace-test') }
 

ee/spec/helpers/ee/trial_helper_spec.rb

@@ -14,6 +14,8 @@ RSpec.describe EE::TrialHelper do
 
     where(:glm_source, :result) do
       'about.gitlab.com'  | false
+      'learn.gitlab.com'  | false
+      'docs.gitlab.com'   | false
       'abouts.gitlab.com' | true
       'about.gitlab.org'  | true
       'about.gitlob.com'  | true