require File.expand_path(File.join(File.dirname(__FILE__),'..','..','..','..','test_helper'))
class NewRelic::Agent::MethodTracer::InstanceMethods::TraceExecutionScopedTest < Test::Unit::TestCase
  require 'new_relic/agent/method_tracer'
  include NewRelic::Agent::MethodTracer::InstanceMethods::TraceExecutionScoped

  def test_trace_disabled_negative
    self.expects(:traced?).returns(false)
    options = {:force => false}
    assert trace_disabled?(options)
  end

  def test_trace_disabled_forced
    self.expects(:traced?).returns(false)
    options = {:force => true}
    assert !(trace_disabled?(options))
  end

  def test_trace_disabled_positive
    self.expects(:traced?).returns(true)
    options = {:force => false}
    assert !(trace_disabled?(options))
  end

  def test_get_stats_unscoped
    fake_engine = mocked_object('stat_engine')
    fake_engine.expects(:get_stats_no_scope).with('foob').returns('fakestats')
    assert_equal 'fakestats', get_stats_unscoped('foob')
  end

  def test_get_stats_scoped_scoped_only
    fake_engine = mocked_object('stat_engine')
    fake_engine.expects(:get_stats).with('foob', true, true).returns('fakestats')
    assert_equal 'fakestats', get_stats_scoped('foob', true)
  end

  def test_get_stats_scoped_no_scoped_only
    fake_engine = mocked_object('stat_engine')
    fake_engine.expects(:get_stats).with('foob', true, false).returns('fakestats')
    assert_equal 'fakestats', get_stats_scoped('foob', false)
  end

  def test_stat_engine
    assert_equal agent_instance.stats_engine, stat_engine
  end

  def test_agent_instance
    assert_equal NewRelic::Agent.instance, agent_instance
  end

  def test_main_stat
    self.expects(:get_stats_scoped).with('hello', true)
    opts = {:scoped_metric_only => true}
    main_stat('hello', opts)
  end

  def test_get_metric_stats_metric
    metrics = ['foo', 'bar', 'baz']
    opts = {:metric => true}
    self.expects(:get_stats_unscoped).twice
    self.expects(:main_stat).with('foo', opts)
    first_name, stats = get_metric_stats(metrics, opts)
    assert_equal 'foo', first_name
    assert_equal 3, stats.length
  end

  def test_get_metric_stats_no_metric
    metrics = ['foo', 'bar', 'baz']
    opts = {:metric => false}
    self.expects(:get_stats_unscoped).twice
    first_name, stats = get_metric_stats(metrics, opts)
    assert_equal 'foo', first_name
    assert_equal 2, stats.length
  end

  def test_set_if_nil
    h = {}
    set_if_nil(h, :foo)
    assert h[:foo]
    h[:bar] = false
    set_if_nil(h, :bar)
    assert !h[:bar]
  end

  def test_push_flag_true
    fake_agent = mocked_object('agent_instance')
    fake_agent.expects(:push_trace_execution_flag).with(true)
    push_flag!(true)
  end

  def test_push_flag_false
    self.expects(:agent_instance).never
    push_flag!(false)
  end

  def test_pop_flag_true
    fake_agent = mocked_object('agent_instance')
    fake_agent.expects(:pop_trace_execution_flag)
    pop_flag!(true)
  end

  def test_pop_flag_false
    self.expects(:agent_instance).never
    pop_flag!(false)
  end

  def test_log_errors_base
    ran = false
    log_errors("name", "metric") do
      ran = true
    end
    assert ran, "should run the contents of the block"
  end

  def test_log_errors_with_return
    ran = false
    return_val = log_errors('name', 'metric') do
      ran = true
      'happy trees'
    end

    assert ran, "should run contents of block"
    assert_equal 'happy trees', return_val, "should return contents of the block"
  end

  def test_log_errors_with_error
    expects_logging(:error, 
      includes("Caught exception in name. Metric name = metric"),
      instance_of(RuntimeError))

    log_errors("name", "metric") do
      raise "should not propagate out of block"
    end
  end

  def test_trace_execution_scoped_header
    options = {:force => false, :deduct_call_time_from_parent => false}
    self.expects(:log_errors).with('trace_execution_scoped header', 'foo').yields
    self.expects(:push_flag!).with(false)
    fakestats = mocked_object('stat_engine')
    fakestats.expects(:push_scope).with('foo', 1.0, false)
    trace_execution_scoped_header('foo', options, 1.0)
  end

  def test_trace_execution_scoped_footer
    t0 = 1.0
    t1 = 2.0
    metric = 'foo'
    metric_stats = [mock('fakestat')]
    metric_stats.first.expects(:trace_call).with(1.0, 0.5)
    expected_scope = 'an expected scope'
    engine = mocked_object('stat_engine')
    scope = mock('scope')
    engine.expects(:pop_scope).with('an expected scope', 1.0, 2.0).returns(scope)
    scope.expects(:children_time).returns(0.5)
    self.expects(:pop_flag!).with(false)
    self.expects(:log_errors).with('trace_method_execution footer', 'foo').yields

    trace_execution_scoped_footer(t0, metric, metric_stats, expected_scope, false, t1)
  end

  def test_trace_execution_scoped_disabled
    self.expects(:trace_disabled?).returns(true)
    # make sure the method doesn't beyond the abort
    self.expects(:set_if_nil).never
    ran = false
    value = trace_execution_scoped(nil, {:options => 'hash'}) do
      ran = true
      1172
    end

    assert ran, 'should run contents of block'
    assert_equal 1172, value, 'should return contents of block'
  end

  def test_trace_execution_scoped_default
    passed_in_opts = {}
    opts_after_correction = {:metric => true, :deduct_call_time_from_parent => true}
    self.expects(:trace_disabled?).returns(false)
    self.expects(:get_metric_stats).with(['metric', 'array'], opts_after_correction).returns(['metric', ['stats']])
    self.expects(:trace_execution_scoped_header).with('metric', opts_after_correction).returns(['start_time', 'expected_scope'])
    self.expects(:trace_execution_scoped_footer).with('start_time', 'metric', ['stats'], 'expected_scope', nil)
    ran = false
    value = trace_execution_scoped(['metric', 'array'], passed_in_opts) do
      ran = true
      1172
    end

    assert ran, 'should run contents of the block'
    assert_equal 1172, value, 'should return the contents of the block'
  end

  def test_trace_execution_scoped_with_error
    passed_in_opts = {}
    opts_after_correction = {:metric => true, :deduct_call_time_from_parent => true}
    self.expects(:trace_disabled?).returns(false)
    self.expects(:get_metric_stats).with(['metric', 'array'], opts_after_correction).returns(['metric', ['stats']])
    self.expects(:trace_execution_scoped_header).with('metric', opts_after_correction).returns(['start_time', 'expected_scope'])
    self.expects(:trace_execution_scoped_footer).with('start_time', 'metric', ['stats'], 'expected_scope', nil)
    ran = false
    assert_raises(RuntimeError) do
      trace_execution_scoped(['metric', 'array'], passed_in_opts) do
        ran = true
        raise 'raising a test error'
      end
    end

    assert ran, 'should run contents of the block'
  end

  private

  def mocked_object(name)
    object = mock(name)
    self.stubs(name).returns(object)
    object
  end

  def mocked_control
    mocked_object('control')
  end
end