1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59


content / public / test / test_web_ui_listener_observer.cc [blame]

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

#include "content/public/test/test_web_ui_listener_observer.h"

#include "base/functional/bind.h"
#include "base/functional/callback.h"
#include "content/public/test/test_web_ui.h"

namespace content {

TestWebUIListenerObserver::TestWebUIListenerObserver(
    content::TestWebUI* web_ui,
    const std::string& listener_name)
    : web_ui_(web_ui), listener_name_(listener_name) {
  web_ui_->AddJavascriptCallObserver(this);
}

TestWebUIListenerObserver::~TestWebUIListenerObserver() {
  web_ui_->RemoveJavascriptCallObserver(this);
}

void TestWebUIListenerObserver::Wait() {
  run_loop_.Run();
}

void TestWebUIListenerObserver::OnJavascriptFunctionCalled(
    const TestWebUI::CallData& call_data) {
  // Ignore subsequent calls after a call matched the listener that this
  // observer is waiting for.
  if (call_args_.has_value())
    return;

  // See WebUIMessageHandler::FireWebUIListener.
  if (call_data.function_name() != "cr.webUIListenerCallback")
    return;
  if (!call_data.arg1() || !call_data.arg1()->is_string() ||
      call_data.arg1()->GetString() != listener_name_) {
    return;
  }

  call_args_.emplace();
  // TestWebUI::CallData supports up to 4 arguments, each of them has a
  // different accessor. arg1() is the listener name for WebUI listeners (see
  // above).
  if (call_data.arg2()) {
    call_args_->Append(call_data.arg2()->Clone());
  }
  if (call_data.arg3()) {
    call_args_->Append(call_data.arg3()->Clone());
  }
  if (call_data.arg4()) {
    call_args_->Append(call_data.arg4()->Clone());
  }
  run_loop_.Quit();
}

}  // namespace content