file_log.cpp 5.34 KB
Newer Older
gabime's avatar
gabime committed
1 2 3 4 5 6 7 8
/*
 * This content is released under the MIT License as specified in https://raw.githubusercontent.com/gabime/spdlog/master/LICENSE
 */
#include "includes.h"

TEST_CASE("simple_file_logger", "[simple_logger]]")
{
    prepare_logdir();
9
    std::string filename = "logs/simple_log";
gabime's avatar
gabime committed
10

11
    auto logger = spdlog::create<spdlog::sinks::basic_file_sink_mt>("logger", filename);
gabime's avatar
gabime committed
12 13 14 15
    logger->set_pattern("%v");

    logger->info("Test message {}", 1);
    logger->info("Test message {}", 2);
gabime's avatar
gabime committed
16

gabime's avatar
gabime committed
17 18 19
    logger->flush();
    REQUIRE(file_contents(filename) == std::string("Test message 1\nTest message 2\n"));
    REQUIRE(count_lines(filename) == 2);
20 21 22 23
}

TEST_CASE("flush_on", "[flush_on]]")
{
gabime's avatar
gabime committed
24
    prepare_logdir();
25
    std::string filename = "logs/simple_log";
gabime's avatar
gabime committed
26

27
    auto logger = spdlog::create<spdlog::sinks::basic_file_sink_mt>("logger", filename);
gabime's avatar
gabime committed
28 29 30 31 32 33 34 35
    logger->set_pattern("%v");
    logger->set_level(spdlog::level::trace);
    logger->flush_on(spdlog::level::info);
    logger->trace("Should not be flushed");
    REQUIRE(count_lines(filename) == 0);

    logger->info("Test message {}", 1);
    logger->info("Test message {}", 2);
36

gabime's avatar
gabime committed
37 38
    REQUIRE(file_contents(filename) == std::string("Should not be flushed\nTest message 1\nTest message 2\n"));
    REQUIRE(count_lines(filename) == 3);
gabime's avatar
gabime committed
39 40 41 42 43
}

TEST_CASE("rotating_file_logger1", "[rotating_logger]]")
{
    prepare_logdir();
gabime's avatar
gabime committed
44
    size_t max_size = 1024 * 10;
gabime's avatar
gabime committed
45
    std::string basename = "logs/rotating_log";
46
    auto logger = spdlog::rotating_logger_mt("logger", basename, max_size, 0);
gabime's avatar
gabime committed
47

gabime's avatar
gabime committed
48
    for (int i = 0; i < 10; ++i)
fogo's avatar
fogo committed
49
    {
gabime's avatar
gabime committed
50
        logger->info("Test message {}", i);
fogo's avatar
fogo committed
51
    }
gabime's avatar
gabime committed
52

gabime's avatar
gabime committed
53
    logger->flush();
54
    auto filename = basename;
gabime's avatar
gabime committed
55 56 57 58 59 60
    REQUIRE(count_lines(filename) == 10);
}

TEST_CASE("rotating_file_logger2", "[rotating_logger]]")
{
    prepare_logdir();
gabime's avatar
gabime committed
61
    size_t max_size = 1024 * 10;
gabime's avatar
gabime committed
62
    std::string basename = "logs/rotating_log";
63
    auto logger = spdlog::rotating_logger_mt("logger", basename, max_size, 1);
gabime's avatar
gabime committed
64 65 66 67
    for (int i = 0; i < 10; ++i)
        logger->info("Test message {}", i);

    logger->flush();
68
    auto filename = basename;
gabime's avatar
gabime committed
69 70
    REQUIRE(count_lines(filename) == 10);
    for (int i = 0; i < 1000; i++)
fogo's avatar
fogo committed
71
    {
gabime's avatar
gabime committed
72

gabime's avatar
gabime committed
73
        logger->info("Test message {}", i);
fogo's avatar
fogo committed
74
    }
gabime's avatar
gabime committed
75 76

    logger->flush();
77
    REQUIRE(get_filesize(filename) <= max_size);
78
    auto filename1 = basename + ".1";
79
    REQUIRE(get_filesize(filename1) <= max_size);
gabime's avatar
gabime committed
80 81
}

