chrome/browser/chromeos/login/test/js_checker.cc

   1 // Copyright 2013 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 "chrome/browser/chromeos/login/test/js_checker.h"
   6
   7 #include "base/strings/utf_string_conversions.h"
   8 #include "content/public/browser/render_frame_host.h"
   9 #include "content/public/browser/web_contents.h"
  10 #include "content/public/test/browser_test_utils.h"
  11 #include "content/public/test/test_utils.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 namespace {
  15
  16 std::string WrapSend(const std::string& expression) {
  17   return "window.domAutomationController.send(" + expression + ")";
  18 }
  19
  20 }  // namespace
  21
  22 namespace chromeos {
  23 namespace test {
  24
  25 JSChecker::JSChecker() : web_contents_(NULL) {}
  26
  27 JSChecker::JSChecker(content::WebContents* web_contents)
  28     : web_contents_(web_contents) {
  29 }
  30
  31 void JSChecker::Evaluate(const std::string& expression) {
  32   CHECK(web_contents_);
  33   ASSERT_TRUE(content::ExecuteScript(web_contents_, expression));
  34 }
  35
  36 void JSChecker::ExecuteAsync(const std::string& expression) {
  37   CHECK(web_contents_);
  38   std::string new_script = expression + ";";
  39   web_contents_->GetMainFrame()->ExecuteJavaScriptWithUserGestureForTests(
  40       base::UTF8ToUTF16(new_script));
  41 }
  42
  43 bool JSChecker::GetBool(const std::string& expression) {
  44   bool result;
  45   GetBoolImpl(expression, &result);
  46   return result;
  47 }
  48
  49 int JSChecker::GetInt(const std::string& expression) {
  50   int result;
  51   GetIntImpl(expression, &result);
  52   return result;
  53 }
  54
  55 std::string JSChecker::GetString(const std::string& expression) {
  56   std::string result;
  57   GetStringImpl(expression, &result);
  58   return result;
  59 }
  60
  61 void JSChecker::ExpectTrue(const std::string& expression) {
  62   EXPECT_TRUE(GetBool(expression)) << expression;
  63 }
  64
  65 void JSChecker::ExpectFalse(const std::string& expression) {
  66   EXPECT_FALSE(GetBool(expression)) << expression;
  67 }
  68
  69 void JSChecker::ExpectEQ(const std::string& expression, int result) {
  70   EXPECT_EQ(GetInt(expression), result) << expression;
  71 }
  72
  73 void JSChecker::ExpectNE(const std::string& expression, int result) {
  74   EXPECT_NE(GetInt(expression), result) << expression;
  75 }
  76
  77 void JSChecker::ExpectEQ(const std::string& expression,
  78                          const std::string& result) {
  79   EXPECT_EQ(GetString(expression), result) << expression;
  80 }
  81
  82 void JSChecker::ExpectNE(const std::string& expression,
  83                          const std::string& result) {
  84   EXPECT_NE(GetString(expression), result) << expression;
  85 }
  86
  87 void JSChecker::GetBoolImpl(const std::string& expression, bool* result) {
  88   CHECK(web_contents_);
  89   ASSERT_TRUE(content::ExecuteScriptAndExtractBool(web_contents_,
  90                                                    "!!" + WrapSend(expression),
  91                                                    result));
  92 }
  93
  94 void JSChecker::GetIntImpl(const std::string& expression, int* result) {
  95   CHECK(web_contents_);
  96   ASSERT_TRUE(content::ExecuteScriptAndExtractInt(web_contents_,
  97                                                   WrapSend(expression),
  98                                                   result));
  99 }
 100
 101 void JSChecker::GetStringImpl(const std::string& expression,
 102                               std::string* result) {
 103   CHECK(web_contents_);
 104   ASSERT_TRUE(content::ExecuteScriptAndExtractString(web_contents_,
 105                                                      WrapSend(expression),
 106                                                      result));
 107 }
 108
 109 }  // namespace test
 110 }  // namespace chromeos