Source code

Revision control

Copy as Markdown

Other Tools

/* -*- 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 {
running_ = true;
mozilla::SyncRunnable::DispatchToThread(
test_utils_->sts_target(),
WrapRunnable(this, &RingbufferDumper::ClearRingBuffer_s));
}
virtual void OnTestEnd(const ::testing::TestInfo& testInfo) override {
running_ = false;
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()) {
if (!running_) {
// Why does gtest sometimes call this without ever calling OnTestStart?
running_ = true;
mozilla::SyncRunnable::DispatchToThread(
test_utils_->sts_target(),
WrapRunnable(this, &RingbufferDumper::ClearRingBuffer_s));
}
// Dump (and empty) the RLogConnector
mozilla::SyncRunnable::DispatchToThread(
test_utils_->sts_target(),
WrapRunnable(this, &RingbufferDumper::DumpRingBuffer_s));
}
}
private:
MtransportTestUtils* test_utils_;
bool running_ = false;
};
} // namespace test
#endif // gtest_ringbuffer_dumper_h__