1 files changed, 79 insertions, 0 deletions
diff --git a/staging_vespalib/src/tests/util/process_memory_stats/process_memory_stats_test.cpp b/staging_vespalib/src/tests/util/process_memory_stats/process_memory_stats_test.cpp
new file mode 100644
index 00000000000..43288d630cf
--- /dev/null
+++ b/staging_vespalib/src/tests/util/process_memory_stats/process_memory_stats_test.cpp
@@ -0,0 +1,79 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+
+#include <vespa/fastos/fastos.h>
+#include <vespa/vespalib/testkit/test_kit.h>
+#include <vespa/vespalib/util/process_memory_stats.h>
+#include <iostream>
+#include <fstream>
+
+
+using namespace vespalib;
+
+namespace {
+
+std::string toString(const ProcessMemoryStats &stats)
+{
+    std::ostringstream os;
+    os << "Mapped("
+       << stats.getMappedVirt() << "," << stats.getMappedRss() <<
+        "), Anonymous("
+       << stats.getAnonymousVirt() << "," << stats.getAnonymousRss() << ")";
+    return os.str();
+}
+
+}
+
+TEST("Simple stats")
+{
+    ProcessMemoryStats stats(ProcessMemoryStats::create());
+    std::cout << toString(stats) << std::endl;
+    EXPECT_LESS(0u, stats.getMappedVirt());
+    EXPECT_LESS(0u, stats.getMappedRss());
+    EXPECT_LESS(0u, stats.getAnonymousVirt());
+    EXPECT_LESS(0u, stats.getAnonymousRss());
+}
+
+TEST("grow anonymous memory")
+{
+    ProcessMemoryStats stats1(ProcessMemoryStats::create());
+    std::cout << toString(stats1) << std::endl;
+    size_t mapLen = 64 * 1024;
+    void *mapAddr = mmap(nullptr, mapLen, PROT_READ | PROT_WRITE,
+                         MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
+    EXPECT_NOT_EQUAL(reinterpret_cast<void *>(-1), mapAddr);
+    ProcessMemoryStats stats2(ProcessMemoryStats::create());
+    std::cout << toString(stats2) << std::endl;
+    EXPECT_LESS_EQUAL(stats1.getAnonymousVirt() + mapLen,
+                      stats2.getAnonymousVirt());
+    memset(mapAddr, 1, mapLen);
+    ProcessMemoryStats stats3(ProcessMemoryStats::create());
+    std::cout << toString(stats3) << std::endl;
+    // Cannot check that resident grows if swap is enabled and system loaded
+    munmap(mapAddr, mapLen);
+}
+
+TEST("grow mapped memory")
+{
+    std::ofstream of("mapfile");
+    size_t mapLen = 64 * 1024;
+    std::vector<char> buf(mapLen, 4);
+    of.write(&buf[0], buf.size());
+    of.close();
+    int mapfileFileDescriptor = open("mapfile", O_RDONLY, 0666);
+    EXPECT_LESS_EQUAL(0, mapfileFileDescriptor);
+    ProcessMemoryStats stats1(ProcessMemoryStats::create());
+    std::cout << toString(stats1) << std::endl;
+    void *mapAddr = mmap(nullptr, mapLen, PROT_READ, MAP_SHARED,
+                         mapfileFileDescriptor, 0);
+    EXPECT_NOT_EQUAL(reinterpret_cast<void *>(-1), mapAddr);
+    ProcessMemoryStats stats2(ProcessMemoryStats::create());
+    std::cout << toString(stats2) << std::endl;
+    EXPECT_LESS_EQUAL(stats1.getMappedVirt() + mapLen, stats2.getMappedVirt());
+    EXPECT_EQUAL(0, memcmp(mapAddr, &buf[0], mapLen));
+    ProcessMemoryStats stats3(ProcessMemoryStats::create());
+    std::cout << toString(stats3) << std::endl;
+    // Cannot check that resident grows if swap is enabled and system loaded
+    munmap(mapAddr, mapLen);
+}
+
+TEST_MAIN() { TEST_RUN_ALL(); }