1 files changed, 183 insertions, 0 deletions
diff --git a/xpcom/tests/gtest/TestLogCommandLineHandler.cpp b/xpcom/tests/gtest/TestLogCommandLineHandler.cpp
new file mode 100644
index 0000000000..ebec4854dd
--- /dev/null
+++ b/xpcom/tests/gtest/TestLogCommandLineHandler.cpp
@@ -0,0 +1,183 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "LogCommandLineHandler.h"
+
+#include <iterator>
+#include "nsString.h"
+#include "nsTArray.h"
+#include "mozilla/UniquePtr.h"
+#include "mozilla/UniquePtrExtensions.h"
+#include "gtest/gtest.h"
+
+using namespace mozilla;
+
+template <class T, size_t N>
+constexpr size_t array_size(T (&)[N]) {
+  return N;
+}
+
+TEST(LogCommandLineHandler, Empty)
+{
+  bool callbackInvoked = false;
+  auto callback = [&](nsACString const& env) mutable {
+    callbackInvoked = true;
+  };
+
+  mozilla::LoggingHandleCommandLineArgs(0, nullptr, callback);
+  EXPECT_FALSE(callbackInvoked);
+
+  char const* argv1[] = {""};
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv1), argv1, callback);
+  EXPECT_FALSE(callbackInvoked);
+}
+
+TEST(LogCommandLineHandler, MOZ_LOG_regular)
+{
+  nsTArray<nsCString> results;
+
+  auto callback = [&](nsACString const& env) mutable {
+    results.AppendElement(env);
+  };
+
+  char const* argv1[] = {"", "-MOZ_LOG", "module1:5,module2:4,sync,timestamp"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv1), argv1, callback);
+  EXPECT_TRUE(results.Length() == 1);
+  EXPECT_TRUE(
+      "MOZ_LOG=module1:5,module2:4,sync,timestamp"_ns.Equals(results[0]));
+
+  char const* argv2[] = {"", "-MOZ_LOG=modules"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv2), argv2, callback);
+  EXPECT_TRUE(results.Length() == 1);
+  EXPECT_TRUE("MOZ_LOG=modules"_ns.Equals(results[0]));
+
+  char const* argv3[] = {"", "--MOZ_LOG", "modules"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv3), argv3, callback);
+  EXPECT_TRUE(results.Length() == 1);
+  EXPECT_TRUE("MOZ_LOG=modules"_ns.Equals(results[0]));
+
+  char const* argv4[] = {"", "--MOZ_LOG=modules"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv4), argv4, callback);
+  EXPECT_TRUE(results.Length() == 1);
+  EXPECT_TRUE("MOZ_LOG=modules"_ns.Equals(results[0]));
+}
+
+TEST(LogCommandLineHandler, MOZ_LOG_and_FILE_regular)
+{
+  nsTArray<nsCString> results;
+
+  auto callback = [&](nsACString const& env) mutable {
+    results.AppendElement(env);
+  };
+
+  char const* argv1[] = {"", "-MOZ_LOG", "modules", "-MOZ_LOG_FILE",
+                         "c:\\file/path"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv1), argv1, callback);
+  EXPECT_TRUE(results.Length() == 2);
+  EXPECT_TRUE("MOZ_LOG=modules"_ns.Equals(results[0]));
+  EXPECT_TRUE("MOZ_LOG_FILE=c:\\file/path"_ns.Equals(results[1]));
+
+  char const* argv2[] = {"", "-MOZ_LOG=modules", "-MOZ_LOG_FILE=file"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv2), argv2, callback);
+  EXPECT_TRUE(results.Length() == 2);
+  EXPECT_TRUE("MOZ_LOG=modules"_ns.Equals(results[0]));
+  EXPECT_TRUE("MOZ_LOG_FILE=file"_ns.Equals(results[1]));
+
+  char const* argv3[] = {"", "--MOZ_LOG", "modules", "--MOZ_LOG_FILE", "file"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv3), argv3, callback);
+  EXPECT_TRUE(results.Length() == 2);
+  EXPECT_TRUE("MOZ_LOG=modules"_ns.Equals(results[0]));
+  EXPECT_TRUE("MOZ_LOG_FILE=file"_ns.Equals(results[1]));
+
+  char const* argv4[] = {"", "--MOZ_LOG=modules", "--MOZ_LOG_FILE=file"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv4), argv4, callback);
+  EXPECT_TRUE(results.Length() == 2);
+  EXPECT_TRUE("MOZ_LOG=modules"_ns.Equals(results[0]));
+  EXPECT_TRUE("MOZ_LOG_FILE=file"_ns.Equals(results[1]));
+
+  char const* argv5[] = {"",    "--MOZ_LOG",      "modules", "-P",
+                         "foo", "--MOZ_LOG_FILE", "file"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv5), argv5, callback);
+  EXPECT_TRUE(results.Length() == 2);
+  EXPECT_TRUE("MOZ_LOG=modules"_ns.Equals(results[0]));
+  EXPECT_TRUE("MOZ_LOG_FILE=file"_ns.Equals(results[1]));
+}
+
+TEST(LogCommandLineHandler, MOZ_LOG_fuzzy)
+{
+  nsTArray<nsCString> results;
+
+  auto callback = [&](nsACString const& env) mutable {
+    results.AppendElement(env);
+  };
+
+  char const* argv1[] = {"", "-MOZ_LOG"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv1), argv1, callback);
+  EXPECT_TRUE(results.Length() == 0);
+
+  char const* argv2[] = {"", "modules"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv2), argv2, callback);
+  EXPECT_TRUE(results.Length() == 0);
+
+  char const* argv3[] = {"", "-MOZ_LOG,modules", "-MOZ_LOG"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv3), argv3, callback);
+  EXPECT_TRUE(results.Length() == 0);
+
+  char const* argv4[] = {"", "-MOZ_LOG", "-MOZ_LOG", "-MOZ_LOG"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv4), argv4, callback);
+  EXPECT_TRUE(results.Length() == 0);
+
+  char const* argv5[] = {"", "-MOZ_LOG", "-diffent_command", "modules"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv5), argv5, callback);
+  EXPECT_TRUE(results.Length() == 0);
+}
+
+TEST(LogCommandLineHandler, MOZ_LOG_overlapping)
+{
+  nsTArray<nsCString> results;
+
+  auto callback = [&](nsACString const& env) mutable {
+    results.AppendElement(env);
+  };
+
+  char const* argv1[] = {"", "-MOZ_LOG=modules1", "-MOZ_LOG=modules2"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv1), argv1, callback);
+  EXPECT_TRUE(results.Length() == 2);
+  EXPECT_TRUE("MOZ_LOG=modules1"_ns.Equals(results[0]));
+  EXPECT_TRUE("MOZ_LOG=modules2"_ns.Equals(results[1]));
+
+  char const* argv2[] = {"", "-MOZ_LOG", "--MOZ_LOG", "modules"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv2), argv2, callback);
+  EXPECT_TRUE(results.Length() == 1);
+  EXPECT_TRUE("MOZ_LOG=modules"_ns.Equals(results[0]));
+
+  char const* argv3[] = {"", "-MOZ_LOG_FILE", "-MOZ_LOG", "modules"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv3), argv3, callback);
+  EXPECT_TRUE(results.Length() == 1);
+  EXPECT_TRUE("MOZ_LOG=modules"_ns.Equals(results[0]));
+
+  char const* argv4[] = {"", "-MOZ_LOG", "-MOZ_LOG_FILE", "-MOZ_LOG"};
+  results.Clear();
+  mozilla::LoggingHandleCommandLineArgs(array_size(argv4), argv4, callback);
+  EXPECT_TRUE(results.Length() == 0);
+}