extensions/test/test_background_page_first_load_observer.cc

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

#include "extensions/test/test_background_page_first_load_observer.h"

#include "base/logging.h"
#include "extensions/browser/extension_host.h"
#include "extensions/browser/process_manager.h"

namespace extensions {

TestBackgroundPageFirstLoadObserver::TestBackgroundPageFirstLoadObserver(
    content::BrowserContext* browser_context,
    const ExtensionId& extension_id)
    : extension_id_(extension_id),
      process_manager_(ProcessManager::Get(browser_context)),
      process_manager_observer_(this) {
  process_manager_observer_.Add(process_manager_);
  extension_host_ =
      process_manager_->GetBackgroundHostForExtension(extension_id_);
  if (extension_host_)
    OnObtainedExtensionHost();
}

TestBackgroundPageFirstLoadObserver::~TestBackgroundPageFirstLoadObserver() {
  if (extension_host_) {
    static_cast<DeferredStartRenderHost*>(extension_host_)
        ->RemoveDeferredStartRenderHostObserver(this);
  }
}

void TestBackgroundPageFirstLoadObserver::Wait() {
  if (!extension_host_ || !extension_host_->has_loaded_once())
    run_loop_.Run();
}

void TestBackgroundPageFirstLoadObserver::OnBackgroundHostCreated(
    ExtensionHost* host) {
  if (host->extension_id() == extension_id_) {
    DCHECK(!extension_host_);
    extension_host_ = host;
    OnObtainedExtensionHost();
  }
}

void TestBackgroundPageFirstLoadObserver::
    OnDeferredStartRenderHostDidStopFirstLoad(
        const DeferredStartRenderHost* host) {
  run_loop_.Quit();
}

void TestBackgroundPageFirstLoadObserver::OnObtainedExtensionHost() {
  static_cast<DeferredStartRenderHost*>(extension_host_)
      ->AddDeferredStartRenderHostObserver(this);
}

}  // namespace extensions