# 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 File.expand_path(File.join(File.dirname(__FILE__),'..','..','test_helper'))

memcached_ready = false
classes = {
#   'memcache' => 'MemCache',
#   'dalli' => 'Dalli::Client',
  'memcached' => 'Memcached'
#   'spymemcached' => 'Spymemcached'
}
begin
  TCPSocket.new('localhost', 11211)
  classes.each do |req, const|
    begin
      require req
      MEMCACHED_CLASS = const.constantize
      puts "Testing #{MEMCACHED_CLASS}"
      memcached_ready = true
    rescue LoadError
    rescue NameError
    end
  end
rescue Errno::ECONNREFUSED
rescue Errno::ETIMEDOUT
end

if memcached_ready
  class NewRelic::Agent::MemcacheInstrumentationTest < Minitest::Test
    include NewRelic::Agent::Instrumentation::ControllerInstrumentation

    def setup
      NewRelic::Agent.manual_start
      @engine = NewRelic::Agent.instance.stats_engine

      case MEMCACHED_CLASS.name
      when 'Memcached'
        @cache = MEMCACHED_CLASS.new('localhost', :support_cas => true)
      when 'Spymemcached'
        @cache = MEMCACHED_CLASS.new('localhost:11211')
      else
        @cache = MEMCACHED_CLASS.new('localhost')
      end
      @key = 'schluessel'
      @cache.set('schluessel', 1)
    end

    def teardown
      if MEMCACHED_CLASS.name == 'Memcached'
        @cache.flush
      elsif MEMCACHED_CLASS.name == 'Spymemcached'
        @cache.flush
        @cache.instance_eval{ @client.shutdown }
      else
        @cache.flush_all
      end
    end

    def _call_test_method_in_web_transaction(method, *args)
      @engine.clear_stats
      perform_action_with_newrelic_trace(:name=>'action', :category => :controller) do
        @cache.send(method.to_sym, *[@key, *args])
      end
    end

    def _call_test_method_in_background_task(method, *args)
      @engine.clear_stats
      perform_action_with_newrelic_trace(:name => 'bg_task', :category => :task) do
        @cache.send(method.to_sym, *[@key, *args])
      end
    end

    def test_reads__web
      commands = ['get']
      commands << 'get_multi' unless MEMCACHED_CLASS.name == 'Spymemcached'
      commands.each do |method|
        if @cache.class.method_defined?(method)
          _call_test_method_in_web_transaction(method)
          compare_metrics ["Memcache/#{method}", "Memcache/allWeb", "Memcache/#{method}:Controller/NewRelic::Agent::MemcacheInstrumentationTest/action"],
          @engine.metrics.select{|m| m =~ /^memcache.*/i}
        end
      end
    end

    def test_writes__web
      %w[delete].each do |method|
        if @cache.class.method_defined?(method)
          _call_test_method_in_web_transaction(method)
          expected_metrics = ["Memcache/#{method}", "Memcache/allWeb", "Memcache/#{method}:Controller/NewRelic::Agent::MemcacheInstrumentationTest/action"]
          compare_metrics expected_metrics, @engine.metrics.select{|m| m =~ /^memcache.*/i}
        end
      end

      %w[set add].each do |method|
        @cache.delete(@key) rescue nil
        if @cache.class.method_defined?(method)
          expected_metrics = ["Memcache/#{method}", "Memcache/allWeb", "Memcache/#{method}:Controller/NewRelic::Agent::MemcacheInstrumentationTest/action"]
          _call_test_method_in_web_transaction(method, 'value')
          compare_metrics expected_metrics, @engine.metrics.select{|m| m =~ /^memcache.*/i}
        end
      end
    end

    def test_reads__background
      commands = ['get']
      commands << 'get_multi' unless MEMCACHED_CLASS.name == 'Spymemcached'
      commands.each do |method|
        if @cache.class.method_defined?(method)
          _call_test_method_in_background_task(method)
          compare_metrics ["Memcache/#{method}", "Memcache/allOther", "Memcache/#{method}:OtherTransaction/Background/NewRelic::Agent::MemcacheInstrumentationTest/bg_task"],
          @engine.metrics.select{|m| m =~ /^memcache.*/i}
        end
      end
    end

    def test_writes__background
      %w[delete].each do |method|
        expected_metrics = ["Memcache/#{method}", "Memcache/allOther", "Memcache/#{method}:OtherTransaction/Background/NewRelic::Agent::MemcacheInstrumentationTest/bg_task"]
        if @cache.class.method_defined?(method)
          _call_test_method_in_background_task(method)
          compare_metrics expected_metrics, @engine.metrics.select{|m| m =~ /^memcache.*/i}
        end
      end

      %w[set add].each do |method|
        @cache.delete(@key) rescue nil
        expected_metrics = ["Memcache/#{method}", "Memcache/allOther", "Memcache/#{method}:OtherTransaction/Background/NewRelic::Agent::MemcacheInstrumentationTest/bg_task"]
        if @cache.class.method_defined?(method)
          _call_test_method_in_background_task(method, 'value')
          compare_metrics expected_metrics, @engine.metrics.select{|m| m =~ /^memcache.*/i}
        end
      end
    end

    def test_handles_cas
      expected_metrics = ["Memcache/cas", "Memcache/allOther", "Memcache/cas:OtherTransaction/Background/NewRelic::Agent::MemcacheInstrumentationTest/bg_task"]
      if @cache.class.method_defined?(:cas)
        @engine.clear_stats
        perform_action_with_newrelic_trace(:name => 'bg_task', :category => :task) do
          @cache.cas(@key) {|val| val += 2 }
        end
        compare_metrics expected_metrics, @engine.metrics.select{|m| m =~ /^memcache.*/i}
        assert_equal 3, @cache.get(@key)
      end
    end
  end
end