82 83
TEST_CASE("daily_logger with dateonly calculator", "[daily_logger_dateonly]]")
{
84
    using sink_type = spdlog::sinks::daily_file_sink<std::mutex, spdlog::sinks::daily_filename_calculator>;
gabime's avatar
gabime committed
85

86
    prepare_logdir();
gabime's avatar
gabime committed
87
    // calculate filename (time based)
88 89
    std::string basename = "logs/daily_dateonly";
    std::tm tm = spdlog::details::os::localtime();
gabime's avatar
gabime committed
90 91
    fmt::memory_buffer w;
    fmt::format_to(w, "{}_{:04d}-{:02d}-{:02d}", basename, tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday);
gabime's avatar
gabime committed
92

93
    auto logger = spdlog::create<sink_type>("logger", basename, 0, 0);
94
    for (int i = 0; i < 10; ++i)
fogo's avatar
fogo committed
95
    {
gabime's avatar
gabime committed
96

97
        logger->info("Test message {}", i);
fogo's avatar
fogo committed
98
    }
gabime's avatar
gabime committed
99
    logger->flush();
gabime's avatar
gabime committed
100
    auto filename = fmt::to_string(w);
101 102
    REQUIRE(count_lines(filename) == 10);
}
gabime's avatar
gabime committed
103

104 105
struct custom_daily_file_name_calculator
{
106
    static spdlog::filename_t calc_filename(const spdlog::filename_t &basename, const tm &now_tm)
107
    {
gabime's avatar
gabime committed
108
        fmt::memory_buffer w;
109
        fmt::format_to(w, "{}{:04d}{:02d}{:02d}", basename, now_tm.tm_year + 1900, now_tm.tm_mon + 1, now_tm.tm_mday);
gabime's avatar
gabime committed
110
        return fmt::to_string(w);
111 112 113 114 115
    }
};

TEST_CASE("daily_logger with custom calculator", "[daily_logger_custom]]")
{
gabime's avatar
gabime committed
116
    using sink_type = spdlog::sinks::daily_file_sink<std::mutex, custom_daily_file_name_calculator>;
gabime's avatar
gabime committed
117

118
    prepare_logdir();
gabime's avatar
gabime committed
119
    // calculate filename (time based)
120 121
    std::string basename = "logs/daily_dateonly";
    std::tm tm = spdlog::details::os::localtime();
gabime's avatar
gabime committed
122 123
    fmt::memory_buffer w;
    fmt::format_to(w, "{}{:04d}{:02d}{:02d}", basename, tm.tm_year + 1900, tm.tm_mon + 1, tm.tm_mday);
gabime's avatar
gabime committed
124

125
    auto logger = spdlog::create<sink_type>("logger", basename, 0, 0);
126
    for (int i = 0; i < 10; ++i)
fogo's avatar
fogo committed
127
    {
128
        logger->info("Test message {}", i);
fogo's avatar
fogo committed
129
    }
gabime's avatar
gabime committed
130

gabime's avatar
gabime committed
131
    logger->flush();
gabime's avatar
gabime committed
132
    auto filename = fmt::to_string(w);
133 134
    REQUIRE(count_lines(filename) == 10);
}
gabime's avatar
gabime committed
135

136 137 138 139 140 141
/*
 * File name calculations
 */

TEST_CASE("rotating_file_sink::calc_filename1", "[rotating_file_sink]]")
{
gabime's avatar
gabime committed
142 143
    auto filename = spdlog::sinks::rotating_file_sink_st::calc_filename("rotated.txt", 3);
    REQUIRE(filename == "rotated.3.txt");
144 145 146 147
}

TEST_CASE("rotating_file_sink::calc_filename2", "[rotating_file_sink]]")
{
gabime's avatar
gabime committed
148 149
    auto filename = spdlog::sinks::rotating_file_sink_st::calc_filename("rotated", 3);
    REQUIRE(filename == "rotated.3");
150 151 152 153
}

TEST_CASE("rotating_file_sink::calc_filename3", "[rotating_file_sink]]")
{
gabime's avatar
gabime committed
154 155
    auto filename = spdlog::sinks::rotating_file_sink_st::calc_filename("rotated.txt", 0);
    REQUIRE(filename == "rotated.txt");
156 157
}

158
// regex supported only from gcc 4.9 and above
gabime's avatar
gabime committed
159
#if defined(_MSC_VER) || !(__GNUC__ <= 4 && __GNUC_MINOR__ < 9)
gabime's avatar
gabime committed
160
#include <regex>
161

162
TEST_CASE("daily_file_sink::daily_filename_calculator", "[daily_file_sink]]")
163
{
gabime's avatar
gabime committed
164
    // daily_YYYY-MM-DD_hh-mm.txt
165
    auto filename = spdlog::sinks::daily_filename_calculator::calc_filename("daily.txt", spdlog::details::os::localtime());
gabime's avatar
gabime committed
166 167 168 169
    // date regex based on https://www.regular-expressions.info/dates.html
    std::regex re(R"(^daily_(19|20)\d\d-(0[1-9]|1[012])-(0[1-9]|[12][0-9]|3[01])\.txt$)");
    std::smatch match;
    REQUIRE(std::regex_match(filename, match, re));
170
}
171
#endif