/* -*- 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__