Enable startup tracing from command line flags in components/tracing

components/tracing/common/trace_config_file_unittest.cc

 // Copyright (c) 2015 The Chromium 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 "components/tracing/common/trace_config_file.h"
+
 #include "base/at_exit.h"
 #include "base/command_line.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
-#include "components/tracing/browser/trace_config_file.h"
 #include "components/tracing/common/tracing_switches.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace tracing {
 
 namespace {
 
 const char kTraceConfig[] =
   "{"
     "\"enable_argument_filter\":true,"
@@ skipping 22 matching lines @@
   if (!result_file.empty()) {
     if (content != "{")
       content += ",";
     content += "\"result_file\":\"" + result_file + "\"";
   }
 
   content += "}";
   return content;
 }
 
-} // namespace
+}  // namespace
 
 TEST(TraceConfigFileTest, TraceStartupEnabled) {
   base::ShadowingAtExitManager sem;
-  base::CommandLine::ForCurrentProcess()->AppendSwitch(
-      switches::kTraceStartup);
+  base::CommandLine::ForCurrentProcess()->AppendSwitch(switches::kTraceStartup);
   base::CommandLine::ForCurrentProcess()->AppendSwitch(
       switches::kTraceConfigFile);
 
   EXPECT_FALSE(TraceConfigFile::GetInstance()->IsEnabled());
 }
 
 TEST(TraceConfigFileTest, TraceShutdownEnabled) {
   base::ShadowingAtExitManager sem;
   base::CommandLine::ForCurrentProcess()->AppendSwitch(
       switches::kTraceShutdown);
@@ skipping 25 matching lines @@
   base::ShadowingAtExitManager sem;
   base::CommandLine::ForCurrentProcess()->AppendSwitchPath(
       switches::kTraceConfigFile,
       base::FilePath(FILE_PATH_LITERAL("invalid-trace-config-file-path")));
 
   EXPECT_FALSE(TraceConfigFile::GetInstance()->IsEnabled());
 }
 
 TEST(TraceConfigFileTest, ValidContent) {
   base::ShadowingAtExitManager sem;
-  std::string content = GetTraceConfigFileContent(
-      kTraceConfig, "10", "trace_result_file.log");
+  std::string content =
+      GetTraceConfigFileContent(kTraceConfig, "10", "trace_result_file.log");
 
   base::FilePath trace_config_file;
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   ASSERT_TRUE(
       base::CreateTemporaryFileInDir(temp_dir.GetPath(), &trace_config_file));
-  ASSERT_NE(-1, base::WriteFile(
-      trace_config_file, content.c_str(), (int)content.length()));
+  ASSERT_NE(-1, base::WriteFile(trace_config_file, content.c_str(),
+                                (int)content.length()));
   base::CommandLine::ForCurrentProcess()->AppendSwitchPath(
       switches::kTraceConfigFile, trace_config_file);
 
   ASSERT_TRUE(TraceConfigFile::GetInstance()->IsEnabled());
   EXPECT_STREQ(
       kTraceConfig,
       TraceConfigFile::GetInstance()->GetTraceConfig().ToString().c_str());
   EXPECT_EQ(10, TraceConfigFile::GetInstance()->GetStartupDuration());
   EXPECT_EQ(base::FilePath(FILE_PATH_LITERAL("trace_result_file.log")),
             TraceConfigFile::GetInstance()->GetResultFile());
 }
 
 TEST(TraceConfigFileTest, ValidContentWithOnlyTraceConfig) {
   base::ShadowingAtExitManager sem;
   std::string content = GetTraceConfigFileContent(kTraceConfig, "", "");
 
   base::FilePath trace_config_file;
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   ASSERT_TRUE(
       base::CreateTemporaryFileInDir(temp_dir.GetPath(), &trace_config_file));
-  ASSERT_NE(-1, base::WriteFile(
-      trace_config_file, content.c_str(), (int)content.length()));
+  ASSERT_NE(-1, base::WriteFile(trace_config_file, content.c_str(),
+                                (int)content.length()));
   base::CommandLine::ForCurrentProcess()->AppendSwitchPath(
       switches::kTraceConfigFile, trace_config_file);
 
   ASSERT_TRUE(TraceConfigFile::GetInstance()->IsEnabled());
   EXPECT_STREQ(
       kTraceConfig,
       TraceConfigFile::GetInstance()->GetTraceConfig().ToString().c_str());
   EXPECT_EQ(0, TraceConfigFile::GetInstance()->GetStartupDuration());
   EXPECT_EQ(base::FilePath(FILE_PATH_LITERAL("chrometrace.log")),
             TraceConfigFile::GetInstance()->GetResultFile());
 }
 
 TEST(TraceConfigFileTest, ContentWithAbsoluteResultFilePath) {
   base::ShadowingAtExitManager sem;
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
 
   const base::FilePath result_file_path =
       temp_dir.GetPath().Append(FILE_PATH_LITERAL("trace_result_file.log"));
   ASSERT_TRUE(result_file_path.IsAbsolute());
 
   std::string result_file_path_str = result_file_path.AsUTF8Unsafe();
-  auto it = std::find(
-      result_file_path_str.begin(), result_file_path_str.end(), '\\');
+  auto it =
+      std::find(result_file_path_str.begin(), result_file_path_str.end(), '\\');
   while (it != result_file_path_str.end()) {
     auto it2 = result_file_path_str.insert(it, '\\');
-    it = std::find(it2+2, result_file_path_str.end(), '\\');
+    it = std::find(it2 + 2, result_file_path_str.end(), '\\');
   }
-  std::string content = GetTraceConfigFileContent(
-      kTraceConfig, "10", result_file_path_str);
+  std::string content =
+      GetTraceConfigFileContent(kTraceConfig, "10", result_file_path_str);
 
   base::FilePath trace_config_file;
   ASSERT_TRUE(
       base::CreateTemporaryFileInDir(temp_dir.GetPath(), &trace_config_file));
-  ASSERT_NE(-1, base::WriteFile(
-      trace_config_file, content.c_str(), (int)content.length()));
+  ASSERT_NE(-1, base::WriteFile(trace_config_file, content.c_str(),
+                                (int)content.length()));
   base::CommandLine::ForCurrentProcess()->AppendSwitchPath(
       switches::kTraceConfigFile, trace_config_file);
 
   ASSERT_TRUE(TraceConfigFile::GetInstance()->IsEnabled());
   EXPECT_EQ(result_file_path, TraceConfigFile::GetInstance()->GetResultFile());
 }
 
 TEST(TraceConfigFileTest, ContentWithNegtiveDuration) {
   base::ShadowingAtExitManager sem;
   std::string content = GetTraceConfigFileContent(kTraceConfig, "-1", "");
 
   base::FilePath trace_config_file;
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   ASSERT_TRUE(
       base::CreateTemporaryFileInDir(temp_dir.GetPath(), &trace_config_file));
-  ASSERT_NE(-1, base::WriteFile(
-      trace_config_file, content.c_str(), (int)content.length()));
+  ASSERT_NE(-1, base::WriteFile(trace_config_file, content.c_str(),
+                                (int)content.length()));
   base::CommandLine::ForCurrentProcess()->AppendSwitchPath(
       switches::kTraceConfigFile, trace_config_file);
 
   ASSERT_TRUE(TraceConfigFile::GetInstance()->IsEnabled());
   EXPECT_STREQ(
       kTraceConfig,
       TraceConfigFile::GetInstance()->GetTraceConfig().ToString().c_str());
   EXPECT_EQ(0, TraceConfigFile::GetInstance()->GetStartupDuration());
   EXPECT_EQ(base::FilePath(FILE_PATH_LITERAL("chrometrace.log")),
             TraceConfigFile::GetInstance()->GetResultFile());
 }
 
 TEST(TraceConfigFileTest, ContentWithoutTraceConfig) {
   base::ShadowingAtExitManager sem;
-  std::string content = GetTraceConfigFileContent(
-      "", "10", "trace_result_file.log");
+  std::string content =
+      GetTraceConfigFileContent("", "10", "trace_result_file.log");
 
   base::FilePath trace_config_file;
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   ASSERT_TRUE(
       base::CreateTemporaryFileInDir(temp_dir.GetPath(), &trace_config_file));
-  ASSERT_NE(-1, base::WriteFile(
-      trace_config_file, content.c_str(), (int)content.length()));
+  ASSERT_NE(-1, base::WriteFile(trace_config_file, content.c_str(),
+                                (int)content.length()));
   base::CommandLine::ForCurrentProcess()->AppendSwitchPath(
       switches::kTraceConfigFile, trace_config_file);
 
   EXPECT_FALSE(TraceConfigFile::GetInstance()->IsEnabled());
 }
 
 TEST(TraceConfigFileTest, InvalidContent) {
   base::ShadowingAtExitManager sem;
   std::string content = "invalid trace config file content";
 
   base::FilePath trace_config_file;
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   ASSERT_TRUE(
       base::CreateTemporaryFileInDir(temp_dir.GetPath(), &trace_config_file));
-  ASSERT_NE(-1, base::WriteFile(
-      trace_config_file, content.c_str(), (int)content.length()));
+  ASSERT_NE(-1, base::WriteFile(trace_config_file, content.c_str(),
+                                (int)content.length()));
   base::CommandLine::ForCurrentProcess()->AppendSwitchPath(
       switches::kTraceConfigFile, trace_config_file);
 
   EXPECT_FALSE(TraceConfigFile::GetInstance()->IsEnabled());
 }
 
 TEST(TraceConfigFileTest, EmptyContent) {
   base::ShadowingAtExitManager sem;
   base::FilePath trace_config_file;
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   ASSERT_TRUE(
       base::CreateTemporaryFileInDir(temp_dir.GetPath(), &trace_config_file));
   base::CommandLine::ForCurrentProcess()->AppendSwitchPath(
       switches::kTraceConfigFile, trace_config_file);
 
   EXPECT_FALSE(TraceConfigFile::GetInstance()->IsEnabled());
 }
 
-} // namespace tracing
+}  // namespace tracing