142 lines
4.8 KiB
C++
142 lines
4.8 KiB
C++
/*
|
|
* Copyright (C) 2015 The Android Open Source Project
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <base/files/file_path.h>
|
|
#include <base/files/file_util.h>
|
|
#include <base/files/scoped_temp_dir.h>
|
|
#include <base/logging.h>
|
|
#include <base/macros.h>
|
|
#include <binder/IBinder.h>
|
|
#include <binderwrapper/binder_test_base.h>
|
|
#include <binderwrapper/stub_binder_wrapper.h>
|
|
|
|
#include "wake_lock_manager.h"
|
|
|
|
namespace android {
|
|
|
|
class WakeLockManagerTest : public BinderTestBase {
|
|
public:
|
|
WakeLockManagerTest() {
|
|
CHECK(temp_dir_.CreateUniqueTempDir());
|
|
lock_path_ = temp_dir_.path().Append("lock");
|
|
unlock_path_ = temp_dir_.path().Append("unlock");
|
|
ClearFiles();
|
|
|
|
manager_.set_paths_for_testing(lock_path_, unlock_path_);
|
|
CHECK(manager_.Init());
|
|
}
|
|
~WakeLockManagerTest() override = default;
|
|
|
|
protected:
|
|
// Returns the contents of |path|.
|
|
std::string ReadFile(const base::FilePath& path) const {
|
|
std::string value;
|
|
CHECK(base::ReadFileToString(path, &value));
|
|
return value;
|
|
}
|
|
|
|
// Clears |lock_path_| and |unlock_path_|.
|
|
void ClearFiles() {
|
|
CHECK(base::WriteFile(lock_path_, "", 0) == 0);
|
|
CHECK(base::WriteFile(unlock_path_, "", 0) == 0);
|
|
}
|
|
|
|
base::ScopedTempDir temp_dir_;
|
|
|
|
// Files within |temp_dir_| simulating /sys/power/wake_lock and wake_unlock.
|
|
base::FilePath lock_path_;
|
|
base::FilePath unlock_path_;
|
|
|
|
WakeLockManager manager_;
|
|
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN(WakeLockManagerTest);
|
|
};
|
|
|
|
TEST_F(WakeLockManagerTest, AddAndRemoveRequests) {
|
|
// A kernel wake lock should be created for the first request.
|
|
sp<BBinder> binder1 = binder_wrapper()->CreateLocalBinder();
|
|
EXPECT_TRUE(manager_.AddRequest(binder1, "1", "1", -1));
|
|
EXPECT_EQ(WakeLockManager::kLockName, ReadFile(lock_path_));
|
|
EXPECT_EQ("", ReadFile(unlock_path_));
|
|
|
|
// Nothing should happen when a second request is made.
|
|
ClearFiles();
|
|
sp<BBinder> binder2 = binder_wrapper()->CreateLocalBinder();
|
|
EXPECT_TRUE(manager_.AddRequest(binder2, "2", "2", -1));
|
|
EXPECT_EQ("", ReadFile(lock_path_));
|
|
EXPECT_EQ("", ReadFile(unlock_path_));
|
|
|
|
// The wake lock should still be held after the first request is withdrawn.
|
|
ClearFiles();
|
|
EXPECT_TRUE(manager_.RemoveRequest(binder1));
|
|
EXPECT_EQ("", ReadFile(lock_path_));
|
|
EXPECT_EQ("", ReadFile(unlock_path_));
|
|
|
|
// When there are no more requests, the wake lock should be released.
|
|
ClearFiles();
|
|
EXPECT_TRUE(manager_.RemoveRequest(binder2));
|
|
EXPECT_EQ("", ReadFile(lock_path_));
|
|
EXPECT_EQ(WakeLockManager::kLockName, ReadFile(unlock_path_));
|
|
}
|
|
|
|
TEST_F(WakeLockManagerTest, DuplicateRequest) {
|
|
sp<BBinder> binder = binder_wrapper()->CreateLocalBinder();
|
|
EXPECT_TRUE(manager_.AddRequest(binder, "foo", "bar", -1));
|
|
EXPECT_EQ(WakeLockManager::kLockName, ReadFile(lock_path_));
|
|
EXPECT_EQ("", ReadFile(unlock_path_));
|
|
|
|
// Send a second request using the same binder and check a new wake lock isn't
|
|
// created.
|
|
ClearFiles();
|
|
EXPECT_TRUE(manager_.AddRequest(binder, "a", "b", -1));
|
|
EXPECT_EQ("", ReadFile(lock_path_));
|
|
EXPECT_EQ("", ReadFile(unlock_path_));
|
|
|
|
ClearFiles();
|
|
EXPECT_TRUE(manager_.RemoveRequest(binder));
|
|
EXPECT_EQ("", ReadFile(lock_path_));
|
|
EXPECT_EQ(WakeLockManager::kLockName, ReadFile(unlock_path_));
|
|
}
|
|
|
|
TEST_F(WakeLockManagerTest, InvalidRemoval) {
|
|
// Trying to remove an unknown binder should fail and not do anything.
|
|
sp<BBinder> binder = binder_wrapper()->CreateLocalBinder();
|
|
EXPECT_FALSE(manager_.RemoveRequest(binder));
|
|
EXPECT_EQ("", ReadFile(lock_path_));
|
|
EXPECT_EQ("", ReadFile(unlock_path_));
|
|
}
|
|
|
|
TEST_F(WakeLockManagerTest, BinderDeath) {
|
|
sp<BBinder> binder = binder_wrapper()->CreateLocalBinder();
|
|
EXPECT_TRUE(manager_.AddRequest(binder, "foo", "bar", -1));
|
|
EXPECT_EQ(WakeLockManager::kLockName, ReadFile(lock_path_));
|
|
EXPECT_EQ("", ReadFile(unlock_path_));
|
|
|
|
// If the binder dies, the wake lock should be released.
|
|
ClearFiles();
|
|
binder_wrapper()->NotifyAboutBinderDeath(binder);
|
|
EXPECT_EQ("", ReadFile(lock_path_));
|
|
EXPECT_EQ(WakeLockManager::kLockName, ReadFile(unlock_path_));
|
|
|
|
// Check that a new request can be created using the same binder.
|
|
ClearFiles();
|
|
EXPECT_TRUE(manager_.AddRequest(binder, "foo", "bar", -1));
|
|
EXPECT_EQ(WakeLockManager::kLockName, ReadFile(lock_path_));
|
|
EXPECT_EQ("", ReadFile(unlock_path_));
|
|
}
|
|
|
|
} // namespace android
|