require "fluent/test"
require "fluent/test/driver/output"
require "fluent/test/helpers"
require "fluent/plugin/out_couch"

class CouchOutputTest < Test::Unit::TestCase
  include Fluent::Test::Helpers

  def setup
    Fluent::Test.setup
  end

  DATABASE_NAME = "fluent_test"
  COUCHHOST = ENV['COUCHHOST'] || "http://127.0.0.1:5984"

  CONFIG = %[
    database #{DATABASE_NAME}
  ]

  CONFIG_UPDATE_DOC = %[
    database #{DATABASE_NAME}
    doc_key_field key
    update_docs true
  ]

  CONFIG_UPDATE_DOC_WITH_TAG_PLACEHOLDER = %[
    database #{DATABASE_NAME}
    doc_key_field ${tag}
    update_docs true
  ]

  CONFIG_JSONPATH = %[
    database #{DATABASE_NAME}
    doc_key_field key
    doc_key_jsonpath $.nested.key
  ]

  CONFIG_JSONPATH_WITH_TAG_PLACEHOLDER = %[
    database #{DATABASE_NAME}
    doc_key_field key
    doc_key_jsonpath $.${tag}.key
  ]

  CONFIG_DESIGN = %[
    database #{DATABASE_NAME}
    doc_key_field key
    refresh_view_index d01
  ]

  def prepare_db
    @cr = CouchRest.new(COUCHHOST)
    @db = @cr.database(DATABASE_NAME)
    @db.delete! rescue nil
    @cr.create_db(DATABASE_NAME)
  end

  def create_driver(config = CONFIG)
    Fluent::Test::Driver::Output.new(Fluent::Plugin::CouchOutput).configure(config)
  end

  def test_configure
    d = create_driver
    assert_equal(DATABASE_NAME, d.instance.database)
    assert_equal('localhost', d.instance.host)
    assert_equal('5984', d.instance.port)
    assert_equal('http', d.instance.protocol)
    assert_nil(d.instance.refresh_view_index)
    assert_nil(d.instance.user)
    assert_nil(d.instance.password)
    assert_false(d.instance.instance_variable_get(:@update_docs))
    assert_nil(d.instance.doc_key_field)
    assert_nil(d.instance.doc_key_jsonpath)
  end

  class WriteTest < self
    def setup
      @d = create_driver
      prepare_db
    end

    def teardown
      @db.delete!
    end

    def test_write
      previous_rows = @d.instance.db.all_docs["total_rows"]
      time = event_time
      @d.run(default_tag: "tag") do
        @d.feed(time, {"message" => "record"})
      end
      rows = @d.instance.db.all_docs["total_rows"]
      assert_equal(rows, previous_rows + 1)
    end
  end

  class WriteWithUpdateDocTest < self
    def setup
      @d = create_driver(CONFIG_UPDATE_DOC)
      prepare_db
    end

    def teardown
      @db.delete!
    end

    def test_write
      time = event_time
      previous_rows = rows = 0
      @d.run(default_tag: "test") do
        @d.feed(time, {"key" => "record-1", "message" => "record"})
        previous_rows = @d.instance.db.all_docs["total_rows"]
        @d.feed(time, {"key" => "record-1", "message" => "record-mod"})
        rows = @d.instance.db.all_docs["total_rows"]
      end
      record = @d.instance.db.get("record-1")
      assert_equal(rows, previous_rows)
      assert_equal("record-mod", record["message"])
    end
  end

  class WriteWithUpdateDocWithTagPlaceHolderTest < self
    def setup
      @d = create_driver(CONFIG_UPDATE_DOC_WITH_TAG_PLACEHOLDER)
      prepare_db
    end

    def teardown
      @db.delete!
    end

    def test_write
      time = event_time
      tag = "replace.placeholder"
      previous_rows = rows = 0
      @d.run(default_tag: tag) do
        @d.feed(time, {tag => "record-placeholder", "message" => "record"})
        previous_rows = @d.instance.db.all_docs["total_rows"]
        @d.feed(time, {tag => "record-placeholder", "message" => "record-placeholder"})
        rows = @d.instance.db.all_docs["total_rows"]
      end
      assert_equal(rows, previous_rows)
      record = @d.instance.db.get("record-placeholder")
      assert_equal("record-placeholder", record["message"])
    end
  end

  class WriteWithJsonpathTest < self
    def setup
      @d = create_driver(CONFIG_JSONPATH)
      prepare_db
    end

    def teardown
      @db.delete!
    end

    def test_write
      time = event_time
      @d.run(default_tag: "test") do
        @d.feed(time, {"nested" => {"key" => "record-nested", "message" => "record"}})
      end
      record = @d.instance.db.get("record-nested")
      assert_equal({"key" => "record-nested", "message" => "record"}, record["nested"])
    end
  end

  class WriteWithJsonpathWithTagPlaceholderTest < self
    def setup
      @d = create_driver(CONFIG_JSONPATH_WITH_TAG_PLACEHOLDER)
      prepare_db
    end

    def teardown
      @db.delete!
    end

    def test_write
      time = event_time
      tag = "couchrecord"
      @d.run(default_tag: tag) do
        @d.feed(time, {tag => {"key" => "record-nested", "message" => "record"}})
      end
      record = @d.instance.db.get("record-nested")
      assert_equal({"key" => "record-nested", "message" => "record"}, record["couchrecord"])
    end
  end

  class WriteWithDesignTest < self
    def setup
      @d = create_driver(CONFIG_DESIGN)
      prepare_db
      setup_design
    end

    def setup_design
      message_by_key = {
        :map => 'function(doc) {
          if (doc._id == "record-design") {
            emit(doc._id, doc.message);
          }
        }',
      }
      @db.delete_doc db.get("_design/d01") rescue nil
      @db.save_doc({
        "_id" => "_design/d01",
        :views => {
          :messages => message_by_key
        }
      })
    end

    def teardown
      @db.delete!
    end

    def test_write
      time = event_time
      @d.run(default_tag: "test") do
        @d.feed(time, {"key" => "record-design", "message" => "record"})
      end
      record = @d.instance.db.get("record-design")
      assert_equal("record", record["message"])
    end
  end
end