| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "modules/wake_lock/ScreenWakeLock.h" | 5 #include "modules/wake_lock/ScreenWakeLock.h" |
| 6 | 6 |
| 7 #include "core/dom/DOMImplementation.h" | 7 #include "core/dom/DOMImplementation.h" |
| 8 #include "core/dom/Document.h" | 8 #include "core/dom/Document.h" |
| 9 #include "core/dom/DocumentInit.h" | 9 #include "core/dom/DocumentInit.h" |
| 10 #include "mojo/public/cpp/bindings/interface_request.h" | 10 #include "mojo/public/cpp/bindings/interface_request.h" |
| 11 #include "mojo/public/cpp/bindings/strong_binding.h" | 11 #include "mojo/public/cpp/bindings/strong_binding.h" |
| 12 #include "platform/testing/URLTestHelpers.h" | 12 #include "platform/testing/URLTestHelpers.h" |
| 13 #include "platform/testing/UnitTestHelpers.h" | 13 #include "platform/testing/UnitTestHelpers.h" |
| 14 #include "public/platform/InterfaceProvider.h" |
| 14 #include "public/platform/Platform.h" | 15 #include "public/platform/Platform.h" |
| 15 #include "public/platform/ServiceRegistry.h" | |
| 16 #include "public/platform/WebPageVisibilityState.h" | 16 #include "public/platform/WebPageVisibilityState.h" |
| 17 #include "public/platform/WebURLLoaderMockFactory.h" | 17 #include "public/platform/WebURLLoaderMockFactory.h" |
| 18 #include "public/web/WebCache.h" | 18 #include "public/web/WebCache.h" |
| 19 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
| 20 #include "web/WebLocalFrameImpl.h" | 20 #include "web/WebLocalFrameImpl.h" |
| 21 #include "web/tests/FrameTestHelpers.h" | 21 #include "web/tests/FrameTestHelpers.h" |
| 22 | 22 |
| 23 #include <memory> | 23 #include <memory> |
| 24 | 24 |
| 25 namespace { | 25 namespace { |
| 26 | 26 |
| 27 using blink::ScreenWakeLock; | 27 using blink::ScreenWakeLock; |
| 28 using blink::mojom::blink::WakeLockService; | 28 using blink::mojom::blink::WakeLockService; |
| 29 using blink::mojom::blink::WakeLockServiceRequest; | 29 using blink::mojom::blink::WakeLockServiceRequest; |
| 30 | 30 |
| 31 // This class allows connecting service requests to a MockWakeLockService. | 31 // This class allows binding interface requests to a MockWakeLockService. |
| 32 class MockServiceRegistry : public blink::ServiceRegistry { | 32 class MockInterfaceProvider : public blink::InterfaceProvider { |
| 33 public: | 33 public: |
| 34 MockServiceRegistry() : m_wakeLockStatus(false) {} | 34 MockInterfaceProvider() : m_wakeLockStatus(false) {} |
| 35 ~MockServiceRegistry() {} | 35 ~MockInterfaceProvider() {} |
| 36 | 36 |
| 37 void connectToRemoteService(const char* name, mojo::ScopedMessagePipeHandle)
override; | 37 void getInterface(const char* name, mojo::ScopedMessagePipeHandle) override; |
| 38 | 38 |
| 39 bool wakeLockStatus() const { return m_wakeLockStatus; } | 39 bool wakeLockStatus() const { return m_wakeLockStatus; } |
| 40 void setWakeLockStatus(bool status) { m_wakeLockStatus = status; } | 40 void setWakeLockStatus(bool status) { m_wakeLockStatus = status; } |
| 41 | 41 |
| 42 private: | 42 private: |
| 43 // A mock WakeLockService used to intercept calls to the mojo methods. | 43 // A mock WakeLockService used to intercept calls to the mojo methods. |
| 44 class MockWakeLockService : public WakeLockService { | 44 class MockWakeLockService : public WakeLockService { |
| 45 public: | 45 public: |
| 46 MockWakeLockService(MockServiceRegistry* registry, WakeLockServiceReques
t request) | 46 MockWakeLockService(MockInterfaceProvider* registry, WakeLockServiceRequ
est request) |
| 47 : m_binding(this, std::move(request)) | 47 : m_binding(this, std::move(request)) |
| 48 , m_registry(registry) {} | 48 , m_registry(registry) {} |
| 49 ~MockWakeLockService() {} | 49 ~MockWakeLockService() {} |
| 50 | 50 |
| 51 private: | 51 private: |
| 52 // mojom::WakeLockService | 52 // mojom::WakeLockService |
| 53 void RequestWakeLock() override { m_registry->setWakeLockStatus(true); } | 53 void RequestWakeLock() override { m_registry->setWakeLockStatus(true); } |
| 54 void CancelWakeLock() override { m_registry->setWakeLockStatus(false); } | 54 void CancelWakeLock() override { m_registry->setWakeLockStatus(false); } |
| 55 | 55 |
| 56 mojo::Binding<WakeLockService> m_binding; | 56 mojo::Binding<WakeLockService> m_binding; |
| 57 MockServiceRegistry* const m_registry; | 57 MockInterfaceProvider* const m_registry; |
| 58 }; | 58 }; |
| 59 std::unique_ptr<MockWakeLockService> m_mockWakeLockService; | 59 std::unique_ptr<MockWakeLockService> m_mockWakeLockService; |
| 60 | 60 |
| 61 bool m_wakeLockStatus; | 61 bool m_wakeLockStatus; |
| 62 }; | 62 }; |
| 63 | 63 |
| 64 void MockServiceRegistry::connectToRemoteService(const char* name, mojo::ScopedM
essagePipeHandle handle) | 64 void MockInterfaceProvider::getInterface(const char* name, mojo::ScopedMessagePi
peHandle handle) |
| 65 { | 65 { |
| 66 m_mockWakeLockService.reset( | 66 m_mockWakeLockService.reset( |
| 67 new MockWakeLockService(this, mojo::MakeRequest<WakeLockService>(std::mo
ve(handle)))); | 67 new MockWakeLockService(this, mojo::MakeRequest<WakeLockService>(std::mo
ve(handle)))); |
| 68 } | 68 } |
| 69 | 69 |
| 70 // A TestWebFrameClient to allow overriding the serviceRegistry() with a mock. | 70 // A TestWebFrameClient to allow overriding the interfaceProvider() with a mock. |
| 71 class TestWebFrameClient : public blink::FrameTestHelpers::TestWebFrameClient { | 71 class TestWebFrameClient : public blink::FrameTestHelpers::TestWebFrameClient { |
| 72 public: | 72 public: |
| 73 ~TestWebFrameClient() override = default; | 73 ~TestWebFrameClient() override = default; |
| 74 blink::ServiceRegistry* serviceRegistry() override { return &m_serviceRegist
ry; } | 74 blink::InterfaceProvider* interfaceProvider() override { return &m_interface
Provider; } |
| 75 | 75 |
| 76 private: | 76 private: |
| 77 MockServiceRegistry m_serviceRegistry; | 77 MockInterfaceProvider m_interfaceProvider; |
| 78 }; | 78 }; |
| 79 | 79 |
| 80 class ScreenWakeLockTest: public testing::Test { | 80 class ScreenWakeLockTest: public testing::Test { |
| 81 protected: | 81 protected: |
| 82 void SetUp() override | 82 void SetUp() override |
| 83 { | 83 { |
| 84 m_webViewHelper.initialize(true, &m_testWebFrameClient); | 84 m_webViewHelper.initialize(true, &m_testWebFrameClient); |
| 85 blink::URLTestHelpers::registerMockedURLFromBaseURL( | 85 blink::URLTestHelpers::registerMockedURLFromBaseURL( |
| 86 blink::WebString::fromUTF8("http://example.com/"), | 86 blink::WebString::fromUTF8("http://example.com/"), |
| 87 blink::WebString::fromUTF8("foo.html")); | 87 blink::WebString::fromUTF8("foo.html")); |
| (...skipping 30 matching lines...) Expand all Loading... |
| 118 } | 118 } |
| 119 | 119 |
| 120 bool screenKeepAwake() | 120 bool screenKeepAwake() |
| 121 { | 121 { |
| 122 DCHECK(screen()); | 122 DCHECK(screen()); |
| 123 return ScreenWakeLock::keepAwake(*screen()); | 123 return ScreenWakeLock::keepAwake(*screen()); |
| 124 } | 124 } |
| 125 | 125 |
| 126 bool clientKeepScreenAwake() | 126 bool clientKeepScreenAwake() |
| 127 { | 127 { |
| 128 return static_cast<MockServiceRegistry*>(m_testWebFrameClient.serviceReg
istry())->wakeLockStatus(); | 128 return static_cast<MockInterfaceProvider*>(m_testWebFrameClient.interfac
eProvider())->wakeLockStatus(); |
| 129 } | 129 } |
| 130 | 130 |
| 131 void setKeepAwake(bool keepAwake) | 131 void setKeepAwake(bool keepAwake) |
| 132 { | 132 { |
| 133 DCHECK(screen()); | 133 DCHECK(screen()); |
| 134 ScreenWakeLock::setKeepAwake(*screen(), keepAwake); | 134 ScreenWakeLock::setKeepAwake(*screen(), keepAwake); |
| 135 // Let the notification sink through the mojo pipes. | 135 // Let the notification sink through the mojo pipes. |
| 136 blink::testing::runPendingTasks(); | 136 blink::testing::runPendingTasks(); |
| 137 } | 137 } |
| 138 | 138 |
| (...skipping 78 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 217 ASSERT_FALSE(clientKeepScreenAwake()); | 217 ASSERT_FALSE(clientKeepScreenAwake()); |
| 218 | 218 |
| 219 setKeepAwake(true); | 219 setKeepAwake(true); |
| 220 loadFrame(); | 220 loadFrame(); |
| 221 | 221 |
| 222 EXPECT_FALSE(screenKeepAwake()); | 222 EXPECT_FALSE(screenKeepAwake()); |
| 223 EXPECT_FALSE(clientKeepScreenAwake()); | 223 EXPECT_FALSE(clientKeepScreenAwake()); |
| 224 } | 224 } |
| 225 | 225 |
| 226 } // namespace | 226 } // namespace |
| OLD | NEW |