| // Copyright 2016 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "src/lib/fxl/log_settings_command_line.h" |
| |
| #include <unistd.h> |
| |
| #include <gtest/gtest.h> |
| |
| #include "src/lib/files/file.h" |
| #include "src/lib/files/scoped_temp_dir.h" |
| #include "src/lib/files/unique_fd.h" |
| #include "src/lib/fxl/command_line.h" |
| #include "src/lib/fxl/log_settings.h" |
| #include "src/lib/fxl/logging.h" |
| |
| namespace fxl { |
| namespace { |
| |
| class LogSettingsFixture : public ::testing::Test { |
| public: |
| LogSettingsFixture() : old_severity_(GetMinLogLevel()), old_stderr_(dup(STDERR_FILENO)) {} |
| ~LogSettingsFixture() { |
| SetLogSettings({.min_log_level = old_severity_}); |
| dup2(old_stderr_.get(), STDERR_FILENO); |
| } |
| |
| private: |
| LogSeverity old_severity_; |
| fbl::unique_fd old_stderr_; |
| }; |
| |
| TEST(LogSettings, ParseValidOptions) { |
| LogSettings settings; |
| settings.min_log_level = LOG_FATAL; |
| |
| EXPECT_TRUE(ParseLogSettings(CommandLineFromInitializerList({"argv0"}), &settings)); |
| EXPECT_EQ(LOG_FATAL, settings.min_log_level); |
| |
| EXPECT_TRUE(ParseLogSettings(CommandLineFromInitializerList({"argv0", "--verbose"}), &settings)); |
| EXPECT_EQ(-1, settings.min_log_level); |
| |
| EXPECT_TRUE( |
| ParseLogSettings(CommandLineFromInitializerList({"argv0", "--verbose=0"}), &settings)); |
| EXPECT_EQ(0, settings.min_log_level); |
| |
| EXPECT_TRUE( |
| ParseLogSettings(CommandLineFromInitializerList({"argv0", "--verbose=3"}), &settings)); |
| EXPECT_EQ(-3, settings.min_log_level); |
| |
| EXPECT_TRUE(ParseLogSettings(CommandLineFromInitializerList({"argv0", "--quiet=0"}), &settings)); |
| EXPECT_EQ(0, settings.min_log_level); |
| |
| EXPECT_TRUE(ParseLogSettings(CommandLineFromInitializerList({"argv0", "--quiet"}), &settings)); |
| EXPECT_EQ(1, settings.min_log_level); |
| |
| EXPECT_TRUE(ParseLogSettings(CommandLineFromInitializerList({"argv0", "--quiet=3"}), &settings)); |
| EXPECT_EQ(3, settings.min_log_level); |
| |
| EXPECT_TRUE(ParseLogSettings( |
| CommandLineFromInitializerList({"argv0", "--log-file=/tmp/custom.log"}), &settings)); |
| EXPECT_EQ("/tmp/custom.log", settings.log_file); |
| } |
| |
| TEST(LogSettings, ParseInvalidOptions) { |
| LogSettings settings; |
| settings.min_log_level = LOG_FATAL; |
| |
| EXPECT_FALSE( |
| ParseLogSettings(CommandLineFromInitializerList({"argv0", "--verbose=-1"}), &settings)); |
| EXPECT_EQ(LOG_FATAL, settings.min_log_level); |
| |
| EXPECT_FALSE(ParseLogSettings(CommandLineFromInitializerList({"argv0", "--verbose=123garbage"}), |
| &settings)); |
| EXPECT_EQ(LOG_FATAL, settings.min_log_level); |
| |
| EXPECT_FALSE( |
| ParseLogSettings(CommandLineFromInitializerList({"argv0", "--quiet=-1"}), &settings)); |
| EXPECT_EQ(LOG_FATAL, settings.min_log_level); |
| |
| EXPECT_FALSE( |
| ParseLogSettings(CommandLineFromInitializerList({"argv0", "--quiet=123garbage"}), &settings)); |
| EXPECT_EQ(LOG_FATAL, settings.min_log_level); |
| } |
| |
| TEST_F(LogSettingsFixture, SetValidOptions) { |
| EXPECT_TRUE( |
| SetLogSettingsFromCommandLine(CommandLineFromInitializerList({"argv0", "--verbose=20"}))); |
| |
| EXPECT_EQ(-20, GetMinLogLevel()); |
| } |
| |
| TEST_F(LogSettingsFixture, SetInvalidOptions) { |
| LogSeverity old_severity = GetMinLogLevel(); |
| |
| EXPECT_FALSE(SetLogSettingsFromCommandLine( |
| CommandLineFromInitializerList({"argv0", "--verbose=garbage"}))); |
| |
| EXPECT_EQ(old_severity, GetMinLogLevel()); |
| } |
| |
| TEST_F(LogSettingsFixture, ToArgv) { |
| LogSettings settings; |
| EXPECT_TRUE(LogSettingsToArgv(settings).empty()); |
| |
| EXPECT_TRUE(ParseLogSettings(CommandLineFromInitializerList({"argv0", "--quiet"}), &settings)); |
| EXPECT_TRUE(LogSettingsToArgv(settings) == std::vector<std::string>{"--quiet=1"}); |
| |
| EXPECT_TRUE(ParseLogSettings(CommandLineFromInitializerList({"argv0", "--quiet=3"}), &settings)); |
| EXPECT_TRUE(LogSettingsToArgv(settings) == std::vector<std::string>{"--quiet=3"}); |
| |
| EXPECT_TRUE(ParseLogSettings(CommandLineFromInitializerList({"argv0", "--verbose"}), &settings)); |
| EXPECT_TRUE(LogSettingsToArgv(settings) == std::vector<std::string>{"--verbose=1"}); |
| |
| EXPECT_TRUE( |
| ParseLogSettings(CommandLineFromInitializerList({"argv0", "--verbose=3"}), &settings)); |
| EXPECT_TRUE(LogSettingsToArgv(settings) == std::vector<std::string>{"--verbose=3"}); |
| |
| // Reset |settings| back to defaults so we don't pick up previous tests. |
| settings = LogSettings{}; |
| EXPECT_TRUE( |
| ParseLogSettings(CommandLineFromInitializerList({"argv0", "--log-file=/foo"}), &settings)); |
| EXPECT_TRUE(LogSettingsToArgv(settings) == std::vector<std::string>{"--log-file=/foo"}) |
| << LogSettingsToArgv(settings)[0]; |
| |
| EXPECT_TRUE(ParseLogSettings( |
| CommandLineFromInitializerList({"argv0", "--verbose", "--log-file=/foo"}), &settings)); |
| EXPECT_TRUE(LogSettingsToArgv(settings) == |
| (std::vector<std::string>{"--verbose=1", "--log-file=/foo"})); |
| } |
| |
| } // namespace |
| } // namespace fxl |