blob: 5fcf52b3f488cc3d2555dd35317675dbeae56df9 [file] [log] [blame]
// Copyright 2016 Google Inc. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include <algorithm>
#include <vector>
#include "base/json/json_reader.h"
#include "base/run_loop.h"
#include "cobalt/dom/document.h"
#include "cobalt/dom/testing/stub_window.h"
#include "cobalt/script/global_environment.h"
#include "cobalt/script/javascript_engine.h"
#include "cobalt/webdriver/script_executor.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
using ::testing::_;
using ::testing::DefaultValue;
using ::testing::Return;
namespace cobalt {
namespace webdriver {
namespace {
class MockElementMapping : public ElementMapping {
public:
MOCK_METHOD1(ElementToId,
protocol::ElementId(const scoped_refptr<dom::Element>&));
MOCK_METHOD1(IdToElement,
scoped_refptr<dom::Element>(const protocol::ElementId& id));
};
class MockScriptExecutorResult : public ScriptExecutorResult::ResultHandler {
public:
MOCK_METHOD1(OnResult, void(const std::string&));
MOCK_METHOD0(OnTimeout, void());
};
class JSONScriptExecutorResult : public ScriptExecutorResult::ResultHandler {
public:
void OnResult(const std::string& result) {
json_result_.reset(base::JSONReader::Read(result.c_str()));
}
void OnTimeout() { NOTREACHED(); }
base::Value* json_result() { return json_result_.get(); }
private:
scoped_ptr<base::Value> json_result_;
};
class ScriptExecutorTest : public ::testing::Test {
protected:
void SetUp() override {
stub_window_.reset(new dom::testing::StubWindow());
script_executor_ =
ScriptExecutor::Create(&element_mapping_, global_environment());
ON_CALL(element_mapping_, IdToElement(_))
.WillByDefault(Return(scoped_refptr<dom::Element>()));
ON_CALL(element_mapping_, ElementToId(_))
.WillByDefault(Return(protocol::ElementId("bad-id")));
}
scoped_refptr<dom::Window> window() { return stub_window_->window(); }
scoped_refptr<script::GlobalEnvironment> global_environment() {
return stub_window_->global_environment();
}
protected:
scoped_ptr<dom::testing::StubWindow> stub_window_;
MockElementMapping element_mapping_;
scoped_refptr<ScriptExecutor> script_executor_;
};
} // namespace
TEST_F(ScriptExecutorTest, CreateSyncScript) {
auto gc_prevented_params =
ScriptExecutorParams::Create(global_environment(), "return 5;", "[]");
ASSERT_TRUE(gc_prevented_params.params);
EXPECT_NE(
reinterpret_cast<intptr_t>(gc_prevented_params.params->function_object()),
NULL);
EXPECT_STREQ(gc_prevented_params.params->json_args().c_str(), "[]");
EXPECT_EQ(gc_prevented_params.params->async_timeout(), base::nullopt);
}
TEST_F(ScriptExecutorTest, CreateAsyncScript) {
auto gc_prevented_params =
ScriptExecutorParams::Create(global_environment(), "return 5;", "[]",
base::TimeDelta::FromMilliseconds(5));
ASSERT_TRUE(gc_prevented_params.params);
EXPECT_NE(
reinterpret_cast<intptr_t>(gc_prevented_params.params->function_object()),
NULL);
EXPECT_STREQ(gc_prevented_params.params->json_args().c_str(), "[]");
EXPECT_EQ(gc_prevented_params.params->async_timeout(), 5);
}
TEST_F(ScriptExecutorTest, CreateInvalidScript) {
auto gc_prevented_params =
ScriptExecutorParams::Create(global_environment(), "retarn 5ish;", "[]");
ASSERT_TRUE(gc_prevented_params.params);
EXPECT_EQ(
reinterpret_cast<intptr_t>(gc_prevented_params.params->function_object()),
NULL);
}
TEST_F(ScriptExecutorTest, ExecuteSync) {
auto gc_prevented_params = ScriptExecutorParams::Create(
global_environment(), "return \"retval\";", "[]");
ASSERT_TRUE(gc_prevented_params.params);
MockScriptExecutorResult result_handler;
EXPECT_CALL(result_handler, OnResult(std::string("\"retval\"")));
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
}
TEST_F(ScriptExecutorTest, ExecuteAsync) {
// Create a script that will call the async callback after 50 ms, with
// an async timeout of 100 ms.
auto gc_prevented_params = ScriptExecutorParams::Create(
global_environment(),
"var callback = arguments[0];"
"window.setTimeout(function() { callback(72); }, 50);",
"[]", base::TimeDelta::FromMilliseconds(100));
ASSERT_TRUE(gc_prevented_params.params);
MockScriptExecutorResult result_handler;
EXPECT_CALL(result_handler, OnResult(std::string("72")));
EXPECT_CALL(result_handler, OnTimeout()).Times(0);
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
// Let the message loop run for 200ms to allow enough time for the async
// script to fire the callback.
base::RunLoop run_loop;
MessageLoop::current()->PostDelayedTask(
FROM_HERE, run_loop.QuitClosure(),
base::TimeDelta::FromMilliseconds(200));
run_loop.Run();
}
TEST_F(ScriptExecutorTest, AsyncTimeout) {
// Create a script that will call the async callback after 10 seconds, with
// an async timeout of 100 ms.
auto gc_prevented_params = ScriptExecutorParams::Create(
global_environment(),
"var callback = arguments[0];"
"window.setTimeout(function() { callback(72); }, 10000);",
"[]", base::TimeDelta::FromMilliseconds(100));
ASSERT_TRUE(gc_prevented_params.params);
MockScriptExecutorResult result_handler;
EXPECT_CALL(result_handler, OnResult(_)).Times(0);
EXPECT_CALL(result_handler, OnTimeout()).Times(1);
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
// Let the message loop run for 200ms to allow enough time for the async
// timeout to fire.
base::RunLoop run_loop;
MessageLoop::current()->PostDelayedTask(
FROM_HERE, run_loop.QuitClosure(),
base::TimeDelta::FromMilliseconds(200));
run_loop.Run();
}
TEST_F(ScriptExecutorTest, ScriptThrowsException) {
auto gc_prevented_params =
ScriptExecutorParams::Create(global_environment(), "throw Error()", "[]");
ASSERT_TRUE(gc_prevented_params.params);
MockScriptExecutorResult result_handler;
EXPECT_FALSE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
}
TEST_F(ScriptExecutorTest, ConvertBoolean) {
auto gc_prevented_params = ScriptExecutorParams::Create(
global_environment(), "return arguments[0];", "[true]");
ASSERT_TRUE(gc_prevented_params.params);
MockScriptExecutorResult result_handler;
EXPECT_CALL(result_handler, OnResult(std::string("true")));
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
gc_prevented_params = ScriptExecutorParams::Create(
global_environment(), "return arguments[0];", "[false]");
ASSERT_TRUE(gc_prevented_params.params);
EXPECT_CALL(result_handler, OnResult(std::string("false")));
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
}
TEST_F(ScriptExecutorTest, ConvertNull) {
auto gc_prevented_params = ScriptExecutorParams::Create(
global_environment(), "return arguments[0];", "[null]");
ASSERT_TRUE(gc_prevented_params.params);
MockScriptExecutorResult result_handler;
EXPECT_CALL(result_handler, OnResult(std::string("null")));
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
}
TEST_F(ScriptExecutorTest, ConvertNumericType) {
auto gc_prevented_params = ScriptExecutorParams::Create(
global_environment(), "return arguments[0];", "[6]");
ASSERT_TRUE(gc_prevented_params.params);
MockScriptExecutorResult result_handler;
EXPECT_CALL(result_handler, OnResult(std::string("6")));
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
gc_prevented_params = ScriptExecutorParams::Create(
global_environment(), "return arguments[0];", "[-6.4]");
ASSERT_TRUE(gc_prevented_params.params);
EXPECT_CALL(result_handler, OnResult(std::string("-6.4")));
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
}
TEST_F(ScriptExecutorTest, ConvertString) {
auto gc_prevented_params = ScriptExecutorParams::Create(
global_environment(), "return arguments[0];", "[\"Mr. T\"]");
ASSERT_TRUE(gc_prevented_params.params);
MockScriptExecutorResult result_handler;
EXPECT_CALL(result_handler, OnResult(std::string("\"Mr. T\"")));
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
}
TEST_F(ScriptExecutorTest, ConvertWebElement) {
// Create a dom::Element for the MockElementMapping to return.
scoped_refptr<dom::Element> element =
window()->document()->CreateElement("p");
EXPECT_CALL(element_mapping_, IdToElement(protocol::ElementId("id123")))
.WillRepeatedly(Return(element));
EXPECT_CALL(element_mapping_, ElementToId(element))
.WillRepeatedly(Return(protocol::ElementId("id123")));
// Create a script that will pass a web element argument as a parameter, and
// return it back. This will invoke the lookup to and from an id.
auto gc_prevented_params =
ScriptExecutorParams::Create(global_environment(), "return arguments[0];",
"[ {\"ELEMENT\": \"id123\"} ]");
ASSERT_TRUE(gc_prevented_params.params);
// Execute the script and parse the result as JSON, ensuring we got the same
// web element.
JSONScriptExecutorResult result_handler;
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
ASSERT_TRUE(result_handler.json_result());
std::string element_id;
base::DictionaryValue* dictionary_value;
ASSERT_TRUE(result_handler.json_result()->GetAsDictionary(&dictionary_value));
EXPECT_TRUE(dictionary_value->GetString(protocol::ElementId::kElementKey,
&element_id));
EXPECT_STREQ(element_id.c_str(), "id123");
}
TEST_F(ScriptExecutorTest, ConvertArray) {
// Create a script that takes an array of numbers as input, and returns an
// array of those numbers incremented by one.
auto gc_prevented_params = ScriptExecutorParams::Create(
global_environment(),
"return [ (arguments[0][0]+1), (arguments[0][1]+1) ];", "[ [5, 6] ]");
ASSERT_TRUE(gc_prevented_params.params);
JSONScriptExecutorResult result_handler;
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
ASSERT_TRUE(result_handler.json_result());
base::ListValue* list_value;
ASSERT_TRUE(result_handler.json_result()->GetAsList(&list_value));
ASSERT_EQ(list_value->GetSize(), 2);
int value;
EXPECT_TRUE(list_value->GetInteger(0, &value));
EXPECT_EQ(value, 6);
EXPECT_TRUE(list_value->GetInteger(1, &value));
EXPECT_EQ(value, 7);
}
TEST_F(ScriptExecutorTest, ConvertObject) {
// Create a script that takes an Object with two properties as input, and
// returns an Object with one property that is the sum of the other Object's
// properties.
auto gc_prevented_params = ScriptExecutorParams::Create(
global_environment(),
"return {\"sum\": arguments[0].a + arguments[0].b};",
"[ {\"a\":5, \"b\":6} ]");
ASSERT_TRUE(gc_prevented_params.params);
JSONScriptExecutorResult result_handler;
EXPECT_TRUE(
script_executor_->Execute(gc_prevented_params.params, &result_handler));
ASSERT_TRUE(result_handler.json_result());
int value;
base::DictionaryValue* dictionary_value;
ASSERT_TRUE(result_handler.json_result()->GetAsDictionary(&dictionary_value));
EXPECT_TRUE(dictionary_value->GetInteger("sum", &value));
EXPECT_EQ(value, 11);
}
} // namespace webdriver
} // namespace cobalt