require 'helper'
require 'webmock/test_unit'
require 'yajl'

WebMock.disable_net_connect!

class SentryOutputTest < Test::Unit::TestCase
  def setup
    Fluent::Test.setup
  end

  CONFIG = %[
    type sentry
    endpoint_url      https://user:password@app.getsentry.com/12345
    hostname_command  hostname -s
    remove_tag_prefix input.
  ]

  def create_driver(conf=CONFIG,tag='test',use_v1=false)
    require 'fluent/version'
    if Gem::Version.new(Fluent::VERSION) < Gem::Version.new('0.12')
      Fluent::Test::OutputTestDriver.new(Fluent::SentryOutput, tag).configure(conf, use_v1)
    else
      Fluent::Test::BufferedOutputTestDriver.new(Fluent::SentryOutput, tag).configure(conf, use_v1)
    end
  end

  def stub_1(url="https://app.getsentry.com/api/12345/envelope/")
    stub_request(:post, "https://app.getsentry.com/api/12345/envelope/").
      with(
        body: "{\"event_id\":\"2aa1f0210fc04c89a8febda09d3e4a5f\",\"dsn\":\"https://user:password@app.getsentry.com/12345\",\"sdk\":{\"name\":\"sentry.ruby\",\"version\":\"4.6.5\"},\"sent_at\":\"2021-09-01T12:00:23Z\"}\n{\"type\":\"event\",\"content_type\":\"application/json\"}\n{\"event_id\":\"2aa1f0210fc04c89a8febda09d3e4a5f\",\"level\":\"warning\",\"timestamp\":\"2021-09-01T12:00:23\",\"environment\":\"default\",\"server_name\":\"reisei-workspace\",\"modules\":{\"rake\":\"13.0.6\",\"bundler\":\"2.2.26\",\"cool.io\":\"1.7.1\",\"http_parser.rb\":\"0.7.0\",\"msgpack\":\"1.4.2\",\"sigdump\":\"0.2.4\",\"serverengine\":\"2.2.4\",\"strptime\":\"0.2.5\",\"concurrent-ruby\":\"1.1.9\",\"tzinfo\":\"2.0.4\",\"tzinfo-data\":\"1.2021.1\",\"webrick\":\"1.7.0\",\"yajl-ruby\":\"1.4.1\",\"fluentd\":\"1.14.0\",\"faraday-em_http\":\"1.0.0\",\"faraday-em_synchrony\":\"1.0.0\",\"faraday-excon\":\"1.1.0\",\"faraday-httpclient\":\"1.0.1\",\"faraday-net_http\":\"1.0.1\",\"faraday-net_http_persistent\":\"1.2.0\",\"faraday-patron\":\"1.0.0\",\"faraday-rack\":\"1.0.0\",\"multipart-post\":\"2.1.1\",\"ruby2_keywords\":\"0.0.5\",\"faraday\":\"1.7.0\",\"sentry-ruby-core\":\"4.6.5\",\"sentry-ruby\":\"4.6.5\",\"ach-fluent-plugin-sentry\":\"0.0.9\",\"public_suffix\":\"4.0.6\",\"addressable\":\"2.8.0\",\"thor\":\"1.1.0\",\"appraisal\":\"2.4.1\",\"rexml\":\"3.2.5\",\"crack\":\"0.4.5\",\"hashdiff\":\"1.0.1\",\"power_assert\":\"2.0.1\",\"test-unit\":\"3.4.4\",\"webmock\":\"3.14.0\"},\"message\":\"error has occoured.\",\"user\":{},\"tags\":{\"tag\":\"app1_error\"},\"contexts\":{},\"extra\":{},\"fingerprint\":[],\"platform\":\"ruby\",\"sdk\":{\"name\":\"sentry.ruby\",\"version\":\"4.6.5\"}}\n",
        headers: {
  	      'Accept'=>'*/*',
  	      'Accept-Encoding'=>'gzip;q=1.0,deflate;q=0.6,identity;q=0.3',
  	      'Content-Encoding'=>'',
  	      'Content-Type'=>'application/x-sentry-envelope',
  	      'User-Agent'=>'sentry-ruby/4.6.5',
  	      'X-Sentry-Auth'=>'Sentry sentry_version=7, sentry_client=sentry-ruby/4.6.5, sentry_timestamp=1630497623, sentry_key=user, sentry_secret=password'
        }).to_return(status: 200, body: "", headers: {})
  end

  def stub_post(url="https://app.getsentry.com/api/12345/store/")
    parser = Yajl::Parser.new
    stub_request(:post, url).with do |req|
      @content_type = req.headers["Content-Type"]
      message = Zlib::Inflate.inflate(Base64.decode64(req.body))
      @body = parser.parse(message)
    end
  end

  def stub_response(url="https://app.getsentry.com/api/12345/envelope/")
    stub_request(:post, url).with do |req|
      @content_type = req.headers["Content-Type"]
      message = Zlib::Inflate.inflate(Base64.decode64(req.body))
      @body = {"eventID"=>"fe0263d6f55d014cade15a8681ae58ed", "tags"=>[["tag", "app1_error"], ["cool", "bar"], ["level", "warning"], ["logger", "zaphod"], ["server_name", "bc40a4be7b2d"], ["sentry:release", "wingnut"]], "nextEventID"=>nil, "dateCreated"=>"2015-10-30T08:25:48Z", "timeSpent"=>13, "user"=>nil, "entries"=>[], "previousEventID"=>nil, "message"=>"error has occoured.", "packages"=>{"fluentd"=>"0.12.16", "sentry-raven"=>"0.15.2", "cool.io"=>"1.4.1", "faraday"=>"0.9.2", "safe_yaml"=>"1.0.4", "msgpack"=>"0.5.12", "bundler"=>"1.10.6", "json"=>"1.8.3", "crack"=>"0.4.2", "tzinfo"=>"1.2.2", "thread_safe"=>"0.3.5", "tzinfo-data"=>"1.2015.7", "rake"=>"10.4.2", "yajl-ruby"=>"1.2.1", "hashdiff"=>"0.2.2", "sigdump"=>"0.2.3", "multipart-post"=>"2.0.0", "string-scrub"=>"0.0.5", "addressable"=>"2.3.8", "webmock"=>"1.22.2", "http_parser.rb"=>"0.6.0", "fluent-plugin-sentry"=>"0.0.1"}, "id"=>"127", "platform"=>"ruby", "context"=>{"something"=>{"foo"=>{"array"=>[1, 2, 3]}, "hash"=>{"nest"=>"data"}}}, "groupID"=>127}
    end
  end

  def test_configure
    assert_raise(Fluent::ConfigError) {
      d = create_driver('')
    }
    d = create_driver(CONFIG)
    assert_equal 'https://user:password@app.getsentry.com/12345', d.instance.config['endpoint_url']
  end

  def test_emit
    stub_request(:post, "https://app.getsentry.com/api/12345/envelope/").
      with(
        body: "{\"event_id\":\"2aa1f0210fc04c89a8febda09d3e4a5f\",\"dsn\":\"https://user:password@app.getsentry.com/12345\",\"sdk\":{\"name\":\"sentry.ruby\",\"version\":\"4.6.5\"},\"sent_at\":\"2021-09-01T12:00:23Z\"}\n{\"type\":\"event\",\"content_type\":\"application/json\"}\n{\"event_id\":\"2aa1f0210fc04c89a8febda09d3e4a5f\",\"level\":\"warning\",\"timestamp\":\"2021-09-01T12:00:23\",\"environment\":\"default\",\"server_name\":\"reisei-workspace\",\"modules\":{\"rake\":\"13.0.6\",\"bundler\":\"2.2.26\",\"cool.io\":\"1.7.1\",\"http_parser.rb\":\"0.7.0\",\"msgpack\":\"1.4.2\",\"sigdump\":\"0.2.4\",\"serverengine\":\"2.2.4\",\"strptime\":\"0.2.5\",\"concurrent-ruby\":\"1.1.9\",\"tzinfo\":\"2.0.4\",\"tzinfo-data\":\"1.2021.1\",\"webrick\":\"1.7.0\",\"yajl-ruby\":\"1.4.1\",\"fluentd\":\"1.14.0\",\"faraday-em_http\":\"1.0.0\",\"faraday-em_synchrony\":\"1.0.0\",\"faraday-excon\":\"1.1.0\",\"faraday-httpclient\":\"1.0.1\",\"faraday-net_http\":\"1.0.1\",\"faraday-net_http_persistent\":\"1.2.0\",\"faraday-patron\":\"1.0.0\",\"faraday-rack\":\"1.0.0\",\"multipart-post\":\"2.1.1\",\"ruby2_keywords\":\"0.0.5\",\"faraday\":\"1.7.0\",\"sentry-ruby-core\":\"4.6.5\",\"sentry-ruby\":\"4.6.5\",\"ach-fluent-plugin-sentry\":\"0.0.9\",\"public_suffix\":\"4.0.6\",\"addressable\":\"2.8.0\",\"thor\":\"1.1.0\",\"appraisal\":\"2.4.1\",\"rexml\":\"3.2.5\",\"crack\":\"0.4.5\",\"hashdiff\":\"1.0.1\",\"power_assert\":\"2.0.1\",\"test-unit\":\"3.4.4\",\"webmock\":\"3.14.0\"},\"message\":\"error has occoured.\",\"user\":{},\"tags\":{\"tag\":\"app1_error\"},\"contexts\":{},\"extra\":{},\"fingerprint\":[],\"platform\":\"ruby\",\"sdk\":{\"name\":\"sentry.ruby\",\"version\":\"4.6.5\"}}\n",
        headers: {
  	      'Accept'=>'*/*',
  	      'Accept-Encoding'=>'gzip;q=1.0,deflate;q=0.6,identity;q=0.3',
  	      'Content-Encoding'=>'',
  	      'Content-Type'=>'application/x-sentry-envelope',
  	      'User-Agent'=>'sentry-ruby/4.6.5',
  	      'X-Sentry-Auth'=>'Sentry sentry_version=7, sentry_client=sentry-ruby/4.6.5, sentry_timestamp=1630497623, sentry_key=user, sentry_secret=password'}).to_return(status: 200, body: "", headers: {})
    #stub_post
    d1 = create_driver(CONFIG, 'input.app1_error')
    emit_level = 'warning'
    emit_message = 'error has occoured.'
    emit_extra = {'foo' => {'array' => [1,2,3]}, 'hash' => {'nest' => 'data'}}
    d1.run do
      d1.emit({
        'level' => emit_level,
        'message' => emit_message,
        'something' => emit_extra
      })
    end
    p @body
    emits = d1.emits
    timestamp = Time.now.utc.strftime('%Y-%m-%dT%H:%M:%S')
    assert_equal 0, emits.length
    #assert_equal 'application/octet-stream', @content_type
    assert_equal emit_message, @body['message']
    assert_equal timestamp, @body['timestamp']
    assert_equal emit_level, @body['level']
    assert_equal 'fluentd', @body['logger']
    assert_equal 'ruby', @body['platform']
    assert_equal 'app1_error', @body['tags']['tag']
    hostname = `#{d1.instance.config['hostname_command']}`.chomp
    assert_equal hostname, @body['server_name']
    extra_message = {'something' => emit_extra}
    assert_equal extra_message, @body['extra']
  end

  def test_emit_mock
    stub_response
    emit_level = 'warning'
    emit_message = 'error has occoured.'
    emit_timestamp = '2015-10-30T08:25:48Z'.force_encoding("UTF-8")
    emit_time_spent = '13'
    emit_logger = 'zaphod'
    emit_server_name = 'bc40a4be7b2d'
    emit_culprit = 'whodonit'
    emit_release = 'wingnut'
    emit_extra = {'foo' => {'array' => [1,2,3]}, 'hash' => {'nest' => 'data'}}
    d1 = create_driver(CONFIG, 'input.app1_error')
    d1.run do
      d1.emit({
        'level' => emit_level,
        'message' => emit_message,
        'something' => emit_extra,
        'tags' => {'cool' => 'bar'},
        'timestamp' => emit_timestamp,
        'logger' => emit_logger,
        'server_name' => emit_server_name,
        'release' => emit_release,
        'time_spent' => emit_time_spent,
        'culprit' => emit_culprit
      })
    end
    p @body
    emits = d1.emits
    extra_message = {'something' => emit_extra}
    assert_equal 0, emits.length
    #assert_equal 'application/octet-stream', @content_type
    assert_equal extra_message, @body['context']
    assert_equal emit_timestamp, @body['dateCreated']
    assert_equal emit_message, @body['message']
    assert_equal emit_level, Hash[ @body['tags'] ]['level']
    assert_equal emit_logger, Hash[ @body['tags'] ]['logger']
    assert_equal emit_server_name, Hash[ @body['tags'] ]['server_name']
    assert_equal emit_release, Hash[ @body['tags'] ]['sentry:release']
    assert_equal 'bar', Hash[ @body['tags'] ]['cool']
    assert_equal 'app1_error', Hash[ @body['tags'] ]['tag']
    assert_equal 'ruby', @body['platform']
    assert_equal nil, @body['user']
    # these values seem to only be visible in the ui. need to find the api to grab them
    #assert_equal emit_culprit, @body['culprit']
    #assert_equal emit_time_spent, @body['timeSpent']
  end
end