test_content_script_load_waiter.h

// Copyright 2021 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.

#ifndef EXTENSIONS_TEST_TEST_CONTENT_SCRIPT_LOAD_WAITER_H_
#define EXTENSIONS_TEST_TEST_CONTENT_SCRIPT_LOAD_WAITER_H_

#include "base/run_loop.h"
#include "base/scoped_observation.h"
#include "extensions/browser/user_script_loader.h"
#include "extensions/common/mojom/host_id.mojom.h"

namespace content {
class BrowserContext;
}

namespace extensions {

// A single-use class to wait for content scripts to be loaded.
class ContentScriptLoadWaiter : public UserScriptLoader::Observer {
 public:
  explicit ContentScriptLoadWaiter(UserScriptLoader* loader);
  ~ContentScriptLoadWaiter();
  ContentScriptLoadWaiter(const ContentScriptLoadWaiter& other) = delete;
  ContentScriptLoadWaiter& operator=(const ContentScriptLoadWaiter& other) =
      delete;

  // Waits until the observed UserScriptLoader completes a script load via the
  // OnScriptsLoaded event.
  void Wait();

 private:
  // UserScriptLoader::Observer:
  void OnScriptsLoaded(UserScriptLoader* loader,
                       content::BrowserContext* browser_context) override;
  void OnUserScriptLoaderDestroyed(UserScriptLoader* loader) override;

  mojom::HostID host_id_;

  base::RunLoop run_loop_;
  base::ScopedObservation<UserScriptLoader, UserScriptLoader::Observer>
      loader_observation_{this};
};

}  // namespace extensions

#endif  // EXTENSIONS_TEST_TEST_CONTENT_SCRIPT_LOAD_WAITER_H_