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 |
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 "BaseJsTest.h" | 18 #include "BaseJsTest.h" |
19 | 19 |
20 using namespace AdblockPlus; | 20 using namespace AdblockPlus; |
21 | 21 |
22 // This define enables NotificationMockWebRequestTest but to run it | 22 // This define enables NotificationMockWebRequestTest but to run it |
23 // one need to set INITIAL_DELAY to about 2000 msec in notification.js. | 23 // one need to set INITIAL_DELAY to about 2000 msec in notification.js. |
24 //#define NotificationMockWebRequestTest_ENABLED | 24 //#define NotificationMockWebRequestTest_ENABLED |
25 | 25 |
26 namespace | 26 namespace |
27 { | 27 { |
28 typedef std::shared_ptr<FilterEngine> FilterEnginePtr; | |
29 | |
30 class NotificationTest : public BaseJsTest | 28 class NotificationTest : public BaseJsTest |
31 { | 29 { |
32 protected: | 30 protected: |
33 FilterEnginePtr filterEngine; | |
34 void SetUp() | 31 void SetUp() |
35 { | 32 { |
36 LazyFileSystem* fileSystem; | 33 LazyFileSystem* fileSystem; |
37 ThrowingPlatformCreationParameters platformParams; | 34 ThrowingPlatformCreationParameters platformParams; |
38 platformParams.timer.reset(new NoopTimer()); | 35 platformParams.timer.reset(new NoopTimer()); |
39 platformParams.fileSystem.reset(fileSystem = new LazyFileSystem()); | 36 platformParams.fileSystem.reset(fileSystem = new LazyFileSystem()); |
40 platformParams.webRequest.reset(new NoopWebRequest()); | 37 platformParams.webRequest.reset(new NoopWebRequest()); |
41 platform.reset(new Platform(std::move(platformParams))); | 38 platform.reset(new Platform(std::move(platformParams))); |
42 filterEngine = CreateFilterEngine(*fileSystem, *platform); | 39 CreateFilterEngine(*fileSystem, *platform); |
43 } | 40 } |
44 | 41 |
45 void AddNotification(const std::string& notification) | 42 void AddNotification(const std::string& notification) |
46 { | 43 { |
47 GetJsEngine().Evaluate("(function()" | 44 GetJsEngine().Evaluate("(function()" |
48 "{" | 45 "{" |
49 "require('notification').Notification.addNotification(" + notification +
");" | 46 "require('notification').Notification.addNotification(" + notification +
");" |
50 "})();"); | 47 "})();"); |
51 } | 48 } |
52 | 49 |
53 std::unique_ptr<Notification> PeekNotification(const std::string& url = std:
:string()) | 50 std::unique_ptr<Notification> PeekNotification(const std::string& url = std:
:string()) |
54 { | 51 { |
55 std::unique_ptr<Notification> retValue; | 52 std::unique_ptr<Notification> retValue; |
56 filterEngine->SetShowNotificationCallback( | 53 auto& filterEngine = platform->GetFilterEngine(); |
| 54 filterEngine.SetShowNotificationCallback( |
57 [&retValue](Notification&& notification) { | 55 [&retValue](Notification&& notification) { |
58 retValue.reset(new Notification(std::move(notification))); | 56 retValue.reset(new Notification(std::move(notification))); |
59 }); | 57 }); |
60 filterEngine->ShowNextNotification(url); | 58 filterEngine.ShowNextNotification(url); |
61 filterEngine->RemoveShowNotificationCallback(); | 59 filterEngine.RemoveShowNotificationCallback(); |
62 return retValue; | 60 return retValue; |
63 } | 61 } |
64 }; | 62 }; |
65 | 63 |
66 class MockWebRequest : public IWebRequest | 64 class MockWebRequest : public IWebRequest |
67 { | 65 { |
68 public: | 66 public: |
69 std::string responseText; | 67 std::string responseText; |
70 explicit MockWebRequest(const std::string& notification) | 68 explicit MockWebRequest(const std::string& notification) |
71 : responseText(notification) | 69 : responseText(notification) |
(...skipping 11 matching lines...) Expand all Loading... |
83 serverResponse.responseStatus = 200; | 81 serverResponse.responseStatus = 200; |
84 serverResponse.responseText = responseText; | 82 serverResponse.responseText = responseText; |
85 getCallback(serverResponse); | 83 getCallback(serverResponse); |
86 } | 84 } |
87 }; | 85 }; |
88 | 86 |
89 #ifdef NotificationMockWebRequestTest_ENABLED | 87 #ifdef NotificationMockWebRequestTest_ENABLED |
90 class NotificationMockWebRequestTest : public BaseJsTest | 88 class NotificationMockWebRequestTest : public BaseJsTest |
91 { | 89 { |
92 protected: | 90 protected: |
93 FilterEnginePtr filterEngine; | |
94 bool isNotificationCallbackCalled; | 91 bool isNotificationCallbackCalled; |
95 void SetUp() | 92 void SetUp() |
96 { | 93 { |
97 BaseJsTest::SetUp(); | 94 BaseJsTest::SetUp(); |
98 isNotificationCallbackCalled = false; | 95 isNotificationCallbackCalled = false; |
99 jsEngine->SetFileSystem( | 96 jsEngine->SetFileSystem( |
100 std::shared_ptr<LazyFileSystem>(new LazyFileSystem())); | 97 std::shared_ptr<LazyFileSystem>(new LazyFileSystem())); |
101 const char* responseJsonText = "{" | 98 const char* responseJsonText = "{" |
102 "\"notifications\": [{" | 99 "\"notifications\": [{" |
103 "\"id\": \"some id\"," | 100 "\"id\": \"some id\"," |
(...skipping 92 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
196 TEST_F(NotificationTest, Links) | 193 TEST_F(NotificationTest, Links) |
197 { | 194 { |
198 AddNotification("{ id: 'id', links: ['link1', 'link2'] }"); | 195 AddNotification("{ id: 'id', links: ['link1', 'link2'] }"); |
199 auto notification = PeekNotification(); | 196 auto notification = PeekNotification(); |
200 ASSERT_TRUE(notification); | 197 ASSERT_TRUE(notification); |
201 std::vector<std::string> notificationLinks = notification->GetLinks(); | 198 std::vector<std::string> notificationLinks = notification->GetLinks(); |
202 ASSERT_EQ(2u, notificationLinks.size()); | 199 ASSERT_EQ(2u, notificationLinks.size()); |
203 EXPECT_EQ("link1", notificationLinks[0]); | 200 EXPECT_EQ("link1", notificationLinks[0]); |
204 EXPECT_EQ("link2", notificationLinks[1]); | 201 EXPECT_EQ("link2", notificationLinks[1]); |
205 } | 202 } |
OLD | NEW |