content/public/test/test_web_ui.cc

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/logging.h"
   6 #include "content/public/browser/web_ui_message_handler.h"
   7 #include "content/public/test/test_web_ui.h"
   8
   9 namespace content {
  10
  11 TestWebUI::TestWebUI() : web_contents_(nullptr) {
  12 }
  13
  14 TestWebUI::~TestWebUI() {
  15   ClearTrackedCalls();
  16 }
  17
  18 void TestWebUI::ClearTrackedCalls() {
  19   call_data_.clear();
  20 }
  21
  22 WebContents* TestWebUI::GetWebContents() const {
  23   return web_contents_;
  24 }
  25
  26 WebUIController* TestWebUI::GetController() const {
  27   return nullptr;
  28 }
  29
  30 float TestWebUI::GetDeviceScaleFactor() const {
  31   return 1.0f;
  32 }
  33
  34 const base::string16& TestWebUI::GetOverriddenTitle() const {
  35   return temp_string_;
  36 }
  37
  38 ui::PageTransition TestWebUI::GetLinkTransitionType() const {
  39   return ui::PAGE_TRANSITION_LINK;
  40 }
  41
  42 int TestWebUI::GetBindings() const {
  43   return 0;
  44 }
  45
  46 void TestWebUI::AddMessageHandler(WebUIMessageHandler* handler) {
  47   handlers_.push_back(handler);
  48 }
  49
  50 void TestWebUI::CallJavascriptFunction(const std::string& function_name) {
  51   call_data_.push_back(new CallData(function_name));
  52 }
  53
  54 void TestWebUI::CallJavascriptFunction(const std::string& function_name,
  55                                        const base::Value& arg1) {
  56   call_data_.push_back(new CallData(function_name));
  57   call_data_.back()->TakeAsArg1(arg1.DeepCopy());
  58 }
  59
  60 void TestWebUI::CallJavascriptFunction(const std::string& function_name,
  61                                        const base::Value& arg1,
  62                                        const base::Value& arg2) {
  63   call_data_.push_back(new CallData(function_name));
  64   call_data_.back()->TakeAsArg1(arg1.DeepCopy());
  65   call_data_.back()->TakeAsArg2(arg2.DeepCopy());
  66 }
  67
  68 void TestWebUI::CallJavascriptFunction(const std::string& function_name,
  69                                        const base::Value& arg1,
  70                                        const base::Value& arg2,
  71                                        const base::Value& arg3) {
  72   NOTREACHED();
  73 }
  74
  75 void TestWebUI::CallJavascriptFunction(const std::string& function_name,
  76                                        const base::Value& arg1,
  77                                        const base::Value& arg2,
  78                                        const base::Value& arg3,
  79                                        const base::Value& arg4) {
  80   NOTREACHED();
  81 }
  82
  83 void TestWebUI::CallJavascriptFunction(
  84     const std::string& function_name,
  85     const std::vector<const base::Value*>& args) {
  86   NOTREACHED();
  87 }
  88
  89 TestWebUI::CallData::CallData(const std::string& function_name)
  90     : function_name_(function_name) {
  91 }
  92
  93 TestWebUI::CallData::~CallData() {
  94 }
  95
  96 void TestWebUI::CallData::TakeAsArg1(base::Value* arg) {
  97   arg1_.reset(arg);
  98 }
  99
 100 void TestWebUI::CallData::TakeAsArg2(base::Value* arg) {
 101   arg2_.reset(arg);
 102 }
 103
 104 }  // namespace content