| 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-2016 Eyeo GmbH | 3 * Copyright (C) 2006-2016 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 |
| 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 12 * GNU General Public License for more details. | 12 * GNU General Public License for more details. |
| 13 * | 13 * |
| 14 * You should have received a copy of the GNU General Public License | 14 * You should have received a copy of the GNU General Public License |
| 15 * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. | 15 * along with Adblock Plus. If not, see <http://www.gnu.org/licenses/>. |
| 16 */ | 16 */ |
| 17 | 17 |
| 18 #include <functional> | 18 #include <functional> |
| 19 | 19 |
| 20 #include "BaseJsTest.h" | 20 #include "BaseJsTest.h" |
| 21 #include "../src/JsEngineTransition.h" |
| 21 | 22 |
| 22 namespace | 23 namespace |
| 23 { | 24 { |
| 24 typedef std::shared_ptr<AdblockPlus::FilterEngine> FilterEnginePtr; | 25 typedef std::shared_ptr<AdblockPlus::FilterEngine> FilterEnginePtr; |
| 25 | 26 |
| 26 void FindAndReplace(std::string& source, const std::string& find, const std::s
tring& replace) | 27 void FindAndReplace(std::string& source, const std::string& find, const std::s
tring& replace) |
| 27 { | 28 { |
| 28 for (size_t pos = 0; (pos = source.find(find), pos) != std::string::npos; po
s += replace.size()) | 29 for (size_t pos = 0; (pos = source.find(find), pos) != std::string::npos; po
s += replace.size()) |
| 29 source.replace(pos, find.size(), replace); | 30 source.replace(pos, find.size(), replace); |
| 30 } | 31 } |
| 31 | 32 |
| 32 std::string previousRequestUrl; | 33 std::string previousRequestUrl; |
| 33 class TestWebRequest : public LazyWebRequest | 34 class TestWebRequest : public LazyWebRequest |
| 34 { | 35 { |
| 35 public: | 36 public: |
| 36 AdblockPlus::ServerResponse response; | 37 AdblockPlus::ServerResponse response; |
| 37 AdblockPlus::ServerResponse GET(const std::string& url, const AdblockPlus::H
eaderList& requestHeaders) const | 38 AdblockPlus::ServerResponse GET(const std::string& url, const AdblockPlus::H
eaderList& requestHeaders) const |
| 38 { | 39 { |
| 39 if (url.find("easylist") != std::string::npos) | 40 if (url.find("easylist") != std::string::npos) |
| 40 return LazyWebRequest::GET(url, requestHeaders); | 41 return LazyWebRequest::GET(url, requestHeaders); |
| 41 | 42 |
| 42 previousRequestUrl = url; | 43 previousRequestUrl = url; |
| 43 return response; | 44 return response; |
| 44 } | 45 } |
| 45 }; | 46 }; |
| 46 | 47 |
| 47 class UpdateCheckTest : public ::testing::Test | 48 class UpdateCheckTest : public ::testing::Test |
| 48 { | 49 { |
| 50 void Construct() |
| 51 { |
| 52 jsEngine = CreateJsEngine(appInfo); |
| 53 jsEngine->SetLogSystem(std::make_shared<LazyLogSystem>()); |
| 54 jsEngine->SetFileSystem(std::make_shared<LazyFileSystem>()); |
| 55 jsEngine->SetWebRequest(webRequest); |
| 56 jsEngine->SetEventCallback("updateAvailable", |
| 57 std::bind(&UpdateCheckTest::EventCallback, this, std::placeholders::_1))
; |
| 58 |
| 59 filterEngine = std::make_shared<AdblockPlus::FilterEngine>(jsEngine); |
| 60 } |
| 61 |
| 62 void Destroy() |
| 63 { |
| 64 webRequest->Cancel(); |
| 65 filterEngine.reset(); |
| 66 ToInternal(jsEngine)->WaitForQuietScheduler(); |
| 67 jsEngine.reset(); |
| 68 ASSERT_EQ(0, jsEngine.use_count()); |
| 69 } |
| 70 |
| 49 protected: | 71 protected: |
| 50 AdblockPlus::AppInfo appInfo; | 72 AdblockPlus::AppInfo appInfo; |
| 51 TestWebRequest* webRequest; | 73 std::shared_ptr<TestWebRequest> webRequest; |
| 52 AdblockPlus::WebRequestPtr webRequestPtr; | |
| 53 AdblockPlus::JsEnginePtr jsEngine; | 74 AdblockPlus::JsEnginePtr jsEngine; |
| 54 FilterEnginePtr filterEngine; | 75 FilterEnginePtr filterEngine; |
| 55 | 76 |
| 56 bool eventCallbackCalled; | 77 bool eventCallbackCalled; |
| 57 AdblockPlus::JsValueList eventCallbackParams; | 78 AdblockPlus::JsValueList eventCallbackParams; |
| 58 bool updateCallbackCalled; | 79 bool updateCallbackCalled; |
| 59 std::string updateError; | 80 std::string updateError; |
| 60 | 81 |
| 61 void SetUp() | 82 void SetUp() |
| 62 { | 83 { |
| 63 webRequest = new TestWebRequest(); | 84 webRequest = std::make_shared<TestWebRequest>(); |
| 64 webRequestPtr.reset(webRequest); | |
| 65 | |
| 66 eventCallbackCalled = false; | 85 eventCallbackCalled = false; |
| 67 updateCallbackCalled = false; | 86 updateCallbackCalled = false; |
| 68 Reset(); | 87 Construct(); |
| 69 } | 88 } |
| 70 | 89 |
| 71 void TearDown() | 90 void TearDown() |
| 72 { | 91 { |
| 73 webRequest->Cancel(); | 92 Destroy(); |
| 93 webRequest.reset(); |
| 74 } | 94 } |
| 75 | 95 |
| 76 void Reset() | 96 void Reset() |
| 77 { | 97 { |
| 78 jsEngine = CreateJsEngine(appInfo); | 98 Destroy(); |
| 79 jsEngine->SetLogSystem(AdblockPlus::LogSystemPtr(new LazyLogSystem)); | 99 Construct(); |
| 80 jsEngine->SetFileSystem(AdblockPlus::FileSystemPtr(new LazyFileSystem)); | |
| 81 jsEngine->SetWebRequest(webRequestPtr); | |
| 82 jsEngine->SetEventCallback("updateAvailable", | |
| 83 std::bind(&UpdateCheckTest::EventCallback, this, std::placeholders::_1
)); | |
| 84 | |
| 85 filterEngine.reset(new AdblockPlus::FilterEngine(jsEngine)); | |
| 86 } | 100 } |
| 87 | 101 |
| 88 void ForceUpdateCheck() | 102 void ForceUpdateCheck() |
| 89 { | 103 { |
| 90 filterEngine->ForceUpdateCheck( | 104 filterEngine->ForceUpdateCheck( |
| 91 std::bind(&UpdateCheckTest::UpdateCallback, this, std::placeholders::_
1)); | 105 std::bind(&UpdateCheckTest::UpdateCallback, this, std::placeholders::_
1)); |
| 92 } | 106 } |
| 93 | 107 |
| 94 void EventCallback(AdblockPlus::JsValueList& params) | 108 void EventCallback(AdblockPlus::JsValueList& params) |
| 95 { | 109 { |
| (...skipping 165 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 261 | 275 |
| 262 Reset(); | 276 Reset(); |
| 263 ForceUpdateCheck(); | 277 ForceUpdateCheck(); |
| 264 | 278 |
| 265 std::this_thread::sleep_for(std::chrono::milliseconds(100)); | 279 std::this_thread::sleep_for(std::chrono::milliseconds(100)); |
| 266 | 280 |
| 267 ASSERT_FALSE(eventCallbackCalled); | 281 ASSERT_FALSE(eventCallbackCalled); |
| 268 ASSERT_TRUE(updateCallbackCalled); | 282 ASSERT_TRUE(updateCallbackCalled); |
| 269 ASSERT_FALSE(updateError.empty()); | 283 ASSERT_FALSE(updateError.empty()); |
| 270 } | 284 } |
| OLD | NEW |