15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 52a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/win/object_watcher.h" 62a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <process.h> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch#include "base/message_loop/message_loop.h" 102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/run_loop.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace base { 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace win { 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class QuitDelegate : public ObjectWatcher::Delegate { 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnObjectSignaled(HANDLE object) { 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) MessageLoop::current()->QuitWhenIdle(); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DecrementCountDelegate : public ObjectWatcher::Delegate { 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit DecrementCountDelegate(int* counter) : counter_(counter) { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void OnObjectSignaled(HANDLE object) { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) --(*counter_); 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int* counter_; 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RunTest_BasicSignal(MessageLoop::Type message_loop_type) { 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MessageLoop message_loop(message_loop_type); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ObjectWatcher watcher; 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(NULL, watcher.GetWatchedObject()); 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A manual-reset event that is not yet signaled. 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE event = CreateEvent(NULL, TRUE, FALSE, NULL); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuitDelegate delegate; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool ok = watcher.StartWatching(event, &delegate); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(ok); 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(event, watcher.GetWatchedObject()); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetEvent(event); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MessageLoop::current()->Run(); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(NULL, watcher.GetWatchedObject()); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseHandle(event); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RunTest_BasicCancel(MessageLoop::Type message_loop_type) { 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MessageLoop message_loop(message_loop_type); 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ObjectWatcher watcher; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A manual-reset event that is not yet signaled. 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE event = CreateEvent(NULL, TRUE, FALSE, NULL); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuitDelegate delegate; 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool ok = watcher.StartWatching(event, &delegate); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(ok); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) watcher.StopWatching(); 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseHandle(event); 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RunTest_CancelAfterSet(MessageLoop::Type message_loop_type) { 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MessageLoop message_loop(message_loop_type); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ObjectWatcher watcher; 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int counter = 1; 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DecrementCountDelegate delegate(&counter); 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A manual-reset event that is not yet signaled. 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE event = CreateEvent(NULL, TRUE, FALSE, NULL); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool ok = watcher.StartWatching(event, &delegate); 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(ok); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SetEvent(event); 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Let the background thread do its business 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Sleep(30); 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) watcher.StopWatching(); 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) RunLoop().RunUntilIdle(); 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Our delegate should not have fired. 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, counter); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseHandle(event); 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RunTest_SignalBeforeWatch(MessageLoop::Type message_loop_type) { 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MessageLoop message_loop(message_loop_type); 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ObjectWatcher watcher; 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // A manual-reset event that is signaled before we begin watching. 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE event = CreateEvent(NULL, TRUE, TRUE, NULL); 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuitDelegate delegate; 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool ok = watcher.StartWatching(event, &delegate); 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(ok); 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MessageLoop::current()->Run(); 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(NULL, watcher.GetWatchedObject()); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseHandle(event); 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RunTest_OutlivesMessageLoop(MessageLoop::Type message_loop_type) { 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Simulate a MessageLoop that dies before an ObjectWatcher. This ordinarily 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // doesn't happen when people use the Thread class, but it can happen when 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // people use the Singleton pattern or atexit. 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HANDLE event = CreateEvent(NULL, TRUE, FALSE, NULL); // not signaled 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ObjectWatcher watcher; 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MessageLoop message_loop(message_loop_type); 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) QuitDelegate delegate; 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) watcher.StartWatching(event, &delegate); 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CloseHandle(event); 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)//----------------------------------------------------------------------------- 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(ObjectWatcherTest, BasicSignal) { 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_BasicSignal(MessageLoop::TYPE_DEFAULT); 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_BasicSignal(MessageLoop::TYPE_IO); 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_BasicSignal(MessageLoop::TYPE_UI); 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(ObjectWatcherTest, BasicCancel) { 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_BasicCancel(MessageLoop::TYPE_DEFAULT); 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_BasicCancel(MessageLoop::TYPE_IO); 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_BasicCancel(MessageLoop::TYPE_UI); 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(ObjectWatcherTest, CancelAfterSet) { 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_CancelAfterSet(MessageLoop::TYPE_DEFAULT); 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_CancelAfterSet(MessageLoop::TYPE_IO); 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_CancelAfterSet(MessageLoop::TYPE_UI); 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(ObjectWatcherTest, SignalBeforeWatch) { 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_SignalBeforeWatch(MessageLoop::TYPE_DEFAULT); 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_SignalBeforeWatch(MessageLoop::TYPE_IO); 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_SignalBeforeWatch(MessageLoop::TYPE_UI); 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(ObjectWatcherTest, OutlivesMessageLoop) { 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_OutlivesMessageLoop(MessageLoop::TYPE_DEFAULT); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_OutlivesMessageLoop(MessageLoop::TYPE_IO); 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) RunTest_OutlivesMessageLoop(MessageLoop::TYPE_UI); 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace win 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace base 175