# Copyright (C) 2015-2018  Kouhei Sutou <kou@clear-code.com>
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License version 2.1 as published by the Free Software Foundation.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA

class QueryLoggerTest < Test::Unit::TestCase
  include GroongaTestUtils

  def setup
    @default_log_path = Groonga::QueryLogger.path
    @default_rotate_threshold_size = Groonga::QueryLogger.rotate_threshold_size
    @default_flags = Groonga::QueryLogger.flags
  end

  def teardown
    Groonga::QueryLogger.path = @default_log_path
    Groonga::QueryLogger.rotate_threshold_size = @default_rotate_threshold_size
    Groonga::QueryLogger.flags = @default_flags
  end

  def test_reopen
    Groonga::QueryLogger.unregister
    Groonga::QueryLogger.path = @query_log_path.to_s
    if @query_log_path.exist?
      FileUtils.mv(@query_log_path, "#{@query_log_path}.old")
    end
    assert do
      not @query_log_path.exist?
    end
    Groonga::QueryLogger.reopen
    assert do
      @query_log_path.exist?
    end
  end

  sub_test_case ".parse" do
    def parse(input, base_flags=nil)
      base_flags ||= Groonga::QueryLogger::Flags::NONE
      Groonga::QueryLogger::Flags.parse(input, base_flags)
    end

    test "nil" do
      assert_equal(Groonga::QueryLogger::Flags::NONE,
                   parse(nil))
    end

    test "Integer" do
      assert_equal(Groonga::QueryLogger::Flags::COMMAND,
                   parse(Groonga::QueryLogger::Flags::COMMAND))
    end

    test "String" do
      assert_equal(Groonga::QueryLogger::Flags::COMMAND,
                   parse("command"))
    end

    test "Symbol" do
      assert_equal(Groonga::QueryLogger::Flags::COMMAND,
                   parse(:command))
    end

    test "Array" do
      assert_equal(Groonga::QueryLogger::Flags::COMMAND |
                   Groonga::QueryLogger::Flags::RESULT_CODE,
                   parse([:command, :result_code]))
    end

    test "Hash" do
      assert_equal(Groonga::QueryLogger::Flags::COMMAND |
                   Groonga::QueryLogger::Flags::DESTINATION,
                   parse({
                           :command => true,
                           :result_code => false,
                           :destination => true,
                         },
                         Groonga::QueryLogger::Flags::COMMAND |
                         Groonga::QueryLogger::Flags::RESULT_CODE))
    end
  end

  sub_test_case ".log" do
    test "no options" do
      messages = []
      Groonga::QueryLogger.register do |action, flag, timestamp, info, message|
        messages << message
      end
      Groonga::QueryLogger.log("1")
      Groonga::QueryLogger.log("2")
      Groonga::QueryLogger.log("3")
      assert_equal(["1", "2", "3"],
                   messages)
    end

    test ":flags" do
      infos = []
      Groonga::QueryLogger.register do |action, flag, timestamp, info, message|
        infos << info
      end
      Groonga::QueryLogger.log("default")
      Groonga::QueryLogger.log("flags", :flags => "command")
      normalized_infos = infos.collect do |info|
        info = info.gsub(/\A0x[a-f\d]+\|/,
                         "context_id|")
        info.gsub(/\|[\d]+ \z/,
                  "|timestamp ")
      end
      assert_equal([
                     "context_id|timestamp ",
                     "context_id|",
                   ],
                   normalized_infos)
    end

    test ":mark" do
      infos = []
      Groonga::QueryLogger.register do |action, flag, timestamp, info, message|
        infos << info
      end
      Groonga::QueryLogger.log("default")
      Groonga::QueryLogger.log("mark", :mark => ":")
      normalized_infos = infos.collect do |info|
        info.gsub(/\A0x[a-f\d]+\|([^\d])?[\d]+ \z/,
                  "context_id|\\1timestamp ")
      end
      assert_equal([
                     "context_id|timestamp ",
                     "context_id|:timestamp ",
                   ],
                   normalized_infos)
    end
  end

  def test_rotate_threshold_size
    Groonga::QueryLogger.unregister
    Groonga::QueryLogger.path = @query_log_path.to_s
    Groonga::QueryLogger.reopen
    Groonga::QueryLogger.rotate_threshold_size = 10
    assert_equal([], Dir.glob("#{@query_log_path}.*"))
    Groonga::QueryLogger.log("command")
    assert_not_equal([], Dir.glob("#{@query_log_path}.*"))
  end

  sub_test_case("flags") do
    test("name") do
      Groonga::QueryLogger.flags = :command
      assert_equal(Groonga::QueryLogger::Flags::COMMAND,
                   Groonga::QueryLogger.flags)
    end
  end
end