Add disableEvictionToMemory unit test.

Author: ss7pro

cachelib/allocator/tests/AllocatorMemoryTiersTest.cpp

@@ -23,12 +23,21 @@ namespace tests {
 using LruAllocatorMemoryTiersTest = AllocatorMemoryTiersTest<LruAllocator>;
 
 // TODO(MEMORY_TIER): add more tests with different eviction policies
-TEST_F(LruAllocatorMemoryTiersTest, MultiTiersInvalid) { this->testMultiTiersInvalid(); }
-TEST_F(LruAllocatorMemoryTiersTest, MultiTiersValid) { this->testMultiTiersValid(); }
-TEST_F(LruAllocatorMemoryTiersTest, MultiTiersValidMixed) { this->testMultiTiersValidMixed(); }
-TEST_F(LruAllocatorMemoryTiersTest, MultiTiersForceTierAllocation) { this->testMultiTiersForceTierAllocation(); }
-TEST_F(LruAllocatorMemoryTiersTest, MultiTiersWatermarkTierAllocation) { this->testMultiTiersWatermarkAllocation(); }
-TEST_F(LruAllocatorMemoryTiersTest, MultiTiersSyncEviction) { this->testSyncEviction(); }
+TEST_F(LruAllocatorMemoryTiersTest, MultiTiersInvalid) {
+    this->testMultiTiersInvalid(); }
+TEST_F(LruAllocatorMemoryTiersTest, MultiTiersValid) {
+    this->testMultiTiersValid(); }
+TEST_F(LruAllocatorMemoryTiersTest, MultiTiersValidMixed) {
+    this->testMultiTiersValidMixed(); }
+TEST_F(LruAllocatorMemoryTiersTest, MultiTiersForceTierAllocation) {
+    this->testMultiTiersForceTierAllocation(); }
+TEST_F(LruAllocatorMemoryTiersTest, MultiTiersWatermarkTierAllocation) {
+    this->testMultiTiersWatermarkAllocation(); }
+TEST_F(LruAllocatorMemoryTiersTest, MultiTiersSyncEviction) {
+    this->testSyncEviction(); }
+TEST_F(LruAllocatorMemoryTiersTest, MultiTiersDisableEvictionToMemory) {
+    this->testMultiTiersDisableEvictionToMemory(); }
+
 
 } // end of namespace tests
 } // end of namespace cachelib

cachelib/allocator/tests/AllocatorMemoryTiersTest.h

@@ -221,6 +221,69 @@ class AllocatorMemoryTiersTest : public AllocatorTest<AllocatorT> {
         alloc.getCacheMemoryStats().slabsApproxFreePercentages[1]);
     }
   }
+
+  void testMultiTiersDisableEvictionToMemory () {
+    auto config = makeDefaultConfig();
+    config.setCacheSize(4 * Slab::kSize);
+    //Always allocate to fist tier.
+    config.forceAllocationTier = 0;
+    //Don't evict to memory.
+    config.disableEvictionToMemory = true;
+    {
+      AllocatorT alloc(AllocatorT::SharedMemNew, config);
+      auto pool = alloc.addPool("default",
+                                alloc.getCacheMemoryStats().cacheSize);
+      {
+        //Allocate first key.
+        auto handle = alloc.allocate(pool, "key1", Slab::kSize / 2);
+        ASSERT(handle != nullptr);
+        std::string data = "lorem ipsum the first";
+        std::memcpy(reinterpret_cast<char*>(handle->getMemory()), data.data(),
+                    data.size());
+        alloc.insertOrReplace(handle);
+        //Item in the first tier.
+        ASSERT_NE(alloc.getCacheMemoryStats().slabsApproxFreePercentages[0],
+                  100.0);
+        //Second tier is empty.
+        ASSERT_EQ(alloc.getCacheMemoryStats().slabsApproxFreePercentages[1],
+                  100.0);
+      }
+      {
+        //Allocate second key.
+        std::cout << alloc.getCacheMemoryStats().slabsApproxFreePercentages[0]
+          << std::endl;
+        std::cout << alloc.getCacheMemoryStats().slabsApproxFreePercentages[1]
+          << std::endl;
+        auto handle = alloc.allocate(pool, "key2", Slab::kSize / 2);
+        ASSERT(handle != nullptr);
+        std::string data2 = "lorem ipsum the second";
+        std::memcpy(reinterpret_cast<char*>(handle->getMemory()), data2.data(),
+                    data2.size());
+        alloc.insertOrReplace(handle);
+        //Item in the first tier.
+        ASSERT_NE(alloc.getCacheMemoryStats().slabsApproxFreePercentages[0],
+                  100.0);
+        //Second tier is empty.
+        ASSERT_EQ(alloc.getCacheMemoryStats().slabsApproxFreePercentages[1],
+                  100.0);
+        std::cout << alloc.getCacheMemoryStats().slabsApproxFreePercentages[0]
+          << std::endl;
+        std::cout << alloc.getCacheMemoryStats().slabsApproxFreePercentages[1]
+          << std::endl;
+      }
+      {
+        //Check if key1 was evicted.
+        auto found = alloc.find("key1");
+        ASSERT_EQ(found, nullptr);
+      }
+      {
+        //Check if key2 is present.
+        auto found = alloc.find("key2");
+        ASSERT_NE(found, nullptr);
+        ASSERT_EQ(found->getSize(), Slab::kSize / 2);
+      }
+    }
+  }
 };
 } // namespace tests
 } // namespace cachelib