src/third_party/webrtc/base/filelock_unittest.cc

   1 /*
   2  *  Copyright 2009 The WebRTC Project Authors. All rights reserved.
   3  *
   4  *  Use of this source code is governed by a BSD-style license
   5  *  that can be found in the LICENSE file in the root of the source
   6  *  tree. An additional intellectual property rights grant can be found
   7  *  in the file PATENTS.  All contributing project authors may
   8  *  be found in the AUTHORS file in the root of the source tree.
   9  */
  10
  11 #include <string>
  12
  13 #include "webrtc/base/event.h"
  14 #include "webrtc/base/filelock.h"
  15 #include "webrtc/base/fileutils.h"
  16 #include "webrtc/base/gunit.h"
  17 #include "webrtc/base/pathutils.h"
  18 #include "webrtc/base/scoped_ptr.h"
  19 #include "webrtc/base/thread.h"
  20 #include "webrtc/test/testsupport/gtest_disable.h"
  21
  22 namespace rtc {
  23
  24 const static std::string kLockFile = "TestLockFile";
  25 const static int kTimeoutMS = 5000;
  26
  27 class FileLockTest : public testing::Test, public Runnable {
  28  public:
  29   FileLockTest() : done_(false, false), thread_lock_failed_(false) {
  30   }
  31
  32   virtual void Run(Thread* t) {
  33     scoped_ptr<FileLock> lock(FileLock::TryLock(temp_file_.pathname()));
  34     // The lock is already owned by the main thread of
  35     // this test, therefore the TryLock(...) call should fail.
  36     thread_lock_failed_ = lock.get() == NULL;
  37     done_.Set();
  38   }
  39
  40  protected:
  41   virtual void SetUp() {
  42     thread_lock_failed_ = false;
  43     Filesystem::GetAppTempFolder(&temp_dir_);
  44     temp_file_ = Pathname(temp_dir_.pathname(), kLockFile);
  45   }
  46
  47   void LockOnThread() {
  48     locker_.Start(this);
  49     done_.Wait(kTimeoutMS);
  50   }
  51
  52   Event done_;
  53   Thread locker_;
  54   bool thread_lock_failed_;
  55   Pathname temp_dir_;
  56   Pathname temp_file_;
  57 };
  58
  59 TEST_F(FileLockTest, TestLockFileDeleted) {
  60   scoped_ptr<FileLock> lock(FileLock::TryLock(temp_file_.pathname()));
  61   EXPECT_TRUE(lock.get() != NULL);
  62   EXPECT_FALSE(Filesystem::IsAbsent(temp_file_.pathname()));
  63   lock->Unlock();
  64   EXPECT_TRUE(Filesystem::IsAbsent(temp_file_.pathname()));
  65 }
  66
  67 TEST_F(FileLockTest, TestLock) {
  68   scoped_ptr<FileLock> lock(FileLock::TryLock(temp_file_.pathname()));
  69   EXPECT_TRUE(lock.get() != NULL);
  70 }
  71
  72 TEST_F(FileLockTest, TestLockX2) {
  73   scoped_ptr<FileLock> lock1(FileLock::TryLock(temp_file_.pathname()));
  74   EXPECT_TRUE(lock1.get() != NULL);
  75
  76   scoped_ptr<FileLock> lock2(FileLock::TryLock(temp_file_.pathname()));
  77   EXPECT_TRUE(lock2.get() == NULL);
  78 }
  79
  80 TEST_F(FileLockTest, TestThreadedLock) {
  81   scoped_ptr<FileLock> lock(FileLock::TryLock(temp_file_.pathname()));
  82   EXPECT_TRUE(lock.get() != NULL);
  83
  84   LockOnThread();
  85   EXPECT_TRUE(thread_lock_failed_);
  86 }
  87
  88 }  // namespace rtc