mock_adapter.rb

module Vanity
  module Adapters
    class << self
      # Creates and returns new MockAdapter.
      #
      # @since 1.4.0
      def mock_connection(spec)
        MockAdapter.new(spec)
      end
    end

    # Mock adapter. You can use this when running in test environment, staging,
    # wherever you don't care for collecting metrics. Doesn't store anything.
    #
    # @since 1.4.0
    class MockAdapter < AbstractAdapter
      def initialize(options)
        @metrics = @@metrics ||= {}
        @experiments = @@experiments ||= {}
      end

      def active?
        !!@metrics
      end

      def disconnect!
        @metrics = nil
        @experiments = nil
      end

      def reconnect!
        @metrics = @@metrics
        @experiments = @@experiments
      end

      def to_s
        "mock:/"
      end

      def flushdb
        @metrics.clear
        @experiments.clear
      end


      # -- Metrics --

      def get_metric_last_update_at(metric)
        @metrics[metric] && @metrics[metric][:last_update_at]
      end

      def metric_track(metric, timestamp, identity, values)
        @metrics[metric] ||= {}
        current = @metrics[metric][timestamp.to_date] ||= []
        values.each_with_index do |v,i|
          current[i] = (current[i] || 0) + v || 0
        end
        @metrics[metric][:last_update_at] = Time.now
      end

      def metric_values(metric, from, to)
        hash = @metrics[metric] || {}
        (from.to_date..to.to_date).map { |date| hash[date] || [] }
      end

      def destroy_metric(metric)
        @metrics.delete metric
      end


      # -- Experiments --

      def experiment_persisted?(experiment)
        !!@experiments[experiment]
      end

      def set_experiment_created_at(experiment, time)
        @experiments[experiment] ||= {}
        @experiments[experiment][:created_at] ||= time
      end

      def get_experiment_created_at(experiment)
        @experiments[experiment] && @experiments[experiment][:created_at]
      end

      def set_experiment_completed_at(experiment, time)
        @experiments[experiment] ||= {}
        @experiments[experiment][:completed_at] ||= time
      end

      def get_experiment_completed_at(experiment)
        @experiments[experiment] && @experiments[experiment][:completed_at]
      end

      def is_experiment_completed?(experiment)
        @experiments[experiment] && @experiments[experiment][:completed_at]
      end

      def set_experiment_enabled(experiment, enabled)
        @experiments[experiment] ||= {}
        @experiments[experiment][:enabled] = enabled
      end

      def is_experiment_enabled?(experiment)
        record = @experiments[experiment]
        if Vanity.configuration.experiments_start_enabled
          record == nil || record[:enabled] != false
        else
          record && record[:enabled] == true
        end
      end

      def ab_counts(experiment, alternative)
        alt = alternative(experiment, alternative)
        { :participants => alt[:participants] ? alt[:participants].size : 0,
          :converted    => alt[:converted] ? alt[:converted].size : 0,
          :conversions  => alt[:conversions] || 0 }
      end

      def ab_show(experiment, identity, alternative)
        @experiments[experiment] ||= {}
        @experiments[experiment][:showing] ||= {}
        @experiments[experiment][:showing][identity] = alternative
      end

      def ab_showing(experiment, identity)
        @experiments[experiment] && @experiments[experiment][:showing] && @experiments[experiment][:showing][identity]
      end

      def ab_not_showing(experiment, identity)
        @experiments[experiment][:showing].delete identity if @experiments[experiment] && @experiments[experiment][:showing]
      end

      def ab_add_participant(experiment, alternative, identity)
        alt = alternative(experiment, alternative)
        alt[:participants] ||= Set.new
        alt[:participants] << identity
      end

      def ab_seen(experiment, identity, alternative_or_id)
        with_ab_seen_deprecation(experiment, identity, alternative_or_id) do |expt, ident, alt_id|
          if ab_assigned(expt, ident) == alt_id
            alt_id
          end
        end
      end

      def ab_assigned(experiment, identity)
        alternatives_for(experiment).each do |alt_id, alt_state|
          return alt_id if alt_state[:participants].include?(identity)
        end

        nil
      end

      def ab_add_conversion(experiment, alternative, identity, count = 1, implicit = false)
        alt = alternative(experiment, alternative)
        alt[:participants] ||= Set.new
        alt[:converted] ||= Set.new
        alt[:conversions] ||= 0
        if implicit
          alt[:participants] << identity
        else
    participating = alt[:participants].include?(identity)
        end
        alt[:converted] << identity if implicit || participating
        alt[:conversions] += count
      end

      def ab_get_outcome(experiment)
        @experiments[experiment] ||= {}
        @experiments[experiment][:outcome]
      end

      def ab_set_outcome(experiment, alternative = 0)
        @experiments[experiment] ||= {}
        @experiments[experiment][:outcome] = alternative
      end

      def destroy_experiment(experiment)
        @experiments.delete experiment
      end

      private

      def alternative(experiment, alternative)
        alternatives_for(experiment)[alternative] ||= {}
        alternatives_for(experiment)[alternative]
      end

      def alternatives_for(experiment)
        @experiments[experiment] ||= {}
        @experiments[experiment][:alternatives] ||= {}

        @experiments[experiment][:alternatives]
      end
    end
  end
end