OLD | NEW |
1 #include <sstream> | 1 #include <sstream> |
2 #include <AdblockPlus.h> | 2 #include <AdblockPlus.h> |
3 #include <gtest/gtest.h> | 3 #include <gtest/gtest.h> |
4 | 4 |
5 #include "../src/Thread.h" | 5 #include "../src/Thread.h" |
6 | 6 |
7 #ifdef HAVE_WININET | |
8 #include "../src/WebRequestWinInet.h" | |
9 #endif | |
10 | |
11 class TestWebRequest : public AdblockPlus::WebRequest | 7 class TestWebRequest : public AdblockPlus::WebRequest |
12 { | 8 { |
13 public: | 9 public: |
14 AdblockPlus::ServerResponse GET(const std::string& url, const AdblockPlus::Hea
derList& requestHeaders) const | 10 AdblockPlus::ServerResponse GET(const std::string& url, const AdblockPlus::Hea
derList& requestHeaders) const |
15 { | 11 { |
16 AdblockPlus::Sleep(50); | 12 AdblockPlus::Sleep(50); |
17 | 13 |
18 AdblockPlus::ServerResponse result; | 14 AdblockPlus::ServerResponse result; |
19 result.status = NS_OK; | 15 result.status = NS_OK; |
20 result.responseStatus = 123; | 16 result.responseStatus = 123; |
21 result.responseHeaders.push_back(std::pair<std::string, std::string>("Foo",
"Bar")); | 17 result.responseHeaders.push_back(std::pair<std::string, std::string>("Foo",
"Bar")); |
22 result.responseText = url + "\n" + requestHeaders[0].first + "\n" + requestH
eaders[0].second; | 18 result.responseText = url + "\n" + requestHeaders[0].first + "\n" + requestH
eaders[0].second; |
23 return result; | 19 return result; |
24 } | 20 } |
25 }; | 21 }; |
26 | 22 |
27 std::string ToString(unsigned int i) | 23 std::string ToString(unsigned int i) |
28 { | 24 { |
29 std::stringstream stream; | 25 std::stringstream stream; |
30 stream << i; | 26 stream << i; |
31 return stream.str(); | 27 return stream.str(); |
32 } | 28 } |
33 | 29 |
34 TEST(WebRequestTest, BadCall) | 30 TEST(WebRequestTest, BadCall) |
35 { | 31 { |
36 AdblockPlus::JsEngine jsEngine(AdblockPlus::AppInfo(), 0, 0, 0); | 32 AdblockPlus::JsEngine jsEngine; |
37 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET()")); | 33 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET()")); |
38 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('', {}, function(){})")); | 34 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('', {}, function(){})")); |
39 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET({toString: false}, {}, fun
ction(){})")); | 35 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET({toString: false}, {}, fun
ction(){})")); |
40 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('http://example.com/', nul
l, function(){})")); | 36 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('http://example.com/', nul
l, function(){})")); |
41 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('http://example.com/', {},
null)")); | 37 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('http://example.com/', {},
null)")); |
42 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('http://example.com/', {},
function(){}, 0)")); | 38 ASSERT_ANY_THROW(jsEngine.Evaluate("_webRequest.GET('http://example.com/', {},
function(){}, 0)")); |
43 } | 39 } |
44 | 40 |
45 TEST(WebRequestTest, TestWebRequest) | 41 TEST(WebRequestTest, TestWebRequest) |
46 { | 42 { |
47 TestWebRequest webRequest; | 43 AdblockPlus::JsEngine jsEngine; |
48 AdblockPlus::JsEngine jsEngine(AdblockPlus::AppInfo(), 0, &webRequest, 0); | 44 jsEngine.SetWebRequest(AdblockPlus::WebRequestPtr(new TestWebRequest())); |
49 jsEngine.Evaluate("_webRequest.GET('http://example.com/', {X: 'Y'}, function(r
esult) {foo = result;} )"); | 45 jsEngine.Evaluate("_webRequest.GET('http://example.com/', {X: 'Y'}, function(r
esult) {foo = result;} )"); |
50 ASSERT_TRUE(jsEngine.Evaluate("this.foo")->IsUndefined()); | 46 ASSERT_TRUE(jsEngine.Evaluate("this.foo")->IsUndefined()); |
51 AdblockPlus::Sleep(200); | 47 AdblockPlus::Sleep(200); |
52 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine.Evaluate("foo.status")->AsI
nt()); | 48 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine.Evaluate("foo.status")->AsI
nt()); |
53 ASSERT_EQ(123, jsEngine.Evaluate("foo.responseStatus")->AsInt()); | 49 ASSERT_EQ(123, jsEngine.Evaluate("foo.responseStatus")->AsInt()); |
54 ASSERT_EQ("http://example.com/\nX\nY", jsEngine.Evaluate("foo.responseText")->
AsString()); | 50 ASSERT_EQ("http://example.com/\nX\nY", jsEngine.Evaluate("foo.responseText")->
AsString()); |
55 ASSERT_EQ("{\"Foo\":\"Bar\"}", jsEngine.Evaluate("JSON.stringify(foo.responseH
eaders)")->AsString()); | 51 ASSERT_EQ("{\"Foo\":\"Bar\"}", jsEngine.Evaluate("JSON.stringify(foo.responseH
eaders)")->AsString()); |
56 } | 52 } |
57 | 53 |
58 #if defined(HAVE_CURL) | 54 #if defined(HAVE_CURL) or defined(HAVE_WININET) |
59 TEST(WebRequestTest, RealWebRequest) | 55 TEST(WebRequestTest, RealWebRequest) |
60 { | 56 { |
61 AdblockPlus::DefaultWebRequest webRequest; | 57 AdblockPlus::JsEngine jsEngine; |
62 AdblockPlus::JsEngine jsEngine(AdblockPlus::AppInfo(), 0, &webRequest, 0); | |
63 | 58 |
64 // This URL should redirect to easylist-downloads.adblockplus.org and we | 59 // This URL should redirect to easylist-downloads.adblockplus.org and we |
65 // should get the actual filter list back. | 60 // should get the actual filter list back. |
66 jsEngine.Evaluate("_webRequest.GET('https://easylist.adblockplus.org/easylist.
txt', {}, function(result) {foo = result;} )"); | 61 jsEngine.Evaluate("_webRequest.GET('https://easylist.adblockplus.org/easylist.
txt', {}, function(result) {foo = result;} )"); |
67 do | 62 do |
68 { | 63 { |
69 AdblockPlus::Sleep(200); | 64 AdblockPlus::Sleep(200); |
70 } while (jsEngine.Evaluate("this.foo")->IsUndefined()); | 65 } while (jsEngine.Evaluate("this.foo")->IsUndefined()); |
71 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine.Evaluate("foo.status")->AsI
nt()); | 66 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine.Evaluate("foo.status")->AsI
nt()); |
72 ASSERT_EQ(200, jsEngine.Evaluate("foo.responseStatus")->AsInt()); | 67 ASSERT_EQ(200, jsEngine.Evaluate("foo.responseStatus")->AsInt()); |
73 ASSERT_EQ("[Adblock Plus ", jsEngine.Evaluate("foo.responseText.substr(0, 14)"
)->AsString()); | 68 ASSERT_EQ("[Adblock Plus ", jsEngine.Evaluate("foo.responseText.substr(0, 14)"
)->AsString()); |
74 ASSERT_EQ("text/plain", jsEngine.Evaluate("foo.responseHeaders['content-type']
.substr(0, 10)")->AsString()); | 69 ASSERT_EQ("text/plain", jsEngine.Evaluate("foo.responseHeaders['content-type']
.substr(0, 10)")->AsString()); |
75 ASSERT_TRUE(jsEngine.Evaluate("foo.responseHeaders['location']")->IsUndefined(
)); | 70 ASSERT_TRUE(jsEngine.Evaluate("foo.responseHeaders['location']")->IsUndefined(
)); |
76 } | 71 } |
77 #elif defined(HAVE_WININET) | |
78 TEST(WebRequestTest, RealWebRequest) | |
79 { | |
80 AdblockPlus::WebRequestWinInet webRequest; | |
81 AdblockPlus::JsEngine jsEngine(AdblockPlus::AppInfo(), 0, &webRequest, 0); | |
82 jsEngine.Evaluate("_webRequest.GET('https://easylist.adblockplus.org/easylist.
txt', {}, function(result) {foo = result;} )"); | |
83 do | |
84 { | |
85 AdblockPlus::Sleep(200); | |
86 } while (jsEngine.Evaluate("typeof foo") == "undefined"); | |
87 ASSERT_EQ(ToString(AdblockPlus::WebRequest::NS_OK), jsEngine.Evaluate("foo.sta
tus")); | |
88 ASSERT_EQ("200", jsEngine.Evaluate("foo.responseStatus")); | |
89 ASSERT_EQ("[Adblock Plus ", jsEngine.Evaluate("foo.responseText.substr(0, 14)"
)); | |
90 //TODO: Shall we be strict with letter casing here? Content-Type vs content-ty
pe | |
91 ASSERT_EQ("text/plain", jsEngine.Evaluate("foo.responseHeaders['Content-Type']
.substr(0, 10)")); | |
92 ASSERT_EQ("undefined", jsEngine.Evaluate("typeof foo.responseHeaders['location
']")); | |
93 } | |
94 | 72 |
| 73 #if !FILTER_ENGINE_STUBS |
95 TEST(WebRequestTest, XMLHttpRequest) | 74 TEST(WebRequestTest, XMLHttpRequest) |
96 { | 75 { |
97 AdblockPlus::DefaultWebRequest webRequest; | 76 AdblockPlus::JsEngine jsEngine; |
98 AdblockPlus::JsEngine jsEngine(AdblockPlus::AppInfo(), 0, &webRequest, 0); | |
99 AdblockPlus::FilterEngine filterEngine(jsEngine); | 77 AdblockPlus::FilterEngine filterEngine(jsEngine); |
100 | 78 |
101 jsEngine.Evaluate("\ | 79 jsEngine.Evaluate("\ |
102 var result;\ | 80 var result;\ |
103 var request = new XMLHttpRequest();\ | 81 var request = new XMLHttpRequest();\ |
104 request.open('GET', 'https://easylist.adblockplus.org/easylist.txt');\ | 82 request.open('GET', 'https://easylist.adblockplus.org/easylist.txt');\ |
105 request.setRequestHeader('X', 'Y');\ | 83 request.setRequestHeader('X', 'Y');\ |
106 request.overrideMimeType('text/plain');\ | 84 request.overrideMimeType('text/plain');\ |
107 request.addEventListener('load', function() {result = request.responseText;}
, false);\ | 85 request.addEventListener('load', function() {result = request.responseText;}
, false);\ |
108 request.addEventListener('error', function() {result = 'error';}, false);\ | 86 request.addEventListener('error', function() {result = 'error';}, false);\ |
109 request.send(null);"); | 87 request.send(null);"); |
110 do | 88 do |
111 { | 89 { |
112 AdblockPlus::Sleep(200); | 90 AdblockPlus::Sleep(200); |
113 } while (jsEngine.Evaluate("result")->IsUndefined()); | 91 } while (jsEngine.Evaluate("result")->IsUndefined()); |
114 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine.Evaluate("request.channel.s
tatus")->AsInt()); | 92 ASSERT_EQ(AdblockPlus::WebRequest::NS_OK, jsEngine.Evaluate("request.channel.s
tatus")->AsInt()); |
115 ASSERT_EQ(200, jsEngine.Evaluate("request.status")->AsInt()); | 93 ASSERT_EQ(200, jsEngine.Evaluate("request.status")->AsInt()); |
116 ASSERT_EQ("[Adblock Plus ", jsEngine.Evaluate("result.substr(0, 14)")->AsStrin
g()); | 94 ASSERT_EQ("[Adblock Plus ", jsEngine.Evaluate("result.substr(0, 14)")->AsStrin
g()); |
117 ASSERT_EQ("text/plain", jsEngine.Evaluate("request.getResponseHeader('Content-
Type').substr(0, 10)")->AsString()); | 95 ASSERT_EQ("text/plain", jsEngine.Evaluate("request.getResponseHeader('Content-
Type').substr(0, 10)")->AsString()); |
118 ASSERT_TRUE(jsEngine.Evaluate("request.getResponseHeader('Location')")->IsNull
()); | 96 ASSERT_TRUE(jsEngine.Evaluate("request.getResponseHeader('Location')")->IsNull
()); |
119 } | 97 } |
| 98 #endif |
| 99 |
120 #else | 100 #else |
121 TEST(WebRequestTest, DummyWebRequest) | 101 TEST(WebRequestTest, DummyWebRequest) |
122 { | 102 { |
123 AdblockPlus::DefaultWebRequest webRequest; | 103 AdblockPlus::JsEngine jsEngine; |
124 AdblockPlus::JsEngine jsEngine(AdblockPlus::AppInfo(), 0, &webRequest, 0); | |
125 jsEngine.Evaluate("_webRequest.GET('https://easylist.adblockplus.org/easylist.
txt', {}, function(result) {foo = result;} )"); | 104 jsEngine.Evaluate("_webRequest.GET('https://easylist.adblockplus.org/easylist.
txt', {}, function(result) {foo = result;} )"); |
126 do | 105 do |
127 { | 106 { |
128 AdblockPlus::Sleep(200); | 107 AdblockPlus::Sleep(200); |
129 } while (jsEngine.Evaluate("this.foo")->IsUndefined()); | 108 } while (jsEngine.Evaluate("this.foo")->IsUndefined()); |
130 ASSERT_EQ(AdblockPlus::WebRequest::NS_ERROR_FAILURE, jsEngine.Evaluate("foo.st
atus")->AsInt()); | 109 ASSERT_EQ(AdblockPlus::WebRequest::NS_ERROR_FAILURE, jsEngine.Evaluate("foo.st
atus")->AsInt()); |
131 ASSERT_EQ(0, jsEngine.Evaluate("foo.responseStatus")->AsInt()); | 110 ASSERT_EQ(0, jsEngine.Evaluate("foo.responseStatus")->AsInt()); |
132 ASSERT_EQ("", jsEngine.Evaluate("foo.responseText")->AsString()); | 111 ASSERT_EQ("", jsEngine.Evaluate("foo.responseText")->AsString()); |
133 ASSERT_EQ("{}", jsEngine.Evaluate("JSON.stringify(foo.responseHeaders)")->AsSt
ring()); | 112 ASSERT_EQ("{}", jsEngine.Evaluate("JSON.stringify(foo.responseHeaders)")->AsSt
ring()); |
134 } | 113 } |
135 | 114 |
| 115 #if !FILTER_ENGINE_STUBS |
136 TEST(WebRequestTest, XMLHttpRequest) | 116 TEST(WebRequestTest, XMLHttpRequest) |
137 { | 117 { |
138 AdblockPlus::DefaultWebRequest webRequest; | 118 AdblockPlus::JsEngine jsEngine; |
139 AdblockPlus::JsEngine jsEngine(AdblockPlus::AppInfo(), 0, &webRequest, 0); | |
140 AdblockPlus::FilterEngine filterEngine(jsEngine); | 119 AdblockPlus::FilterEngine filterEngine(jsEngine); |
141 | 120 |
142 jsEngine.Evaluate("\ | 121 jsEngine.Evaluate("\ |
143 var result;\ | 122 var result;\ |
144 var request = new XMLHttpRequest();\ | 123 var request = new XMLHttpRequest();\ |
145 request.open('GET', 'https://easylist.adblockplus.org/easylist.txt');\ | 124 request.open('GET', 'https://easylist.adblockplus.org/easylist.txt');\ |
146 request.setRequestHeader('X', 'Y');\ | 125 request.setRequestHeader('X', 'Y');\ |
147 request.overrideMimeType('text/plain');\ | 126 request.overrideMimeType('text/plain');\ |
148 request.addEventListener('load', function() {result = request.responseText;}
, false);\ | 127 request.addEventListener('load', function() {result = request.responseText;}
, false);\ |
149 request.addEventListener('error', function() {result = 'error';}, false);\ | 128 request.addEventListener('error', function() {result = 'error';}, false);\ |
150 request.send(null);"); | 129 request.send(null);"); |
151 do | 130 do |
152 { | 131 { |
153 AdblockPlus::Sleep(200); | 132 AdblockPlus::Sleep(200); |
154 } while (jsEngine.Evaluate("result")->IsUndefined()); | 133 } while (jsEngine.Evaluate("result")->IsUndefined()); |
155 ASSERT_EQ(AdblockPlus::WebRequest::NS_ERROR_FAILURE, jsEngine.Evaluate("reques
t.channel.status")->AsInt()); | 134 ASSERT_EQ(AdblockPlus::WebRequest::NS_ERROR_FAILURE, jsEngine.Evaluate("reques
t.channel.status")->AsInt()); |
156 ASSERT_EQ(0, jsEngine.Evaluate("request.status")->AsInt()); | 135 ASSERT_EQ(0, jsEngine.Evaluate("request.status")->AsInt()); |
157 ASSERT_EQ("error", jsEngine.Evaluate("result")->AsString()); | 136 ASSERT_EQ("error", jsEngine.Evaluate("result")->AsString()); |
158 ASSERT_TRUE(jsEngine.Evaluate("request.getResponseHeader('Content-Type')")->Is
Null()); | 137 ASSERT_TRUE(jsEngine.Evaluate("request.getResponseHeader('Content-Type')")->Is
Null()); |
159 } | 138 } |
160 #endif | 139 #endif |
| 140 |
| 141 #endif |
OLD | NEW |