| OLD | NEW |
| 1 /* | 1 /* |
| 2 * This file is part of Adblock Plus <https://adblockplus.org/>, | 2 * This file is part of Adblock Plus <https://adblockplus.org/>, |
| 3 * Copyright (C) 2006-2017 eyeo GmbH | 3 * Copyright (C) 2006-2017 eyeo GmbH |
| 4 * | 4 * |
| 5 * Adblock Plus is free software: you can redistribute it and/or modify | 5 * Adblock Plus is free software: you can redistribute it and/or modify |
| 6 * it under the terms of the GNU General Public License version 3 as | 6 * it under the terms of the GNU General Public License version 3 as |
| 7 * published by the Free Software Foundation. | 7 * published by the Free Software Foundation. |
| 8 * | 8 * |
| 9 * Adblock Plus is distributed in the hope that it will be useful, | 9 * Adblock Plus is distributed in the hope that it will be useful, |
| 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of | 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| (...skipping 104 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 115 request.addEventListener('error', function() {result = 'error';}, false);\ | 115 request.addEventListener('error', function() {result = 'error';}, false);\ |
| 116 "); | 116 "); |
| 117 return url; | 117 return url; |
| 118 } | 118 } |
| 119 | 119 |
| 120 void WaitForVariable(const std::string& variable, const AdblockPlus::JsEngineP
tr& jsEngine) | 120 void WaitForVariable(const std::string& variable, const AdblockPlus::JsEngineP
tr& jsEngine) |
| 121 { | 121 { |
| 122 do | 122 do |
| 123 { | 123 { |
| 124 AdblockPlus::Sleep(60); | 124 AdblockPlus::Sleep(60); |
| 125 } while (jsEngine->Evaluate(variable)->IsUndefined()); | 125 } while (jsEngine->Evaluate(variable).IsUndefined()); |
| 126 } | 126 } |
| 127 | 127 |
| 128 } | 128 } |
| 129 | 129 |
| 130 TEST_F(MockWebRequestTest, BadCall) | 130 TEST_F(MockWebRequestTest, BadCall) |
| 131 { | 131 { |
| 132 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET()")); | 132 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET()")); |
| 133 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('', {}, function(){})")); | 133 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('', {}, function(){})")); |
| 134 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET({toString: false}, {}, fu
nction(){})")); | 134 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET({toString: false}, {}, fu
nction(){})")); |
| 135 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', nu
ll, function(){})")); | 135 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', nu
ll, function(){})")); |
| 136 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', {}
, null)")); | 136 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', {}
, null)")); |
| 137 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', {}
, function(){}, 0)")); | 137 ASSERT_ANY_THROW(jsEngine->Evaluate("_webRequest.GET('http://example.com/', {}
, function(){}, 0)")); |
| 138 } | 138 } |
| 139 | 139 |
| 140 TEST_F(MockWebRequestTest, SuccessfulRequest) | 140 TEST_F(MockWebRequestTest, SuccessfulRequest) |
| 141 { | 141 { |
| 142 jsEngine->Evaluate("_webRequest.GET('http://example.com/', {X: 'Y'}, function(
result) {foo = result;} )"); | 142 jsEngine->Evaluate("_webRequest.GET('http://example.com/', {X: 'Y'}, function(
result) {foo = result;} )"); |
| 143 ASSERT_TRUE(jsEngine->Evaluate("this.foo")->IsUndefined()); | 143 ASSERT_TRUE(jsEngine->Evaluate("this.foo").IsUndefined()); |
| 144 AdblockPlus::Sleep(200); | 144 AdblockPlus::Sleep(200); |
| 145 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine->Evaluate("foo.status")->As
Int()); | 145 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine->Evaluate("foo.status").AsI
nt()); |
| 146 ASSERT_EQ(123, jsEngine->Evaluate("foo.responseStatus")->AsInt()); | 146 ASSERT_EQ(123, jsEngine->Evaluate("foo.responseStatus").AsInt()); |
| 147 ASSERT_EQ("http://example.com/\nX\nY", jsEngine->Evaluate("foo.responseText")-
>AsString()); | 147 ASSERT_EQ("http://example.com/\nX\nY", jsEngine->Evaluate("foo.responseText").
AsString()); |
| 148 ASSERT_EQ("{\"Foo\":\"Bar\"}", jsEngine->Evaluate("JSON.stringify(foo.response
Headers)")->AsString()); | 148 ASSERT_EQ("{\"Foo\":\"Bar\"}", jsEngine->Evaluate("JSON.stringify(foo.response
Headers)").AsString()); |
| 149 } | 149 } |
| 150 | 150 |
| 151 #if defined(HAVE_CURL) || defined(_WIN32) | 151 #if defined(HAVE_CURL) || defined(_WIN32) |
| 152 TEST_F(DefaultWebRequestTest, RealWebRequest) | 152 TEST_F(DefaultWebRequestTest, RealWebRequest) |
| 153 { | 153 { |
| 154 // This URL should redirect to easylist-downloads.adblockplus.org and we | 154 // This URL should redirect to easylist-downloads.adblockplus.org and we |
| 155 // should get the actual filter list back. | 155 // should get the actual filter list back. |
| 156 jsEngine->Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.or
g/easylist.txt', {}, function(result) {foo = result;} )"); | 156 jsEngine->Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.or
g/easylist.txt', {}, function(result) {foo = result;} )"); |
| 157 WaitForVariable("this.foo", jsEngine); | 157 WaitForVariable("this.foo", jsEngine); |
| 158 ASSERT_EQ("text/plain", jsEngine->Evaluate("foo.responseHeaders['content-type'
].substr(0, 10)")->AsString()); | 158 ASSERT_EQ("text/plain", jsEngine->Evaluate("foo.responseHeaders['content-type'
].substr(0, 10)").AsString()); |
| 159 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine->Evaluate("foo.status")->As
Int()); | 159 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine->Evaluate("foo.status").AsI
nt()); |
| 160 ASSERT_EQ(200, jsEngine->Evaluate("foo.responseStatus")->AsInt()); | 160 ASSERT_EQ(200, jsEngine->Evaluate("foo.responseStatus").AsInt()); |
| 161 ASSERT_EQ("[Adblock Plus ", jsEngine->Evaluate("foo.responseText.substr(0, 14)
")->AsString()); | 161 ASSERT_EQ("[Adblock Plus ", jsEngine->Evaluate("foo.responseText.substr(0, 14)
").AsString()); |
| 162 ASSERT_EQ("text/plain", jsEngine->Evaluate("foo.responseHeaders['content-type'
].substr(0, 10)")->AsString()); | 162 ASSERT_EQ("text/plain", jsEngine->Evaluate("foo.responseHeaders['content-type'
].substr(0, 10)").AsString()); |
| 163 #if defined(HAVE_CURL) | 163 #if defined(HAVE_CURL) |
| 164 ASSERT_EQ("gzip", jsEngine->Evaluate("foo.responseHeaders['content-encoding'].
substr(0, 4)")->AsString()); | 164 ASSERT_EQ("gzip", jsEngine->Evaluate("foo.responseHeaders['content-encoding'].
substr(0, 4)").AsString()); |
| 165 #endif | 165 #endif |
| 166 ASSERT_TRUE(jsEngine->Evaluate("foo.responseHeaders['location']")->IsUndefined
()); | 166 ASSERT_TRUE(jsEngine->Evaluate("foo.responseHeaders['location']").IsUndefined(
)); |
| 167 } | 167 } |
| 168 | 168 |
| 169 TEST_F(DefaultWebRequestTest, XMLHttpRequest) | 169 TEST_F(DefaultWebRequestTest, XMLHttpRequest) |
| 170 { | 170 { |
| 171 auto filterEngine = AdblockPlus::FilterEngine::Create(jsEngine); | 171 auto filterEngine = AdblockPlus::FilterEngine::Create(jsEngine); |
| 172 | 172 |
| 173 ResetTestXHR(jsEngine, "https://easylist-downloads.adblockplus.org/easylist.tx
t"); | 173 ResetTestXHR(jsEngine, "https://easylist-downloads.adblockplus.org/easylist.tx
t"); |
| 174 jsEngine->Evaluate("\ | 174 jsEngine->Evaluate("\ |
| 175 request.setRequestHeader('X', 'Y');\ | 175 request.setRequestHeader('X', 'Y');\ |
| 176 request.setRequestHeader('X2', 'Y2');\ | 176 request.setRequestHeader('X2', 'Y2');\ |
| 177 request.send(null);"); | 177 request.send(null);"); |
| 178 WaitForVariable("result", jsEngine); | 178 WaitForVariable("result", jsEngine); |
| 179 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine->Evaluate("request.channel.
status")->AsInt()); | 179 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine->Evaluate("request.channel.
status").AsInt()); |
| 180 ASSERT_EQ(200, jsEngine->Evaluate("request.status")->AsInt()); | 180 ASSERT_EQ(200, jsEngine->Evaluate("request.status").AsInt()); |
| 181 ASSERT_EQ("[Adblock Plus ", jsEngine->Evaluate("result.substr(0, 14)")->AsStri
ng()); | 181 ASSERT_EQ("[Adblock Plus ", jsEngine->Evaluate("result.substr(0, 14)").AsStrin
g()); |
| 182 ASSERT_EQ("text/plain", jsEngine->Evaluate("request.getResponseHeader('Content
-Type').substr(0, 10)")->AsString()); | 182 ASSERT_EQ("text/plain", jsEngine->Evaluate("request.getResponseHeader('Content
-Type').substr(0, 10)").AsString()); |
| 183 #if defined(HAVE_CURL) | 183 #if defined(HAVE_CURL) |
| 184 ASSERT_EQ("gzip", jsEngine->Evaluate("request.getResponseHeader('Content-Encod
ing').substr(0, 4)")->AsString()); | 184 ASSERT_EQ("gzip", jsEngine->Evaluate("request.getResponseHeader('Content-Encod
ing').substr(0, 4)").AsString()); |
| 185 #endif | 185 #endif |
| 186 ASSERT_TRUE(jsEngine->Evaluate("request.getResponseHeader('Location')")->IsNul
l()); | 186 ASSERT_TRUE(jsEngine->Evaluate("request.getResponseHeader('Location')").IsNull
()); |
| 187 } | 187 } |
| 188 #else | 188 #else |
| 189 TEST_F(DefaultWebRequestTest, DummyWebRequest) | 189 TEST_F(DefaultWebRequestTest, DummyWebRequest) |
| 190 { | 190 { |
| 191 jsEngine->Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.or
g/easylist.txt', {}, function(result) {foo = result;} )"); | 191 jsEngine->Evaluate("_webRequest.GET('https://easylist-downloads.adblockplus.or
g/easylist.txt', {}, function(result) {foo = result;} )"); |
| 192 WaitForVariable("this.foo", jsEngine); | 192 WaitForVariable("this.foo", jsEngine); |
| 193 ASSERT_EQ(AdblockPlus::WebRequest::NS_ERROR_FAILURE, jsEngine->Evaluate("foo.s
tatus")->AsInt()); | 193 ASSERT_EQ(AdblockPlus::WebRequest::NS_ERROR_FAILURE, jsEngine->Evaluate("foo.s
tatus").AsInt()); |
| 194 ASSERT_EQ(0, jsEngine->Evaluate("foo.responseStatus")->AsInt()); | 194 ASSERT_EQ(0, jsEngine->Evaluate("foo.responseStatus").AsInt()); |
| 195 ASSERT_EQ("", jsEngine->Evaluate("foo.responseText")->AsString()); | 195 ASSERT_EQ("", jsEngine->Evaluate("foo.responseText").AsString()); |
| 196 ASSERT_EQ("{}", jsEngine->Evaluate("JSON.stringify(foo.responseHeaders)")->AsS
tring()); | 196 ASSERT_EQ("{}", jsEngine->Evaluate("JSON.stringify(foo.responseHeaders)").AsSt
ring()); |
| 197 } | 197 } |
| 198 | 198 |
| 199 TEST_F(DefaultWebRequestTest, XMLHttpRequest) | 199 TEST_F(DefaultWebRequestTest, XMLHttpRequest) |
| 200 { | 200 { |
| 201 auto filterEngine = AdblockPlus::FilterEngine::Create(jsEngine); | 201 auto filterEngine = AdblockPlus::FilterEngine::Create(jsEngine); |
| 202 | 202 |
| 203 ResetTestXHR(jsEngine); | 203 ResetTestXHR(jsEngine); |
| 204 jsEngine->Evaluate("\ | 204 jsEngine->Evaluate("\ |
| 205 request.setRequestHeader('X', 'Y');\ | 205 request.setRequestHeader('X', 'Y');\ |
| 206 request.send(null);"); | 206 request.send(null);"); |
| 207 WaitForVariable("result", jsEngine); | 207 WaitForVariable("result", jsEngine); |
| 208 ASSERT_EQ(AdblockPlus::WebRequest::NS_ERROR_FAILURE, jsEngine->Evaluate("reque
st.channel.status")->AsInt()); | 208 ASSERT_EQ(AdblockPlus::WebRequest::NS_ERROR_FAILURE, jsEngine->Evaluate("reque
st.channel.status").AsInt()); |
| 209 ASSERT_EQ(0, jsEngine->Evaluate("request.status")->AsInt()); | 209 ASSERT_EQ(0, jsEngine->Evaluate("request.status").AsInt()); |
| 210 ASSERT_EQ("error", jsEngine->Evaluate("result")->AsString()); | 210 ASSERT_EQ("error", jsEngine->Evaluate("result").AsString()); |
| 211 ASSERT_TRUE(jsEngine->Evaluate("request.getResponseHeader('Content-Type')")->I
sNull()); | 211 ASSERT_TRUE(jsEngine->Evaluate("request.getResponseHeader('Content-Type')").Is
Null()); |
| 212 } | 212 } |
| 213 | 213 |
| 214 #endif | 214 #endif |
| 215 | 215 |
| 216 namespace | 216 namespace |
| 217 { | 217 { |
| 218 class CatchLogSystem : public AdblockPlus::LogSystem | 218 class CatchLogSystem : public AdblockPlus::LogSystem |
| 219 { | 219 { |
| 220 public: | 220 public: |
| 221 AdblockPlus::LogSystem::LogLevel lastLogLevel; | 221 AdblockPlus::LogSystem::LogLevel lastLogLevel; |
| (...skipping 118 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 340 EXPECT_EQ(AdblockPlus::LogSystem::LOG_LEVEL_TRACE, catchLogSystem->lastLogLeve
l); | 340 EXPECT_EQ(AdblockPlus::LogSystem::LOG_LEVEL_TRACE, catchLogSystem->lastLogLeve
l); |
| 341 EXPECT_EQ("", catchLogSystem->lastMessage); | 341 EXPECT_EQ("", catchLogSystem->lastMessage); |
| 342 WaitForVariable("result", jsEngine); | 342 WaitForVariable("result", jsEngine); |
| 343 { | 343 { |
| 344 auto headersRequest = webRequest->headersForRequest(url); | 344 auto headersRequest = webRequest->headersForRequest(url); |
| 345 EXPECT_TRUE(headersRequest.first); | 345 EXPECT_TRUE(headersRequest.first); |
| 346 const auto& headers = headersRequest.second; | 346 const auto& headers = headersRequest.second; |
| 347 EXPECT_FALSE(headers.cend() == headers.find("Security")); | 347 EXPECT_FALSE(headers.cend() == headers.find("Security")); |
| 348 } | 348 } |
| 349 } | 349 } |
| OLD | NEW |