headless/test/headless_protocol_browsertest.h

// Copyright 2020 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 HEADLESS_TEST_HEADLESS_PROTOCOL_BROWSERTEST_H_
#define HEADLESS_TEST_HEADLESS_PROTOCOL_BROWSERTEST_H_

#include <memory>
#include <string>
#include <vector>

#include "content/public/test/browser_test.h"
#include "headless/public/devtools/domains/runtime.h"
#include "headless/public/headless_devtools_client.h"
#include "headless/test/headless_browser_test.h"

namespace headless {

class HeadlessProtocolBrowserTest
    : public HeadlessAsyncDevTooledBrowserTest,
      public HeadlessDevToolsClient::RawProtocolListener,
      public runtime::ExperimentalObserver {
 public:
  HeadlessProtocolBrowserTest();

 protected:
  void SetUpCommandLine(base::CommandLine* command_line) override;

  virtual std::vector<std::string> GetPageUrlExtraParams();

 private:
  // HeadlessWebContentsObserver implementation.
  void RunDevTooledTest() override;
  void BindingCreated(std::unique_ptr<headless::runtime::AddBindingResult>);

  // runtime::Observer implementation.
  void OnBindingCalled(const runtime::BindingCalledParams& params) override;

  // HeadlessDevToolsClient::RawProtocolListener
  bool OnProtocolMessage(base::span<const uint8_t> json_message,
                         const base::DictionaryValue& parsed_message) override;

  void SendMessageToJS(base::StringPiece message);
  void FinishTest();

 protected:
  bool test_finished_ = false;
  std::string test_folder_;
  std::string script_name_;
};

}  // namespace headless

#endif  // HEADLESS_TEST_HEADLESS_PROTOCOL_BROWSERTEST_H_