// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "extensions/test/result_catcher.h"

#include "base/run_loop.h"
#include "content/public/test/test_utils.h"

namespace extensions {

ResultCatcher::ResultCatcher() : browser_context_restriction_(nullptr) {
  test_api_observation_.Observe(TestApiObserverRegistry::GetInstance());
}

ResultCatcher::~ResultCatcher() = default;

bool ResultCatcher::GetNextResult() {
  // Depending on the tests, multiple results can come in from a single call to
  // RunLoop::Run() so we maintain a queue of results and just pull them off as
  // the test calls this, going to the run loop only when the queue is empty.
  if (results_.empty()) {
    base::RunLoop run_loop(base::RunLoop::Type::kNestableTasksAllowed);
    quit_closure_ = content::GetDeferredQuitTaskForRunLoop(&run_loop);
    run_loop.Run();
  }

  if (!results_.empty()) {
    bool ret = results_.front();
    results_.pop_front();
    message_ = messages_.front();
    messages_.pop_front();
    return ret;
  }

  NOTREACHED();
  return false;
}

void ResultCatcher::OnTestPassed(content::BrowserContext* browser_context) {
  if (browser_context_restriction_ &&
      browser_context != browser_context_restriction_) {
    return;
  }

  VLOG(1) << "Got chrome.test.notifyPass notification.";
  results_.push_back(true);
  messages_.push_back(std::string());
  if (!quit_closure_.is_null())
    std::move(quit_closure_).Run();
}

void ResultCatcher::OnTestFailed(content::BrowserContext* browser_context,
                                 const std::string& message) {
  if (browser_context_restriction_ &&
      browser_context != browser_context_restriction_) {
    return;
  }

  VLOG(1) << "Got chrome.test.notifyFail notification.";
  results_.push_back(false);
  messages_.push_back(message);
  if (!quit_closure_.is_null())
    std::move(quit_closure_).Run();
}

}  // namespace extensions