Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright 2015 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/process/private_working_set_snapshot.h" | |
| 6 | |
| 7 #include "base/win/windows_version.h" | |
| 8 #include "testing/gtest/include/gtest/gtest.h" | |
| 9 | |
| 10 using PrivateWorkingSetSnapshotWinTest = testing::Test; | |
| 11 | |
| 12 TEST_F(PrivateWorkingSetSnapshotWinTest, FindPidSelfTest) { | |
| 13 // The Pdh APIs are supported on Windows XP, but the "Working Set - Private" | |
| 14 // counter that PrivateWorkingSetSnapshot depends on is not defined until | |
| 15 // Windows Vista. Early-out to avoid test failure. | |
| 16 if (base::win::GetVersion() < base::win::VERSION_VISTA) | |
| 17 return; | |
| 18 | |
| 19 // Sample this process. | |
| 20 base::ProcessId pid = base::GetCurrentProcId(); | |
| 21 | |
| 22 base::PrivateWorkingSetSnapshot private_ws_snapshot; | |
| 23 | |
| 24 private_ws_snapshot.AddToMonitorList("base_unittests"); | |
| 25 private_ws_snapshot.Sample(); | |
| 26 | |
| 27 size_t private_ws = private_ws_snapshot.GetPrivateWorkingSet(pid); | |
| 28 // Private working set is difficult to predict but should be at least several | |
| 29 // MB. Initial tests show a value of ~4 MB to 25 MB depending on how many | |
| 30 // tests and processes are used. Anomalously small or large values would | |
| 31 // warrant investigation. | |
| 32 EXPECT_GT(private_ws, 2000000u); | |
| 33 // Check that the WS is less than 500 MB. This is set very high to reduce the | |
| 34 // chance that unrelated changes could ever make this fail. This mostly just | |
| 35 // checks against some uncaught error that might return 0xFFFFFFFF. | |
| 36 EXPECT_LT(private_ws, 500000000u); | |
| 37 | |
| 38 // Allocate and touch a large block of memory (vector's constructor will zero | |
| 39 // every entry). This will increase the private working set. | |
| 40 const size_t alloc_size = 10000000; | |
| 41 std::vector<char> big_memory(alloc_size); | |
| 42 | |
| 43 size_t private_ws2 = private_ws_snapshot.GetPrivateWorkingSet(pid); | |
| 44 EXPECT_EQ(private_ws, private_ws2) << "GetPrivateWorkingSet should be " | |
| 45 "consistent until the next call to " | |
| 46 "Sample()"; | |
| 47 | |
| 48 private_ws_snapshot.Sample(); | |
| 49 size_t private_ws3 = private_ws_snapshot.GetPrivateWorkingSet(pid); | |
| 50 EXPECT_GT(private_ws3, private_ws2 + alloc_size / 2) | |
| 51 << "GetPrivateWorkingSet should increase as we allocate more memory"; | |
|
ncarter (slow)
2015/06/29 20:03:19
Fingers crossed on this test not being flaky.
brucedawson
2015/06/29 22:09:52
Done.
| |
| 52 } | |
| OLD | NEW |