base/threading/worker_pool_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/threading/worker_pool.h"
   6
   7 #include "base/bind.h"
   8 #include "base/bind_helpers.h"
   9 #include "base/location.h"
  10 #include "base/message_loop/message_loop.h"
  11 #include "base/run_loop.h"
  12 #include "base/synchronization/waitable_event.h"
  13 #include "base/test/test_timeouts.h"
  14 #include "base/threading/thread_checker_impl.h"
  15 #include "base/time/time.h"
  16 #include "testing/gtest/include/gtest/gtest.h"
  17 #include "testing/platform_test.h"
  18
  19 typedef PlatformTest WorkerPoolTest;
  20
  21 namespace base {
  22
  23 namespace {
  24
  25 class PostTaskAndReplyTester
  26     : public base::RefCountedThreadSafe<PostTaskAndReplyTester> {
  27  public:
  28   PostTaskAndReplyTester() : finished_(false), test_event_(false, false) {}
  29
  30   void RunTest() {
  31     ASSERT_TRUE(thread_checker_.CalledOnValidThread());
  32     WorkerPool::PostTaskAndReply(
  33       FROM_HERE,
  34       base::Bind(&PostTaskAndReplyTester::OnWorkerThread, this),
  35       base::Bind(&PostTaskAndReplyTester::OnOriginalThread, this),
  36       false);
  37
  38     test_event_.Wait();
  39   }
  40
  41   void OnWorkerThread() {
  42     // We're not on the original thread.
  43     EXPECT_FALSE(thread_checker_.CalledOnValidThread());
  44
  45     test_event_.Signal();
  46   }
  47
  48   void OnOriginalThread() {
  49     EXPECT_TRUE(thread_checker_.CalledOnValidThread());
  50     finished_ = true;
  51   }
  52
  53   bool finished() const {
  54     return finished_;
  55   }
  56
  57  private:
  58   friend class base::RefCountedThreadSafe<PostTaskAndReplyTester>;
  59   ~PostTaskAndReplyTester() {}
  60
  61   bool finished_;
  62   WaitableEvent test_event_;
  63
  64   // The Impl version performs its checks even in release builds.
  65   ThreadCheckerImpl thread_checker_;
  66 };
  67
  68 }  // namespace
  69
  70 TEST_F(WorkerPoolTest, PostTask) {
  71   WaitableEvent test_event(false, false);
  72   WaitableEvent long_test_event(false, false);
  73
  74   WorkerPool::PostTask(FROM_HERE,
  75                        base::Bind(&WaitableEvent::Signal,
  76                                   base::Unretained(&test_event)),
  77                        false);
  78   WorkerPool::PostTask(FROM_HERE,
  79                        base::Bind(&WaitableEvent::Signal,
  80                                   base::Unretained(&long_test_event)),
  81                        true);
  82
  83   test_event.Wait();
  84   long_test_event.Wait();
  85 }
  86
  87 #if defined(OS_WIN) || defined(OS_LINUX)
  88 // Flaky on Windows and Linux (http://crbug.com/130337)
  89 #define MAYBE_PostTaskAndReply DISABLED_PostTaskAndReply
  90 #else
  91 #define MAYBE_PostTaskAndReply PostTaskAndReply
  92 #endif
  93
  94 TEST_F(WorkerPoolTest, MAYBE_PostTaskAndReply) {
  95   MessageLoop message_loop;
  96   scoped_refptr<PostTaskAndReplyTester> tester(new PostTaskAndReplyTester());
  97   tester->RunTest();
  98
  99   const TimeDelta kMaxDuration = TestTimeouts::tiny_timeout();
 100   TimeTicks start = TimeTicks::Now();
 101   while (!tester->finished() && TimeTicks::Now() - start < kMaxDuration) {
 102 #if defined(OS_IOS)
 103     // Ensure that the other thread has a chance to run even on a single-core
 104     // device.
 105     pthread_yield_np();
 106 #endif
 107     RunLoop().RunUntilIdle();
 108   }
 109   EXPECT_TRUE(tester->finished());
 110 }
 111
 112 }  // namespace base