1 // Copyright (c) 2011 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/file_util.h"
6 #include "base/files/scoped_temp_dir.h"
7 #include "base/path_service.h"
8 #include "base/strings/string_util.h"
9 #include "build/build_config.h"
10 #include "chrome/browser/importer/firefox_profile_lock.h"
11 #include "chrome/common/chrome_paths.h"
12 #include "testing/gtest/include/gtest/gtest.h"
13
14 class FirefoxProfileLockTest : public testing::Test {
15 protected:
SetUp()16 virtual void SetUp() {
17 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
18 }
19
20 base::ScopedTempDir temp_dir_;
21 };
22
TEST_F(FirefoxProfileLockTest,LockTest)23 TEST_F(FirefoxProfileLockTest, LockTest) {
24 FirefoxProfileLock lock1(temp_dir_.path());
25 ASSERT_TRUE(lock1.HasAcquired());
26 lock1.Unlock();
27 ASSERT_FALSE(lock1.HasAcquired());
28 lock1.Lock();
29 ASSERT_TRUE(lock1.HasAcquired());
30 }
31
32 // Tests basic functionality and verifies that the lock file is deleted after
33 // use.
TEST_F(FirefoxProfileLockTest,ProfileLock)34 TEST_F(FirefoxProfileLockTest, ProfileLock) {
35 base::FilePath test_path = temp_dir_.path();
36 base::FilePath lock_file_path =
37 test_path.Append(FirefoxProfileLock::kLockFileName);
38
39 scoped_ptr<FirefoxProfileLock> lock;
40 EXPECT_EQ(static_cast<FirefoxProfileLock*>(NULL), lock.get());
41 EXPECT_FALSE(base::PathExists(lock_file_path));
42 lock.reset(new FirefoxProfileLock(test_path));
43 EXPECT_TRUE(lock->HasAcquired());
44 EXPECT_TRUE(base::PathExists(lock_file_path));
45 lock->Unlock();
46 EXPECT_FALSE(lock->HasAcquired());
47
48 // In the posix code, we don't delete the file when releasing the lock.
49 #if !defined(OS_POSIX)
50 EXPECT_FALSE(base::PathExists(lock_file_path));
51 #endif // !defined(OS_POSIX)
52 lock->Lock();
53 EXPECT_TRUE(lock->HasAcquired());
54 EXPECT_TRUE(base::PathExists(lock_file_path));
55 lock->Lock();
56 EXPECT_TRUE(lock->HasAcquired());
57 lock->Unlock();
58 EXPECT_FALSE(lock->HasAcquired());
59 // In the posix code, we don't delete the file when releasing the lock.
60 #if !defined(OS_POSIX)
61 EXPECT_FALSE(base::PathExists(lock_file_path));
62 #endif // !defined(OS_POSIX)
63 }
64
65 // If for some reason the lock file is left behind by the previous owner, we
66 // should still be able to lock it, at least in the Windows implementation.
TEST_F(FirefoxProfileLockTest,ProfileLockOrphaned)67 TEST_F(FirefoxProfileLockTest, ProfileLockOrphaned) {
68 base::FilePath test_path = temp_dir_.path();
69 base::FilePath lock_file_path =
70 test_path.Append(FirefoxProfileLock::kLockFileName);
71
72 // Create the orphaned lock file.
73 FILE* lock_file = base::OpenFile(lock_file_path, "w");
74 ASSERT_TRUE(lock_file);
75 base::CloseFile(lock_file);
76 EXPECT_TRUE(base::PathExists(lock_file_path));
77
78 scoped_ptr<FirefoxProfileLock> lock;
79 EXPECT_EQ(static_cast<FirefoxProfileLock*>(NULL), lock.get());
80 lock.reset(new FirefoxProfileLock(test_path));
81 EXPECT_TRUE(lock->HasAcquired());
82 lock->Unlock();
83 EXPECT_FALSE(lock->HasAcquired());
84 }
85
86 // This is broken on POSIX since the same process is allowed to reacquire a
87 // lock.
88 #if !defined(OS_POSIX)
89 // Tests two locks contending for the same lock file.
TEST_F(FirefoxProfileLockTest,ProfileLockContention)90 TEST_F(FirefoxProfileLockTest, ProfileLockContention) {
91 base::FilePath test_path = temp_dir_.path();
92
93 scoped_ptr<FirefoxProfileLock> lock1;
94 EXPECT_EQ(static_cast<FirefoxProfileLock*>(NULL), lock1.get());
95 lock1.reset(new FirefoxProfileLock(test_path));
96 EXPECT_TRUE(lock1->HasAcquired());
97
98 scoped_ptr<FirefoxProfileLock> lock2;
99 EXPECT_EQ(static_cast<FirefoxProfileLock*>(NULL), lock2.get());
100 lock2.reset(new FirefoxProfileLock(test_path));
101 EXPECT_FALSE(lock2->HasAcquired());
102
103 lock1->Unlock();
104 EXPECT_FALSE(lock1->HasAcquired());
105
106 lock2->Lock();
107 EXPECT_TRUE(lock2->HasAcquired());
108 lock2->Unlock();
109 EXPECT_FALSE(lock2->HasAcquired());
110 }
111 #endif
112