Home | History | Annotate | Download | only in base 
      1 // Copyright 2014 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/test/base/extension_js_browser_test.h"
      6 
      7 #include "base/callback.h"
      8 #include "base/json/json_reader.h"
      9 #include "base/strings/string_util.h"
     10 #include "base/strings/utf_string_conversions.h"
     11 #include "base/values.h"
     12 #include "chrome/browser/extensions/browsertest_util.h"
     13 #include "chrome/browser/profiles/profile.h"
     14 #include "content/public/browser/web_contents.h"
     15 #include "content/public/test/browser_test_utils.h"
     16 
     17 ExtensionJSBrowserTest::ExtensionJSBrowserTest() : libs_loaded_(false) {
     18 }
     19 
     20 ExtensionJSBrowserTest::~ExtensionJSBrowserTest() {
     21 }
     22 
     23 void ExtensionJSBrowserTest::WaitForExtension(const char* extension_id,
     24                                               const base::Closure& load_cb) {
     25   load_waiter_.reset(new ExtensionLoadWaiterOneShot());
     26   load_waiter_->WaitForExtension(extension_id, load_cb);
     27 }
     28 
     29 bool ExtensionJSBrowserTest::RunJavascriptTestF(bool is_async,
     30                                                 const std::string& test_fixture,
     31                                                 const std::string& test_name) {
     32   EXPECT_TRUE(load_waiter_->browser_context());
     33   if (!load_waiter_->browser_context())
     34     return false;
     35   ConstValueVector args;
     36   args.push_back(new base::StringValue(test_fixture));
     37   args.push_back(new base::StringValue(test_name));
     38   std::vector<base::string16> scripts;
     39   if (!libs_loaded_) {
     40     BuildJavascriptLibraries(&scripts);
     41     libs_loaded_ = true;
     42   }
     43   scripts.push_back(BuildRunTestJSCall(is_async, "RUN_TEST_F", args));
     44 
     45   base::string16 script_16 = JoinString(scripts, '\n');
     46   std::string script = base::UTF16ToUTF8(script_16);
     47 
     48   std::string result =
     49       extensions::browsertest_util::ExecuteScriptInBackgroundPage(
     50           Profile::FromBrowserContext(load_waiter_->browser_context()),
     51           load_waiter_->extension_id(),
     52           script);
     53 
     54   base::JSONReader reader;
     55   scoped_ptr<base::Value> value_result;
     56   value_result.reset(reader.Read(result));
     57   CHECK_EQ(base::Value::TYPE_DICTIONARY, value_result->GetType());
     58   base::DictionaryValue* dict_value =
     59       static_cast<base::DictionaryValue*>(value_result.get());
     60   bool test_result;
     61   std::string test_result_message;
     62   CHECK(dict_value->GetBoolean("result", &test_result));
     63   CHECK(dict_value->GetString("message", &test_result_message));
     64   if (!test_result_message.empty())
     65     ADD_FAILURE() << test_result_message;
     66   return test_result;
     67 }
     68