commit e41ed9d0ef0de03e76aaa13cec74003598af8224
author Philipp Schrader <philipp.schrader@gmail.com> Sun Mar 15 22:57:13 2015 +0000
committer Philipp Schrader <philipp.schrader@gmail.com> Sat Mar 28 15:04:28 2015 +0000
tree 132b0f647385c6dceada127c1c93c7f250bf6dd6
parent 16032d878f7328f74c92d95704ef2737f39d830b

Add log-related command-line options to unit tests.

You can now specify the following when running a unit test:
  -p, --print-logs
      Print the log messages as they are being generated.
  -o, --log-file=FILE
      Print all log messages to FILE instead of standard output

Also, when a test fails the messages no longer get dumped to standard
output. They only get dumped when specifying --print-logs.

Change-Id: Icc265734f302c27c24829409ca6f7c8fce2979d4

aos/common/queue_testutils.cc

diff --git a/aos/common/queue_testutils.cc b/aos/common/queue_testutils.cc
index 824cb89..d71d023 100644
--- a/aos/common/queue_testutils.cc
+++ b/aos/common/queue_testutils.cc
@@ -41,22 +41,41 @@
     }
   }
 
+  void SetOutputFile(const char *filename) {
+    if (strcmp("-", filename) != 0) {
+      FILE *newfile = fopen(filename, "w");
+
+      if (newfile) {
+        output_file_ = newfile;
+      }
+    }
+  }
+
+  void PrintMessagesAsTheyComeIn() { print_as_messages_come_in_ = true; }
+
  private:
   TestLogImplementation() {}
+  ~TestLogImplementation() {
+    if (output_file_ != stdout) {
+      fclose(output_file_);
+    }
+  }
 
   static TestLogImplementation *CreateInstance() {
     return new TestLogImplementation();
   }
 
   virtual void HandleMessage(const LogMessage &message) override {
-    if (message.level == FATAL) {
-      logging::internal::PrintMessage(stdout, message);
+    if (message.level == FATAL || print_as_messages_come_in_) {
+      logging::internal::PrintMessage(output_file_, message);
     }
 
     messages_.push_back(message);
   }
 
   ::std::vector<LogMessage> messages_;
+  bool print_as_messages_come_in_ = false;
+  FILE *output_file_ = stdout;
 };
 
 class MyTestEventListener : public ::testing::EmptyTestEventListener {
@@ -65,13 +84,8 @@
   }
   virtual void OnTestEnd(const ::testing::TestInfo &test_info) {
     if (test_info.result()->Failed()) {
-      printf("Test %s failed. Printing out all log messages.\n",
+      printf("Test %s failed. Use '--print-logs' to see all log messages.\n",
              test_info.name());
-      fputs("\tThis will include already printed WARNING and up messages.\n",
-            stdout);
-      fputs("\tIt will also include duplicates of all gtest failures.\n",
-            stdout);
-      TestLogImplementation::GetInstance()->PrintAllMessages();
     }
   }
 
@@ -145,6 +159,14 @@
   CHECK_EQ(atexit(TerminateExitHandler), 0);
 }
 
+void SetLogFileName(const char* filename) {
+  TestLogImplementation::GetInstance()->SetOutputFile(filename);
+}
+
+void ForcePrintLogsDuringTests() {
+  TestLogImplementation::GetInstance()->PrintMessagesAsTheyComeIn();
+}
+
 }  // namespace testing
 }  // namespace common
 }  // namespace aos