ENV['SKIP_RAILS'] = 'true'
require File.expand_path(File.join(File.dirname(__FILE__),'..', '..', 'test_helper'))

class NewRelic::Agent::StatsTest < Test::Unit::TestCase
  def mock_plusequals(first, second, method, first_value, second_value)
    first.expects(method).returns(first_value)
    second.expects(method).returns(second_value)
    first.expects("#{method}=".to_sym).with(first_value + second_value)
  end

  def test_update_totals
    attrs = [:total_call_time, :total_exclusive_time, :sum_of_squares]
    merged = setup_merge do |a, b|
      attrs.each do |attr|
        a.send("#{attr}=", 2)
        b.send("#{attr}=", 3)
      end
    end
    attrs.each do |attr|
      assert_equal(5, merged.send(attr))
    end
  end

  def setup_merge
    a = NewRelic::Agent::Stats.new
    b = NewRelic::Agent::Stats.new
    a.reset
    b.reset
    yield a, b
    a.merge(b)
  end

  def test_merge_expands_min_max_call_time
    merged = setup_merge do |a, b|
      a.call_count = 1
      b.call_count = 1
      a.min_call_time = 0.5
      a.max_call_time = 3.0
      b.min_call_time = 1.0
      b.max_call_time = 4.0
    end
    assert_equal(0.5, merged.min_call_time)
    assert_equal(4.0, merged.max_call_time)
  end

  def test_simple
    stats = NewRelic::Agent::Stats.new
    validate stats, 0, 0, 0, 0

    assert_equal stats.call_count,0
    stats.trace_call 10
    stats.trace_call 20
    stats.trace_call 30

    validate stats, 3, (10+20+30), 10, 30
  end

  def test_to_s
    s1 = NewRelic::Agent::Stats.new
    s1.trace_call 10
    assert_equal("[ 1 calls 10.0000s]", s1.to_s)
  end

  def test_apdex_recording
    s = NewRelic::Agent::Stats.new

    s.record_apdex_s
    s.record_apdex_t

    s.record_apdex_f
    s.record_apdex_t

    assert_equal(1, s.apdex_s)
    assert_equal(1, s.apdex_f)
    assert_equal(2, s.apdex_t)
  end

  def test_merge
    s1 = NewRelic::Agent::Stats.new
    s2 = NewRelic::Agent::Stats.new

    s1.trace_call 10
    s2.trace_call 20
    s2.freeze

    validate s2, 1, 20, 20, 20
    s3 = s1.merge s2
    validate s3, 2, (10+20), 10, 20
    validate s1, 1, 10, 10, 10
    validate s2, 1, 20, 20, 20

    s1.merge! s2
    validate s1, 2, (10+20), 10, 20
    validate s2, 1, 20, 20, 20
  end

  def test_merge_with_exclusive
    s1 = NewRelic::Agent::Stats.new

    s2 = NewRelic::Agent::Stats.new

    s1.trace_call 10, 5
    s2.trace_call 20, 10
    s2.freeze

    validate s2, 1, 20, 20, 20, 10
    s3 = s1.merge s2
    validate s3, 2, (10+20), 10, 20, (10+5)
    validate s1, 1, 10, 10, 10, 5
    validate s2, 1, 20, 20, 20, 10

    s1.merge! s2
    validate s1, 2, (10+20), 10, 20, (5+10)
    validate s2, 1, 20, 20, 20, 10
  end

  def test_merge_array
    s1 = NewRelic::Agent::Stats.new
    merges = []
    merges << (NewRelic::Agent::Stats.new.trace_call 1)
    merges << (NewRelic::Agent::Stats.new.trace_call 1)
    merges << (NewRelic::Agent::Stats.new.trace_call 1)

    s1.merge! merges
    validate s1, 3, 3, 1, 1
  end

  def test_freeze
    s1 = NewRelic::Agent::Stats.new

    s1.trace_call 10
    s1.freeze

    begin
      # the following should throw an exception because s1 is frozen
      s1.trace_call 20
      assert false
    rescue StandardError
      assert s1.frozen?
      validate s1, 1, 10, 10, 10
    end
  end

  def test_sum_of_squares_merge
    s1 = NewRelic::Agent::Stats.new
    s1.trace_call 4
    s1.trace_call 7

    s2 = NewRelic::Agent::Stats.new
    s2.trace_call 13
    s2.trace_call 16

    s3 = s1.merge(s2)

    assert_equal(s1.sum_of_squares, 4*4 + 7*7)
    assert_equal(s3.sum_of_squares, 4*4 + 7*7 + 13*13 + 16*16, "check sum of squares")
  end

  def test_chained_stats_proxies_calls_to_scoped_and_unscoped
    scoped_stats = NewRelic::Agent::Stats.new
    unscoped_stats = NewRelic::Agent::Stats.new
    chained_stats = NewRelic::Agent::ChainedStats.new(scoped_stats, unscoped_stats)
    chained_stats.record_data_point(42)
    assert_equal(1, scoped_stats.call_count)
    assert_equal(1, unscoped_stats.call_count)
    assert_equal(42, scoped_stats.total_call_time)
    assert_equal(42, unscoped_stats.total_call_time)
  end

  def test_chained_stats_returns_values_from_scoped
    scoped_stats = NewRelic::Agent::Stats.new
    unscoped_stats = NewRelic::Agent::Stats.new
    chained_stats = NewRelic::Agent::ChainedStats.new(scoped_stats, unscoped_stats)
    scoped_stats.record_data_point(42)
    assert_equal(1, chained_stats.call_count)
    assert_equal(42, chained_stats.total_call_time)
  end

  if RUBY_VERSION >= '1.9'
    def test_to_json_enforces_float_values
      s1 = NewRelic::Agent::Stats.new
      s1.trace_call 3.to_r
      s1.trace_call 7.to_r

      assert_equal 3.0, JSON.load(s1.to_json)['min_call_time']
    end
  end

  private
  def validate (stats, count, total, min, max, exclusive = nil)
    assert_equal count, stats.call_count
    assert_equal total, stats.total_call_time
    assert_equal min,   stats.min_call_time
    assert_equal max,   stats.max_call_time
    assert_equal exclusive, stats.total_exclusive_time if exclusive
  end
end