Home | History | Annotate | Download | only in base
      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, DISABLED_ON_MAC(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
     89