/* 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 "rlogconnector.h"
#include "runnable_utils.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_