# encoding: utf-8
# This file is distributed under New Relic's license terms.
# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.

require './app'

# These tests only return consistent results for REE or MRI >= 1.9.2
if NewRelic::LanguageSupport.ree? ||
    (RUBY_VERSION >= '1.9.2' &&
     !NewRelic::LanguageSupport.jruby? &&
     !NewRelic::LanguageSupport.rubinius?)

class GcController < ApplicationController
  def gc_action
    begin
      NewRelic::Agent::StatsEngine::GCProfiler.init
      initial_gc_count = current_gc_count

      Timeout.timeout(5) do
        until current_gc_count > initial_gc_count
          long_string = "01234567" * 100_000
          long_string = nil
          another_long_string = "01234567" * 100_000
          another_long_string = nil
        end
      end
    rescue Timeout::Error
      puts "Timed out waiting for GC..."
    end

    render :text => 'ha'
  end

  def current_gc_count
    if NewRelic::LanguageSupport.ree?
      ::GC.collections
    elsif RUBY_VERSION >= '1.9.2'
      ::GC.count
    end
  end
end

class GCRailsInstrumentationTest < ActionController::TestCase
  tests GcController

  include MultiverseHelpers

  setup_and_teardown_agent do
    NewRelic::Agent.drop_buffered_data
    NewRelic::Agent::StatsEngine::GCProfiler.reset
    enable_gc_stats
    @controller = GcController.new
  end

  def test_records_accurate_time_for_gc_activity
    start = Time.now
    get :gc_action
    elapsed = Time.now.to_f - start.to_f

    assert_in_range(elapsed, get_call_time('GC/Transaction/allWeb'))
    assert_in_range(elapsed, get_call_time('GC/Transaction/allWeb', 'Controller/gc/gc_action'))
  end

  def test_records_transaction_param_for_gc_activity
    start = Time.now.to_f
    get :gc_action
    elapsed = Time.now.to_f - start

    trace = NewRelic::Agent.instance.transaction_sampler.last_sample
    assert_in_range(elapsed, trace.params[:custom_params][:gc_time])
  end

  def assert_in_range(duration, gc_time)
    assert gc_time > 0.0, "GC Time wasn't recorded!"
    assert gc_time < duration, "GC Time #{gc_time} can't be more than elapsed #{duration}!"
  end

  def get_call_time(name, scope=nil)
    NewRelic::Agent.agent.stats_engine.
      get_stats(name, true, false, scope).
      total_call_time
  end

  def enable_gc_stats
    if NewRelic::LanguageSupport.ree?
      GC.enable_stats
    elsif RUBY_VERSION >= '1.9.2'
      GC::Profiler.enable
    end
  end
end

end