gitlab-experiment 1.2.0 → 1.4.0

data/lib/gitlab/experiment/configuration.rb

@@ -44,6 +44,15 @@ module Gitlab
         "#{experiment.name}_id"
       end
 
+      # Allow forced variant assignment via query parameter and cookie.
+      #
+      # When enabled, experiments will check for a `glex_force` query
+      # parameter and a `_glex_force` cookie to override variant assignment.
+      # This is intended for QA/UAT testing in staging and production.
+      #
+      # Defaults to false (disabled).
+      @allow_forced_assignment = false
+
       # Mark experiment cookies as secure (HTTPS only).
       #
       # When set to true, cookies will have the secure flag set, meaning they
@@ -187,6 +196,7 @@ module Gitlab
           :cookie_domain,
           :cookie_name,
           :secure_cookie,
+          :allow_forced_assignment,
           :context_key_secret,
           :context_key_bit_length,
           :mount_at,

data/lib/gitlab/experiment/dsl.rb

@@ -7,10 +7,19 @@ module Gitlab
         klass.include(self).tap { |base| base.helper_method(:experiment) if with_helper }
       end
 
+      # An explicit `request:` keyword argument always takes precedence; when
+      # present, no auto-detection runs.
+      #
+      # Otherwise, `request` is auto-detected through implicit_experiment_request in this order:
+      #   1. A `request` method on `self` (the controller pattern).
+      #   2. A `@request` instance variable on `self` (the service pattern).
+      #
+      # Callers with `request` as a pure local variable should pass
+      # `request: request` explicitly.
       def experiment(name, variant_name = nil, **context, &block)
         raise ArgumentError, 'name is required' if name.nil?
 
-        context[:request] ||= request if respond_to?(:request)
+        context[:request] ||= implicit_experiment_request
 
         base = Configuration.base_class.constantize
         klass = base.constantize(name) || base
@@ -20,6 +29,16 @@ module Gitlab
 
         instance.context.frozen? ? instance.run : instance.tap(&:run)
       end
+
+      private
+
+      def implicit_experiment_request
+        if respond_to?(:request)
+          request
+        elsif instance_variable_defined?(:@request)
+          instance_variable_get(:@request)
+        end
+      end
     end
   end
 end

data/lib/gitlab/experiment/force_assignment.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module Gitlab
+  class Experiment
+    module ForceAssignment
+      PARAM_NAME = 'glex_force'
+
+      private
+
+      def forced_variant_name
+        return unless Configuration.allow_forced_assignment
+        return unless enabled?
+
+        param = context&.request&.params.try(:[], PARAM_NAME)
+        return if param.blank?
+
+        experiment_name, variant_name = param.split(':', 2)
+        return if experiment_name.blank? || variant_name.blank?
+        return unless experiment_name == name
+
+        variant_sym = variant_name.to_sym
+        return unless behaviors.key?(variant_sym)
+
+        variant_sym
+      end
+    end
+  end
+end

data/lib/gitlab/experiment/version.rb

@@ -2,6 +2,6 @@
 
 module Gitlab
   class Experiment
-    VERSION = '1.2.0'
+    VERSION = '1.4.0'
   end
 end

data/lib/gitlab/experiment.rb

@@ -13,6 +13,7 @@ require 'gitlab/experiment/callbacks'
 require 'gitlab/experiment/rollout'
 require 'gitlab/experiment/configuration'
 require 'gitlab/experiment/cookies'
+require 'gitlab/experiment/force_assignment'
 require 'gitlab/experiment/context'
 require 'gitlab/experiment/dsl'
 require 'gitlab/experiment/middleware'
@@ -26,6 +27,7 @@ module Gitlab
     include BaseInterface
     include Cache
     include Callbacks
+    include ForceAssignment
     include Nestable
 
     class << self
@@ -107,6 +109,8 @@ module Gitlab
     end
 
     def assigned(value = nil)
+      # Skip force assignment if a variant was already set (e.g., via constructor or explicit #assigned call).
+      value ||= forced_variant_name unless @_assigned_variant_name
       @_assigned_variant_name = cache_variant(value) if value.present?
       return Variant.new(name: @_assigned_variant_name || :unresolved) if @_assigned_variant_name || @_resolving_variant
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gitlab-experiment
 version: !ruby/object:Gem::Version
-  version: 1.2.0
+  version: 1.4.0
 platform: ruby
 authors:
 - GitLab
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-12-16 00:00:00.000000000 Z
+date: 2026-05-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -223,6 +223,7 @@ files:
 - lib/gitlab/experiment/dsl.rb
 - lib/gitlab/experiment/engine.rb
 - lib/gitlab/experiment/errors.rb
+- lib/gitlab/experiment/force_assignment.rb
 - lib/gitlab/experiment/middleware.rb
 - lib/gitlab/experiment/nestable.rb
 - lib/gitlab/experiment/rollout.rb