michael@0: // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
michael@0: // Use of this source code is governed by a BSD-style license that can be
michael@0: // found in the LICENSE file.
michael@0: 
michael@0: #include "sandbox/win/src/win2k_threadpool.h"
michael@0: #include "testing/gtest/include/gtest/gtest.h"
michael@0: 
michael@0: void __stdcall EmptyCallBack(void*, unsigned char) {
michael@0: }
michael@0: 
michael@0: void __stdcall TestCallBack(void* context, unsigned char) {
michael@0:   HANDLE event = reinterpret_cast<HANDLE>(context);
michael@0:   ::SetEvent(event);
michael@0: }
michael@0: 
michael@0: namespace sandbox {
michael@0: 
michael@0: // Test that register and unregister work, part 1.
michael@0: TEST(IPCTest, ThreadPoolRegisterTest1) {
michael@0:   Win2kThreadPool thread_pool;
michael@0: 
michael@0:   EXPECT_EQ(0, thread_pool.OutstandingWaits());
michael@0: 
michael@0:   HANDLE event1 = ::CreateEventW(NULL, FALSE, FALSE, NULL);
michael@0:   HANDLE event2 = ::CreateEventW(NULL, FALSE, FALSE, NULL);
michael@0: 
michael@0:   uint32 context = 0;
michael@0:   EXPECT_FALSE(thread_pool.RegisterWait(0, event1, EmptyCallBack, &context));
michael@0:   EXPECT_EQ(0, thread_pool.OutstandingWaits());
michael@0: 
michael@0:   EXPECT_TRUE(thread_pool.RegisterWait(this, event1, EmptyCallBack, &context));
michael@0:   EXPECT_EQ(1, thread_pool.OutstandingWaits());
michael@0:   EXPECT_TRUE(thread_pool.RegisterWait(this, event2, EmptyCallBack, &context));
michael@0:   EXPECT_EQ(2, thread_pool.OutstandingWaits());
michael@0: 
michael@0:   EXPECT_TRUE(thread_pool.UnRegisterWaits(this));
michael@0:   EXPECT_EQ(0, thread_pool.OutstandingWaits());
michael@0: 
michael@0:   EXPECT_EQ(TRUE, ::CloseHandle(event1));
michael@0:   EXPECT_EQ(TRUE, ::CloseHandle(event2));
michael@0: }
michael@0: 
michael@0: // Test that register and unregister work, part 2.
michael@0: TEST(IPCTest, ThreadPoolRegisterTest2) {
michael@0:   Win2kThreadPool thread_pool;
michael@0: 
michael@0:   HANDLE event1 = ::CreateEventW(NULL, FALSE, FALSE, NULL);
michael@0:   HANDLE event2 = ::CreateEventW(NULL, FALSE, FALSE, NULL);
michael@0: 
michael@0:   uint32 context = 0;
michael@0:   uint32 c1 = 0;
michael@0:   uint32 c2 = 0;
michael@0: 
michael@0:   EXPECT_TRUE(thread_pool.RegisterWait(&c1, event1, EmptyCallBack, &context));
michael@0:   EXPECT_EQ(1, thread_pool.OutstandingWaits());
michael@0:   EXPECT_TRUE(thread_pool.RegisterWait(&c2, event2, EmptyCallBack, &context));
michael@0:   EXPECT_EQ(2, thread_pool.OutstandingWaits());
michael@0: 
michael@0:   EXPECT_TRUE(thread_pool.UnRegisterWaits(&c2));
michael@0:   EXPECT_EQ(1, thread_pool.OutstandingWaits());
michael@0:   EXPECT_TRUE(thread_pool.UnRegisterWaits(&c2));
michael@0:   EXPECT_EQ(1, thread_pool.OutstandingWaits());
michael@0: 
michael@0:   EXPECT_TRUE(thread_pool.UnRegisterWaits(&c1));
michael@0:   EXPECT_EQ(0, thread_pool.OutstandingWaits());
michael@0: 
michael@0:   EXPECT_EQ(TRUE, ::CloseHandle(event1));
michael@0:   EXPECT_EQ(TRUE, ::CloseHandle(event2));
michael@0: }
michael@0: 
michael@0: // Test that the thread pool has at least a thread that services an event.
michael@0: // Test that when the event is un-registered is no longer serviced.
michael@0: TEST(IPCTest, ThreadPoolSignalAndWaitTest) {
michael@0:   Win2kThreadPool thread_pool;
michael@0: 
michael@0:   // The events are auto reset and start not signaled.
michael@0:   HANDLE event1 = ::CreateEventW(NULL, FALSE, FALSE, NULL);
michael@0:   HANDLE event2 = ::CreateEventW(NULL, FALSE, FALSE, NULL);
michael@0: 
michael@0:   EXPECT_TRUE(thread_pool.RegisterWait(this, event1, TestCallBack, event2));
michael@0: 
michael@0:   EXPECT_EQ(WAIT_OBJECT_0, ::SignalObjectAndWait(event1, event2, 5000, FALSE));
michael@0:   EXPECT_EQ(WAIT_OBJECT_0, ::SignalObjectAndWait(event1, event2, 5000, FALSE));
michael@0: 
michael@0:   EXPECT_TRUE(thread_pool.UnRegisterWaits(this));
michael@0:   EXPECT_EQ(0, thread_pool.OutstandingWaits());
michael@0: 
michael@0:   EXPECT_EQ(WAIT_TIMEOUT, ::SignalObjectAndWait(event1, event2, 1000, FALSE));
michael@0: 
michael@0:   EXPECT_EQ(TRUE, ::CloseHandle(event1));
michael@0:   EXPECT_EQ(TRUE, ::CloseHandle(event2));
michael@0: }
michael@0: 
michael@0: }  // namespace sandbox