| Index: ui/compositor/compositor_unittest.cc
|
| diff --git a/ui/compositor/compositor_unittest.cc b/ui/compositor/compositor_unittest.cc
|
| index e62e0b8c430a84131a25aeb4a6c3ee619a3eab11..a5e9ca2655cdf7dd086030c1aa59a1da8e582f7b 100644
|
| --- a/ui/compositor/compositor_unittest.cc
|
| +++ b/ui/compositor/compositor_unittest.cc
|
| @@ -62,31 +62,39 @@ class CompositorTest : public testing::Test {
|
| DISALLOW_COPY_AND_ASSIGN(CompositorTest);
|
| };
|
|
|
| +class MockCompositorLockClient
|
| + : NON_EXPORTED_BASE(public ui::CompositorLockClient) {
|
| + public:
|
| + MOCK_METHOD0(CompositorLockTimedOut, void());
|
| +};
|
| +
|
| } // namespace
|
|
|
| TEST_F(CompositorTest, LocksTimeOut) {
|
| scoped_refptr<ui::CompositorLock> lock;
|
| +
|
| + base::TimeDelta timeout = base::TimeDelta::FromMilliseconds(100);
|
| +
|
| {
|
| + testing::StrictMock<MockCompositorLockClient> lock_client;
|
| base::RunLoop run_loop;
|
| - // Ensure that the lock times out by default.
|
| - lock = compositor()->GetCompositorLock();
|
| + // This lock has a timeout.
|
| + lock = compositor()->GetCompositorLock(&lock_client, timeout);
|
| EXPECT_TRUE(compositor()->IsLocked());
|
| - task_runner()->PostDelayedTask(
|
| - FROM_HERE, run_loop.QuitClosure(),
|
| - base::TimeDelta::FromMilliseconds(kCompositorLockTimeoutMs));
|
| + task_runner()->PostDelayedTask(FROM_HERE, run_loop.QuitClosure(), timeout);
|
| + EXPECT_CALL(lock_client, CompositorLockTimedOut()).Times(1);
|
| run_loop.Run();
|
| EXPECT_FALSE(compositor()->IsLocked());
|
| }
|
|
|
| {
|
| + testing::StrictMock<MockCompositorLockClient> lock_client;
|
| base::RunLoop run_loop;
|
| - // Ensure that the lock does not time out when set.
|
| - compositor()->SetLocksWillTimeOut(false);
|
| - lock = compositor()->GetCompositorLock();
|
| + // This lock has no timeout.
|
| + lock = compositor()->GetCompositorLock(&lock_client, base::TimeDelta());
|
| EXPECT_TRUE(compositor()->IsLocked());
|
| - task_runner()->PostDelayedTask(
|
| - FROM_HERE, run_loop.QuitClosure(),
|
| - base::TimeDelta::FromMilliseconds(kCompositorLockTimeoutMs));
|
| + task_runner()->PostDelayedTask(FROM_HERE, run_loop.QuitClosure(), timeout);
|
| + EXPECT_CALL(lock_client, CompositorLockTimedOut()).Times(0);
|
| run_loop.Run();
|
| EXPECT_TRUE(compositor()->IsLocked());
|
| }
|
|
|