1 files changed, 78 insertions, 0 deletions
diff --git a/dom/media/webrtc/transport/test/gtest_ringbuffer_dumper.h b/dom/media/webrtc/transport/test/gtest_ringbuffer_dumper.h
new file mode 100644
index 0000000000..25e85c2155
--- /dev/null
+++ b/dom/media/webrtc/transport/test/gtest_ringbuffer_dumper.h
@@ -0,0 +1,78 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=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/. */
+
+// Original author: bcampen@mozilla.com
+
+#ifndef gtest_ringbuffer_dumper_h__
+#define gtest_ringbuffer_dumper_h__
+
+#include "mozilla/SyncRunnable.h"
+
+#define GTEST_HAS_RTTI 0
+#include "gtest/gtest.h"
+
+#include "mtransport_test_utils.h"
+#include "runnable_utils.h"
+#include "rlogconnector.h"
+
+using mozilla::RLogConnector;
+using mozilla::WrapRunnable;
+
+namespace test {
+class RingbufferDumper : public ::testing::EmptyTestEventListener {
+ public:
+  explicit RingbufferDumper(MtransportTestUtils* test_utils)
+      : test_utils_(test_utils) {}
+
+  void ClearRingBuffer_s() {
+    RLogConnector::CreateInstance();
+    // Set limit to zero to clear the ringbuffer
+    RLogConnector::GetInstance()->SetLogLimit(0);
+    RLogConnector::GetInstance()->SetLogLimit(UINT32_MAX);
+  }
+
+  void DestroyRingBuffer_s() { RLogConnector::DestroyInstance(); }
+
+  void DumpRingBuffer_s() {
+    std::deque<std::string> logs;
+    // Get an unlimited number of log lines, with no filter
+    RLogConnector::GetInstance()->GetAny(0, &logs);
+    for (auto l = logs.begin(); l != logs.end(); ++l) {
+      std::cout << *l << std::endl;
+    }
+    ClearRingBuffer_s();
+  }
+
+  virtual void OnTestStart(const ::testing::TestInfo& testInfo) override {
+    mozilla::SyncRunnable::DispatchToThread(
+        test_utils_->sts_target(),
+        WrapRunnable(this, &RingbufferDumper::ClearRingBuffer_s));
+  }
+
+  virtual void OnTestEnd(const ::testing::TestInfo& testInfo) override {
+    mozilla::SyncRunnable::DispatchToThread(
+        test_utils_->sts_target(),
+        WrapRunnable(this, &RingbufferDumper::DestroyRingBuffer_s));
+  }
+
+  // Called after a failed assertion or a SUCCEED() invocation.
+  virtual void OnTestPartResult(
+      const ::testing::TestPartResult& testResult) override {
+    if (testResult.failed()) {
+      // Dump (and empty) the RLogConnector
+      mozilla::SyncRunnable::DispatchToThread(
+          test_utils_->sts_target(),
+          WrapRunnable(this, &RingbufferDumper::DumpRingBuffer_s));
+    }
+  }
+
+ private:
+  MtransportTestUtils* test_utils_;
+};
+
+}  // namespace test
+
+#endif  // gtest_ringbuffer_dumper_h__