242 lines
9.1 KiB
C++
242 lines
9.1 KiB
C++
/*
|
|
* Copyright (C) 2017 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 specic language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#include <android-base/file.h>
|
|
#include <android-base/stringprintf.h>
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <algorithm>
|
|
#include <thread>
|
|
|
|
#include "perfmgr/FileNode.h"
|
|
|
|
namespace android {
|
|
namespace perfmgr {
|
|
|
|
using std::literals::chrono_literals::operator""ms;
|
|
|
|
constexpr double kTIMING_TOLERANCE_MS = std::chrono::milliseconds(25).count();
|
|
constexpr auto kSLEEP_TOLERANCE_MS = 2ms;
|
|
|
|
static inline void _VerifyPathValue(const std::string& path,
|
|
const std::string& value) {
|
|
std::string s;
|
|
EXPECT_TRUE(android::base::ReadFileToString(path, &s)) << strerror(errno);
|
|
EXPECT_EQ(value, s);
|
|
}
|
|
|
|
// Test init with no default value
|
|
TEST(FileNodeTest, NoInitDefaultTest) {
|
|
TemporaryFile tf;
|
|
FileNode t("t", tf.path, {{"value0"}, {"value1"}, {"value2"}}, 1, false, false);
|
|
t.Update(false);
|
|
_VerifyPathValue(tf.path, "");
|
|
}
|
|
|
|
// Test init with default value
|
|
TEST(FileNodeTest, InitDefaultTest) {
|
|
TemporaryFile tf;
|
|
FileNode t("t", tf.path, {{"value0"}, {"value1"}, {"value2"}}, 1, true, true);
|
|
t.Update(false);
|
|
_VerifyPathValue(tf.path, "value1");
|
|
TemporaryFile tf2;
|
|
FileNode t2("t2", tf2.path, {{"value0"}, {"value1"}, {"value2"}}, 0, true, true);
|
|
t2.Update(false);
|
|
_VerifyPathValue(tf2.path, "value0");
|
|
}
|
|
|
|
// Test DumpToFd
|
|
TEST(FileNodeTest, DumpToFdTest) {
|
|
TemporaryFile tf;
|
|
FileNode t("test_dump", tf.path, {{"value0"}, {"value1"}, {"value2"}}, 1,
|
|
true, true);
|
|
t.Update(false);
|
|
TemporaryFile dumptf;
|
|
t.DumpToFd(dumptf.fd);
|
|
fsync(dumptf.fd);
|
|
std::string buf(android::base::StringPrintf(
|
|
"Node Name\t"
|
|
"Node Path\t"
|
|
"Current Index\t"
|
|
"Current Value\t"
|
|
"Hold FD\t"
|
|
"Truncate\n"
|
|
"%s\t%s\t%zu\t%s\t%d\t%d\n",
|
|
"test_dump", tf.path, static_cast<size_t>(1), "value1", 0, 1));
|
|
_VerifyPathValue(dumptf.path, buf);
|
|
}
|
|
|
|
// Test GetValueIndex
|
|
TEST(FileNodeTest, GetValueIndexTest) {
|
|
TemporaryFile tf;
|
|
FileNode t("t", tf.path, {{"value0"}, {"value1"}, {"value2"}}, 1, false, false);
|
|
std::size_t index = 0;
|
|
EXPECT_TRUE(t.GetValueIndex("value2", &index));
|
|
EXPECT_EQ(2u, index);
|
|
index = 1234;
|
|
EXPECT_FALSE(t.GetValueIndex("NON_EXIST", &index));
|
|
EXPECT_EQ(1234u, index);
|
|
}
|
|
|
|
// Test GetValues
|
|
TEST(FileNodeTest, GetValuesTest) {
|
|
TemporaryFile tf;
|
|
FileNode t("t", tf.path, {{"value0"}, {"value1"}, {"value2"}}, 1, false, false);
|
|
std::vector values = t.GetValues();
|
|
EXPECT_EQ(3u, values.size());
|
|
EXPECT_EQ("value0", values[0]);
|
|
EXPECT_EQ("value1", values[1]);
|
|
EXPECT_EQ("value2", values[2]);
|
|
}
|
|
|
|
// Test get more properties
|
|
TEST(FileNodeTest, GetPropertiesTest) {
|
|
std::string test_name = "TESTREQ_1";
|
|
std::string test_path = "TEST_PATH";
|
|
FileNode t(test_name, test_path, {}, 0, false, false, true);
|
|
EXPECT_EQ(test_name, t.GetName());
|
|
EXPECT_EQ(test_path, t.GetPath());
|
|
EXPECT_EQ(0u, t.GetValues().size());
|
|
EXPECT_EQ(0u, t.GetDefaultIndex());
|
|
EXPECT_FALSE(t.GetResetOnInit());
|
|
EXPECT_TRUE(t.GetHoldFd());
|
|
}
|
|
|
|
// Test add request fail and retry
|
|
TEST(FileNodeTest, AddRequestTestFail) {
|
|
FileNode t("t", "/sys/android/nonexist_node_test",
|
|
{{"value0"}, {"value1"}, {"value2"}}, 2, true, true);
|
|
auto start = std::chrono::steady_clock::now();
|
|
EXPECT_TRUE(t.AddRequest(1, "INTERACTION", start + 200ms));
|
|
std::chrono::milliseconds expire_time = t.Update(true);
|
|
// Add request @ value1
|
|
EXPECT_NEAR(std::chrono::milliseconds(200).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Add request @ value0 higher prio than value1
|
|
EXPECT_TRUE(t.AddRequest(0, "LAUNCH", start + 2000ms));
|
|
expire_time = t.Update(true);
|
|
// Retry in 500 ms
|
|
EXPECT_NEAR(std::chrono::milliseconds(500).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
}
|
|
|
|
// Test add request
|
|
TEST(FileNodeTest, AddRequestTest) {
|
|
TemporaryFile tf;
|
|
FileNode t("t", tf.path, {{"value0"}, {"value1"}, {"value2"}}, 2, true, true);
|
|
auto start = std::chrono::steady_clock::now();
|
|
EXPECT_TRUE(t.AddRequest(1, "INTERACTION", start + 500ms));
|
|
std::chrono::milliseconds expire_time = t.Update(true);
|
|
// Add request @ value1
|
|
_VerifyPathValue(tf.path, "value1");
|
|
EXPECT_NEAR(std::chrono::milliseconds(500).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Add request @ value0 higher prio than value1
|
|
EXPECT_TRUE(t.AddRequest(0, "LAUNCH", start + 200ms));
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value0");
|
|
EXPECT_NEAR(std::chrono::milliseconds(200).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Let high prio request timeout, now only request @ value1 active
|
|
std::this_thread::sleep_for(expire_time + kSLEEP_TOLERANCE_MS);
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value1");
|
|
EXPECT_NEAR(std::chrono::milliseconds(300).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Let all requests timeout, now default value2
|
|
std::this_thread::sleep_for(expire_time + kSLEEP_TOLERANCE_MS);
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value2");
|
|
EXPECT_EQ(std::chrono::milliseconds::max(), expire_time);
|
|
}
|
|
|
|
// Test remove request
|
|
TEST(FileNodeTest, RemoveRequestTest) {
|
|
TemporaryFile tf;
|
|
FileNode t("t", tf.path, {{"value0"}, {"value1"}, {"value2"}}, 2, true, true);
|
|
auto start = std::chrono::steady_clock::now();
|
|
EXPECT_TRUE(t.AddRequest(1, "INTERACTION", start + 500ms));
|
|
std::chrono::milliseconds expire_time = t.Update(true);
|
|
// Add request @ value1
|
|
_VerifyPathValue(tf.path, "value1");
|
|
EXPECT_NEAR(std::chrono::milliseconds(500).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Add request @ value0 higher prio than value1
|
|
EXPECT_TRUE(t.AddRequest(0, "LAUNCH", start + 200ms));
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value0");
|
|
EXPECT_NEAR(std::chrono::milliseconds(200).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Remove high prio request, now only request @ value1 active
|
|
t.RemoveRequest("LAUNCH");
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value1");
|
|
EXPECT_NEAR(std::chrono::milliseconds(500).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Remove request, now default value2
|
|
t.RemoveRequest("INTERACTION");
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value2");
|
|
EXPECT_EQ(std::chrono::milliseconds::max(), expire_time);
|
|
}
|
|
|
|
// Test add request with holding fd
|
|
TEST(FileNodeTest, AddRequestTestHoldFdOverride) {
|
|
TemporaryFile tf;
|
|
FileNode t("t", tf.path, {{"value0"}, {"value1"}, {"value2"}}, 2, true,
|
|
true, true);
|
|
EXPECT_TRUE(t.GetHoldFd());
|
|
auto start = std::chrono::steady_clock::now();
|
|
EXPECT_TRUE(t.AddRequest(1, "INTERACTION", start + 500ms));
|
|
std::chrono::milliseconds expire_time = t.Update(true);
|
|
// Add request @ value1
|
|
_VerifyPathValue(tf.path, "value1");
|
|
EXPECT_NEAR(std::chrono::milliseconds(500).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Add request @ value0 higher prio than value1
|
|
EXPECT_TRUE(t.AddRequest(0, "LAUNCH", start + 200ms));
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value0");
|
|
EXPECT_NEAR(std::chrono::milliseconds(200).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Add request @ value0 shorter
|
|
EXPECT_TRUE(t.AddRequest(0, "LAUNCH", start + 100ms));
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value0");
|
|
EXPECT_NEAR(std::chrono::milliseconds(200).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Add request @ value0 longer
|
|
EXPECT_TRUE(t.AddRequest(0, "LAUNCH", start + 300ms));
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value0");
|
|
EXPECT_NEAR(std::chrono::milliseconds(300).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Remove high prio request, now only request @ value1 active
|
|
t.RemoveRequest("LAUNCH");
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value1");
|
|
EXPECT_NEAR(std::chrono::milliseconds(500).count(), expire_time.count(),
|
|
kTIMING_TOLERANCE_MS);
|
|
// Remove request, now default value2
|
|
t.RemoveRequest("INTERACTION");
|
|
expire_time = t.Update(true);
|
|
_VerifyPathValue(tf.path, "value2");
|
|
EXPECT_EQ(std::chrono::milliseconds::max(), expire_time);
|
|
}
|
|
|
|
} // namespace perfmgr
|
|
} // namespace android
|