// Copyright 2011 The Kyua Authors.

// All rights reserved.

//

// Redistribution and use in source and binary forms, with or without

// modification, are permitted provided that the following conditions are

// met:

//

// * Redistributions of source code must retain the above copyright

//   notice, this list of conditions and the following disclaimer.

// * Redistributions in binary form must reproduce the above copyright

//   notice, this list of conditions and the following disclaimer in the

//   documentation and/or other materials provided with the distribution.

// * Neither the name of Google Inc. nor the names of its contributors

//   may be used to endorse or promote products derived from this software

//   without specific prior written permission.

//

// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR

// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT

// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#include "utils/logging/operations.hpp"

extern "C" {

#include <unistd.h>

}

#include <fstream>

#include <string>

#include <atf-c++.hpp>

#include "utils/datetime.hpp"

#include "utils/format/macros.hpp"

#include "utils/fs/operations.hpp"

#include "utils/fs/path.hpp"

namespace datetime = utils::datetime;

namespace fs = utils::fs;

namespace logging = utils::logging;

ATF_TEST_CASE_WITHOUT_HEAD(generate_log_name__before_log);

ATF_TEST_CASE_BODY(generate_log_name__before_log)

{

    datetime::set_mock_now(2011, 2, 21, 18, 10, 0, 0);

    ATF_REQUIRE_EQ(fs::path("/some/dir/foobar.20110221-181000.log"),

                   logging::generate_log_name(fs::path("/some/dir"), "foobar"));

    datetime::set_mock_now(2011, 2, 21, 18, 10, 1, 987654);

    logging::log(logging::level_info, "file", 123, "A message");

    datetime::set_mock_now(2011, 2, 21, 18, 10, 2, 123);

    ATF_REQUIRE_EQ(fs::path("/some/dir/foobar.20110221-181000.log"),

                   logging::generate_log_name(fs::path("/some/dir"), "foobar"));

}

ATF_TEST_CASE_WITHOUT_HEAD(generate_log_name__after_log);

ATF_TEST_CASE_BODY(generate_log_name__after_log)

{

    datetime::set_mock_now(2011, 2, 21, 18, 15, 0, 0);

    logging::log(logging::level_info, "file", 123, "A message");

    datetime::set_mock_now(2011, 2, 21, 18, 15, 1, 987654);

    logging::log(logging::level_info, "file", 123, "A message");

    datetime::set_mock_now(2011, 2, 21, 18, 15, 2, 123);

    ATF_REQUIRE_EQ(fs::path("/some/dir/foobar.20110221-181500.log"),

                   logging::generate_log_name(fs::path("/some/dir"), "foobar"));

    datetime::set_mock_now(2011, 2, 21, 18, 15, 3, 1);

    logging::log(logging::level_info, "file", 123, "A message");

    datetime::set_mock_now(2011, 2, 21, 18, 15, 4, 91);

    ATF_REQUIRE_EQ(fs::path("/some/dir/foobar.20110221-181500.log"),

                   logging::generate_log_name(fs::path("/some/dir"), "foobar"));

}

ATF_TEST_CASE_WITHOUT_HEAD(log);

ATF_TEST_CASE_BODY(log)

{

    logging::set_inmemory();

    datetime::set_mock_now(2011, 2, 21, 18, 10, 0, 0);

    logging::log(logging::level_debug, "f1", 1, "Debug message");

    datetime::set_mock_now(2011, 2, 21, 18, 10, 1, 987654);

    logging::log(logging::level_error, "f2", 2, "Error message");

    logging::set_persistency("debug", fs::path("test.log"));

    datetime::set_mock_now(2011, 2, 21, 18, 10, 2, 123);

    logging::log(logging::level_info, "f3", 3, "Info message");

    datetime::set_mock_now(2011, 2, 21, 18, 10, 3, 456);

    logging::log(logging::level_warning, "f4", 4, "Warning message");

    std::ifstream input("test.log");

    ATF_REQUIRE(input);

    const pid_t pid = ::getpid();

    std::string line;

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110221-181000 D %s f1:1: Debug message") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110221-181001 E %s f2:2: Error message") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110221-181002 I %s f3:3: Info message") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110221-181003 W %s f4:4: Warning message") % pid).str(), line);

}

ATF_TEST_CASE_WITHOUT_HEAD(set_inmemory__reset);

ATF_TEST_CASE_BODY(set_inmemory__reset)

{

    logging::set_persistency("debug", fs::path("test.log"));

    datetime::set_mock_now(2011, 2, 21, 18, 20, 0, 654321);

    logging::log(logging::level_debug, "file", 123, "Debug message");

    logging::set_inmemory();

    logging::log(logging::level_debug, "file", 123, "Debug message 2");

    std::ifstream input("test.log");

    ATF_REQUIRE(input);

    const pid_t pid = ::getpid();

    std::string line;

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110221-182000 D %s file:123: Debug message") % pid).str(), line);

}

ATF_TEST_CASE_WITHOUT_HEAD(set_persistency__no_backlog);

ATF_TEST_CASE_BODY(set_persistency__no_backlog)

{

    logging::set_persistency("debug", fs::path("test.log"));

    datetime::set_mock_now(2011, 2, 21, 18, 20, 0, 654321);

    logging::log(logging::level_debug, "file", 123, "Debug message");

    std::ifstream input("test.log");

    ATF_REQUIRE(input);

    const pid_t pid = ::getpid();

    std::string line;

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110221-182000 D %s file:123: Debug message") % pid).str(), line);

}

/// Creates a log for testing purposes, buffering messages on start.

