summaryrefslogtreecommitdiffstats
path: root/dom/media/webrtc/transport/test/gtest_ringbuffer_dumper.h
blob: 25e85c2155cacbdd41ba016ac8b1338ac89c54b2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
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__