| Index: test/WebRequest.cpp |
| diff --git a/test/WebRequest.cpp b/test/WebRequest.cpp |
| index cd14e207ff1a5b1d91969a378e8979d56653afe7..fbad8f9ae59b7b082ef19c1b2fa0ef798a34597b 100644 |
| --- a/test/WebRequest.cpp |
| +++ b/test/WebRequest.cpp |
| @@ -25,7 +25,7 @@ using namespace AdblockPlus; |
| namespace |
| { |
| - class BaseWebRequestTest : public ::testing::Test |
| + class BaseWebRequestTest : public BaseJsTest |
| { |
| protected: |
| void SetUp() |
| @@ -36,7 +36,6 @@ namespace |
| platformParams.fileSystem.reset(fileSystem = new LazyFileSystem()); |
| platformParams.webRequest = CreateWebRequest(); |
| platform.reset(new Platform(std::move(platformParams))); |
| - jsEngine = platform->GetJsEngine(); |
| } |
| virtual WebRequestPtr CreateWebRequest() = 0; |
| @@ -46,8 +45,6 @@ namespace |
| return LogSystemPtr(new ThrowingLogSystem()); |
| } |
| - std::unique_ptr<Platform> platform; |
| - JsEnginePtr jsEngine; |
| LazyFileSystem* fileSystem; |
| }; |
| @@ -62,9 +59,9 @@ namespace |
| } |
| std::list<SchedulerTask> webRequestTasks; |
| protected: |
| - void WaitForVariable(const std::string& variable, const AdblockPlus::JsEnginePtr& jsEngine) |
| + void WaitForVariable(const std::string& variable, AdblockPlus::JsEngine& jsEngine) |
| { |
| - while (jsEngine->Evaluate(variable).IsUndefined() && !webRequestTasks.empty()) |
| + while (jsEngine.Evaluate(variable).IsUndefined() && !webRequestTasks.empty()) |
| { |
| (*webRequestTasks.begin())(); |
| webRequestTasks.pop_front(); |
| @@ -127,7 +124,7 @@ namespace |
| }; |
| // we return the url of the XHR. |
| - std::string ResetTestXHR(const AdblockPlus::JsEnginePtr& jsEngine, const std::string& defaultUrl = "") |
| + std::string ResetTestXHR(AdblockPlus::JsEngine& jsEngine, const std::string& defaultUrl = "") |
| { |
| std::string url = defaultUrl; |
| // make up a unique URL if we don't have one. |
| @@ -137,7 +134,7 @@ namespace |
| url += std::to_string(std::chrono::system_clock::to_time_t(std::chrono::system_clock::now())); |
| } |
| - jsEngine->Evaluate(std::string("\ |
| + jsEngine.Evaluate(std::string("\ |
| var result;\ |
| var request = new XMLHttpRequest();\ |
| request.open('GET', '") + url + "'); \ |
| @@ -151,86 +148,92 @@ namespace |
| TEST_F(MockWebRequestTest, BadCall) |
| { |
| - ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET()")); |
| - ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('', {}, function(){})")); |
| - ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET({toString: false}, {}, function(){})")); |
| - ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', null, function(){})")); |
| - ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', {}, null)")); |
| - ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', {}, function(){}, 0)")); |
| + auto& jsEngine = GetJsEngine(); |
| + ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET()")); |
| + ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('', {}, function(){})")); |
| + ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET({toString: false}, {}, function(){})")); |
| + ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('http://example.com/', null, function(){})")); |
| + ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('http://example.com/', {}, null)")); |
| + ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('http://example.com/', {}, function(){}, 0)")); |
| } |
| TEST_F(MockWebRequestTest, SuccessfulRequest) |
| { |
| - jsEngine->Evaluate("_webRequest.GET('http://example.com/', {X: 'Y'}, function(result) {foo = result;} )"); |
| - ASSERT_TRUE(jsEngine->Evaluate("this.foo").IsUndefined()); |
| + auto& jsEngine = GetJsEngine(); |
| + jsEngine.Evaluate("_webRequest.GET('http://example.com/', {X: 'Y'}, function(result) {foo = result;} )"); |
| + ASSERT_TRUE(jsEngine.Evaluate("this.foo").IsUndefined()); |
| ProcessPendingWebRequests(); |
| - ASSERT_EQ(IWebRequest::NS_OK, jsEngine->Evaluate("foo.status").AsInt()); |
| - ASSERT_EQ(123, jsEngine->Evaluate("foo.responseStatus").AsInt()); |
| - ASSERT_EQ("http://example.com/\nX\nY", jsEngine->Evaluate("foo.responseText").AsString()); |
| - ASSERT_EQ("{\"Foo\":\"Bar\"}", jsEngine->Evaluate("JSON.stringify(foo.responseHeaders)").AsString()); |
| + ASSERT_EQ(IWebRequest::NS_OK, jsEngine.Evaluate("foo.status").AsInt()); |
| + ASSERT_EQ(123, jsEngine.Evaluate("foo.responseStatus").AsInt()); |
| + ASSERT_EQ("http://example.com/\nX\nY", jsEngine.Evaluate("foo.responseText").AsString()); |
| + ASSERT_EQ("{\"Foo\":\"Bar\"}", jsEngine.Evaluate("JSON.stringify(foo.responseHeaders)").AsString()); |
| } |
| #if defined(HAVE_CURL) || defined(_WIN32) |
| TEST_F(DefaultWebRequestTest, RealWebRequest) |
| { |
| + auto& jsEngine = GetJsEngine(); |
| // This URL should redirect to easylist-downloads.adblockplus.org and we |
| // should get the actual filter list back. |
| - jsEngine->Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.org/easylist.txt', {}, function(result) {foo = result;} )"); |
| + jsEngine.Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.org/easylist.txt', {}, function(result) {foo = result;} )"); |
| WaitForVariable("this.foo", jsEngine); |
| - ASSERT_EQ("text/plain", jsEngine->Evaluate("foo.responseHeaders['content-type'].substr(0, 10)").AsString()); |
| - ASSERT_EQ(IWebRequest::NS_OK, jsEngine->Evaluate("foo.status").AsInt()); |
| - ASSERT_EQ(200, jsEngine->Evaluate("foo.responseStatus").AsInt()); |
| - ASSERT_EQ("[Adblock Plus ", jsEngine->Evaluate("foo.responseText.substr(0, 14)").AsString()); |
| - ASSERT_EQ("text/plain", jsEngine->Evaluate("foo.responseHeaders['content-type'].substr(0, 10)").AsString()); |
| + ASSERT_EQ("text/plain", jsEngine.Evaluate("foo.responseHeaders['content-type'].substr(0, 10)").AsString()); |
| + ASSERT_EQ(IWebRequest::NS_OK, jsEngine.Evaluate("foo.status").AsInt()); |
| + ASSERT_EQ(200, jsEngine.Evaluate("foo.responseStatus").AsInt()); |
| + ASSERT_EQ("[Adblock Plus ", jsEngine.Evaluate("foo.responseText.substr(0, 14)").AsString()); |
| + ASSERT_EQ("text/plain", jsEngine.Evaluate("foo.responseHeaders['content-type'].substr(0, 10)").AsString()); |
| #if defined(HAVE_CURL) |
| - ASSERT_EQ("gzip", jsEngine->Evaluate("foo.responseHeaders['content-encoding'].substr(0, 4)").AsString()); |
| + ASSERT_EQ("gzip", jsEngine.Evaluate("foo.responseHeaders['content-encoding'].substr(0, 4)").AsString()); |
| #endif |
| - ASSERT_TRUE(jsEngine->Evaluate("foo.responseHeaders['location']").IsUndefined()); |
| + ASSERT_TRUE(jsEngine.Evaluate("foo.responseHeaders['location']").IsUndefined()); |
| } |
| TEST_F(DefaultWebRequestTest, XMLHttpRequest) |
| { |
| + auto& jsEngine = GetJsEngine(); |
| auto filterEngine = CreateFilterEngine(*fileSystem, *platform); |
| ResetTestXHR(jsEngine, "https://easylist-downloads.adblockplus.org/easylist.txt"); |
| - jsEngine->Evaluate("\ |
| + jsEngine.Evaluate("\ |
| request.setRequestHeader('X', 'Y');\ |
| request.setRequestHeader('X2', 'Y2');\ |
| request.send(null);"); |
| WaitForVariable("result", jsEngine); |
| - ASSERT_EQ(IWebRequest::NS_OK, jsEngine->Evaluate("request.channel.status").AsInt()); |
| - ASSERT_EQ(200, jsEngine->Evaluate("request.status").AsInt()); |
| - ASSERT_EQ("[Adblock Plus ", jsEngine->Evaluate("result.substr(0, 14)").AsString()); |
| - ASSERT_EQ("text/plain", jsEngine->Evaluate("request.getResponseHeader('Content-Type').substr(0, 10)").AsString()); |
| + ASSERT_EQ(IWebRequest::NS_OK, jsEngine.Evaluate("request.channel.status").AsInt()); |
| + ASSERT_EQ(200, jsEngine.Evaluate("request.status").AsInt()); |
| + ASSERT_EQ("[Adblock Plus ", jsEngine.Evaluate("result.substr(0, 14)").AsString()); |
| + ASSERT_EQ("text/plain", jsEngine.Evaluate("request.getResponseHeader('Content-Type').substr(0, 10)").AsString()); |
| #if defined(HAVE_CURL) |
| - ASSERT_EQ("gzip", jsEngine->Evaluate("request.getResponseHeader('Content-Encoding').substr(0, 4)").AsString()); |
| + ASSERT_EQ("gzip", jsEngine.Evaluate("request.getResponseHeader('Content-Encoding').substr(0, 4)").AsString()); |
| #endif |
| - ASSERT_TRUE(jsEngine->Evaluate("request.getResponseHeader('Location')").IsNull()); |
| + ASSERT_TRUE(jsEngine.Evaluate("request.getResponseHeader('Location')").IsNull()); |
| } |
| #else |
| TEST_F(DefaultWebRequestTest, DummyWebRequest) |
| { |
| - jsEngine->Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.org/easylist.txt', {}, function(result) {foo = result;} )"); |
| + auto& jsEngine = GetJsEngine(); |
| + jsEngine.Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.org/easylist.txt', {}, function(result) {foo = result;} )"); |
| WaitForVariable("this.foo", jsEngine); |
| - ASSERT_EQ(IWebRequest::NS_ERROR_FAILURE, jsEngine->Evaluate("foo.status").AsInt()); |
| - ASSERT_EQ(0, jsEngine->Evaluate("foo.responseStatus").AsInt()); |
| - ASSERT_EQ("", jsEngine->Evaluate("foo.responseText").AsString()); |
| - ASSERT_EQ("{}", jsEngine->Evaluate("JSON.stringify(foo.responseHeaders)").AsString()); |
| + ASSERT_EQ(IWebRequest::NS_ERROR_FAILURE, jsEngine.Evaluate("foo.status").AsInt()); |
| + ASSERT_EQ(0, jsEngine.Evaluate("foo.responseStatus").AsInt()); |
| + ASSERT_EQ("", jsEngine.Evaluate("foo.responseText").AsString()); |
| + ASSERT_EQ("{}", jsEngine.Evaluate("JSON.stringify(foo.responseHeaders)").AsString()); |
| } |
| TEST_F(DefaultWebRequestTest, XMLHttpRequest) |
| { |
| + auto& jsEngine = GetJsEngine(); |
| auto filterEngine = CreateFilterEngine(*fileSystem, *platform); |
| ResetTestXHR(jsEngine); |
| - jsEngine->Evaluate("\ |
| + jsEngine.Evaluate("\ |
| request.setRequestHeader('X', 'Y');\ |
| request.send(null);"); |
| WaitForVariable("result", jsEngine); |
| - ASSERT_EQ(IWebRequest::NS_ERROR_FAILURE, jsEngine->Evaluate("request.channel.status").AsInt()); |
| - ASSERT_EQ(0, jsEngine->Evaluate("request.status").AsInt()); |
| - ASSERT_EQ("error", jsEngine->Evaluate("result").AsString()); |
| - ASSERT_TRUE(jsEngine->Evaluate("request.getResponseHeader('Content-Type')").IsNull()); |
| + ASSERT_EQ(IWebRequest::NS_ERROR_FAILURE, jsEngine.Evaluate("request.channel.status").AsInt()); |
| + ASSERT_EQ(0, jsEngine.Evaluate("request.status").AsInt()); |
| + ASSERT_EQ("error", jsEngine.Evaluate("result").AsString()); |
| + ASSERT_TRUE(jsEngine.Evaluate("request.getResponseHeader('Content-Type')").IsNull()); |
| } |
| #endif |
| @@ -276,6 +279,7 @@ namespace |
| TEST_F(MockWebRequestAndLogSystemTest, RequestHeaderValidation) |
| { |
| + auto& jsEngine = GetJsEngine(); |
| auto filterEngine = CreateFilterEngine(*fileSystem, *platform); |
| const std::string msg = "Attempt to set a forbidden header was denied: "; |
| @@ -287,7 +291,7 @@ TEST_F(MockWebRequestAndLogSystemTest, RequestHeaderValidation) |
| // test 'Accept-Encoding' is rejected |
| catchLogSystem->clear(); |
| std::string url = ResetTestXHR(jsEngine); |
| - jsEngine->Evaluate("\ |
| + jsEngine.Evaluate("\ |
| request.setRequestHeader('Accept-Encoding', 'gzip');\nrequest.send();"); |
| EXPECT_EQ(AdblockPlus::LogSystem::LOG_LEVEL_WARN, catchLogSystem->lastLogLevel); |
| EXPECT_EQ(msg + "Accept-Encoding", catchLogSystem->lastMessage); |
| @@ -302,7 +306,7 @@ TEST_F(MockWebRequestAndLogSystemTest, RequestHeaderValidation) |
| // test 'DNT' is rejected |
| catchLogSystem->clear(); |
| url = ResetTestXHR(jsEngine); |
| - jsEngine->Evaluate("\ |
| + jsEngine.Evaluate("\ |
| request.setRequestHeader('DNT', '1');\nrequest.send();"); |
| EXPECT_EQ(AdblockPlus::LogSystem::LOG_LEVEL_WARN, catchLogSystem->lastLogLevel); |
| EXPECT_EQ(msg + "DNT", catchLogSystem->lastMessage); |
| @@ -317,7 +321,7 @@ TEST_F(MockWebRequestAndLogSystemTest, RequestHeaderValidation) |
| // test random 'X' header is accepted |
| catchLogSystem->clear(); |
| url = ResetTestXHR(jsEngine); |
| - jsEngine->Evaluate("\ |
| + jsEngine.Evaluate("\ |
| request.setRequestHeader('X', 'y');\nrequest.send();"); |
| EXPECT_EQ(AdblockPlus::LogSystem::LOG_LEVEL_TRACE, catchLogSystem->lastLogLevel); |
| EXPECT_EQ("", catchLogSystem->lastMessage); |
| @@ -332,7 +336,7 @@ TEST_F(MockWebRequestAndLogSystemTest, RequestHeaderValidation) |
| // test /^Proxy-/ is rejected. |
| catchLogSystem->clear(); |
| url = ResetTestXHR(jsEngine); |
| - jsEngine->Evaluate("\ |
| + jsEngine.Evaluate("\ |
| request.setRequestHeader('Proxy-foo', 'bar');\nrequest.send();"); |
| EXPECT_EQ(AdblockPlus::LogSystem::LOG_LEVEL_WARN, catchLogSystem->lastLogLevel); |
| EXPECT_EQ(msg + "Proxy-foo", catchLogSystem->lastMessage); |
| @@ -347,7 +351,7 @@ TEST_F(MockWebRequestAndLogSystemTest, RequestHeaderValidation) |
| // test /^Sec-/ is rejected. |
| catchLogSystem->clear(); |
| url = ResetTestXHR(jsEngine); |
| - jsEngine->Evaluate("\ |
| + jsEngine.Evaluate("\ |
| request.setRequestHeader('Sec-foo', 'bar');\nrequest.send();"); |
| EXPECT_EQ(AdblockPlus::LogSystem::LOG_LEVEL_WARN, catchLogSystem->lastLogLevel); |
| EXPECT_EQ(msg + "Sec-foo", catchLogSystem->lastMessage); |
| @@ -362,7 +366,7 @@ TEST_F(MockWebRequestAndLogSystemTest, RequestHeaderValidation) |
| // test 'Security' is accepted. |
| catchLogSystem->clear(); |
| url = ResetTestXHR(jsEngine); |
| - jsEngine->Evaluate("\ |
| + jsEngine.Evaluate("\ |
| request.setRequestHeader('Security', 'theater');\nrequest.send();"); |
| EXPECT_EQ(AdblockPlus::LogSystem::LOG_LEVEL_TRACE, catchLogSystem->lastLogLevel); |
| EXPECT_EQ("", catchLogSystem->lastMessage); |