///

/// \param level The level of the desired log.

/// \param path The output file.

static void

create_log(const std::string& level, const std::string& path)

{

    logging::set_inmemory();

    datetime::set_mock_now(2011, 3, 19, 11, 40, 0, 100);

    logging::log(logging::level_debug, "file1", 11, "Debug 1");

    datetime::set_mock_now(2011, 3, 19, 11, 40, 1, 200);

    logging::log(logging::level_error, "file2", 22, "Error 1");

    datetime::set_mock_now(2011, 3, 19, 11, 40, 2, 300);

    logging::log(logging::level_info, "file3", 33, "Info 1");

    datetime::set_mock_now(2011, 3, 19, 11, 40, 3, 400);

    logging::log(logging::level_warning, "file4", 44, "Warning 1");

    logging::set_persistency(level, fs::path(path));

    datetime::set_mock_now(2011, 3, 19, 11, 40, 4, 500);

    logging::log(logging::level_debug, "file1", 11, "Debug 2");

    datetime::set_mock_now(2011, 3, 19, 11, 40, 5, 600);

    logging::log(logging::level_error, "file2", 22, "Error 2");

    datetime::set_mock_now(2011, 3, 19, 11, 40, 6, 700);

    logging::log(logging::level_info, "file3", 33, "Info 2");

    datetime::set_mock_now(2011, 3, 19, 11, 40, 7, 800);

    logging::log(logging::level_warning, "file4", 44, "Warning 2");

}

ATF_TEST_CASE_WITHOUT_HEAD(set_persistency__some_backlog__debug);

ATF_TEST_CASE_BODY(set_persistency__some_backlog__debug)

{

    create_log("debug", "test.log");

    std::ifstream input("test.log");

    ATF_REQUIRE(input);

    const pid_t pid = ::getpid();

    std::string line;

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114000 D %s file1:11: Debug 1") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114001 E %s file2:22: Error 1") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114002 I %s file3:33: Info 1") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114003 W %s file4:44: Warning 1") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114004 D %s file1:11: Debug 2") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114005 E %s file2:22: Error 2") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114006 I %s file3:33: Info 2") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114007 W %s file4:44: Warning 2") % pid).str(), line);

}

ATF_TEST_CASE_WITHOUT_HEAD(set_persistency__some_backlog__error);

ATF_TEST_CASE_BODY(set_persistency__some_backlog__error)

{

    create_log("error", "test.log");

    std::ifstream input("test.log");

    ATF_REQUIRE(input);

    const pid_t pid = ::getpid();

    std::string line;

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114001 E %s file2:22: Error 1") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114005 E %s file2:22: Error 2") % pid).str(), line);

}

ATF_TEST_CASE_WITHOUT_HEAD(set_persistency__some_backlog__info);

ATF_TEST_CASE_BODY(set_persistency__some_backlog__info)

{

    create_log("info", "test.log");

    std::ifstream input("test.log");

    ATF_REQUIRE(input);

    const pid_t pid = ::getpid();

    std::string line;

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114001 E %s file2:22: Error 1") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114002 I %s file3:33: Info 1") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114003 W %s file4:44: Warning 1") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114005 E %s file2:22: Error 2") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114006 I %s file3:33: Info 2") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114007 W %s file4:44: Warning 2") % pid).str(), line);

}

ATF_TEST_CASE_WITHOUT_HEAD(set_persistency__some_backlog__warning);

ATF_TEST_CASE_BODY(set_persistency__some_backlog__warning)

{

    create_log("warning", "test.log");

    std::ifstream input("test.log");

    ATF_REQUIRE(input);

    const pid_t pid = ::getpid();

    std::string line;

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114001 E %s file2:22: Error 1") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114003 W %s file4:44: Warning 1") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114005 E %s file2:22: Error 2") % pid).str(), line);

    ATF_REQUIRE(std::getline(input, line).good());

    ATF_REQUIRE_EQ(

        (F("20110319-114007 W %s file4:44: Warning 2") % pid).str(), line);

}

ATF_TEST_CASE(set_persistency__fail);

ATF_TEST_CASE_HEAD(set_persistency__fail)

{

    set_md_var("require.user", "unprivileged");

}

ATF_TEST_CASE_BODY(set_persistency__fail)

{

    ATF_REQUIRE_THROW_RE(std::range_error, "'foobar'",

                         logging::set_persistency("foobar", fs::path("log")));

    fs::mkdir(fs::path("dir"), 0644);

    ATF_REQUIRE_THROW_RE(std::runtime_error, "dir/fail.log",

                         logging::set_persistency("debug",

                                                  fs::path("dir/fail.log")));

}

ATF_INIT_TEST_CASES(tcs)

{

    ATF_ADD_TEST_CASE(tcs, generate_log_name__before_log);

    ATF_ADD_TEST_CASE(tcs, generate_log_name__after_log);

    ATF_ADD_TEST_CASE(tcs, log);

    ATF_ADD_TEST_CASE(tcs, set_inmemory__reset);

    ATF_ADD_TEST_CASE(tcs, set_persistency__no_backlog);

    ATF_ADD_TEST_CASE(tcs, set_persistency__some_backlog__debug);

    ATF_ADD_TEST_CASE(tcs, set_persistency__some_backlog__error);

    ATF_ADD_TEST_CASE(tcs, set_persistency__some_backlog__info);

    ATF_ADD_TEST_CASE(tcs, set_persistency__some_backlog__warning);

    ATF_ADD_TEST_CASE(tcs, set_persistency__fail);

}