summaryrefslogtreecommitdiffstats
path: root/media/mtransport/test/gtest_ringbuffer_dumper.h
blob: 81eddbb2f6fa1c9c3137cbbfbf6a3b948d91bc0c (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
79
80
81
82
83
/* -*- 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) {
      mozilla::SyncRunnable::DispatchToThread(
          test_utils_->sts_target(),
          WrapRunnable(this, &RingbufferDumper::ClearRingBuffer_s));
    }

    virtual void OnTestEnd(const ::testing::TestInfo& testInfo) {
      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) {
      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__