7ba8228c46
Summary: - Create types folder in storage/common - Move locking and kvstore to storage/common - Add storage/distributed/rpc folder Reviewers: teon.banek, ipaljak, msantl Reviewed By: msantl Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1713
56 lines
1.6 KiB
C++
56 lines
1.6 KiB
C++
#include <gtest/gtest.h>
|
|
|
|
#include "storage/single_node/mvcc/record.hpp"
|
|
#include "storage/single_node/deferred_deleter.hpp"
|
|
#include "storage/single_node/vertex.hpp"
|
|
|
|
#include "mvcc_gc_common.hpp"
|
|
|
|
// Add and count objects.
|
|
TEST(DeferredDeleter, AddObjects) {
|
|
DeferredDeleter<Vertex> deleter;
|
|
for (int i = 0; i < 10; ++i) {
|
|
deleter.AddObject(new Vertex, 5);
|
|
deleter.AddObject(new Vertex, 5);
|
|
EXPECT_EQ(deleter.Count(), (i + 1) * 2);
|
|
}
|
|
deleter.FreeExpiredObjects(tx::Transaction::MaxId());
|
|
}
|
|
|
|
// Check that the deleter can't be destroyed while it still has objects.
|
|
TEST(DeferredDeleter, Destructor) {
|
|
std::atomic<int> count{0};
|
|
DeferredDeleter<DestrCountRec> *deleter = new DeferredDeleter<DestrCountRec>;
|
|
for (int i = 0; i < 10; ++i) {
|
|
deleter->AddObject(new DestrCountRec(count), 5);
|
|
deleter->AddObject(new DestrCountRec(count), 5);
|
|
EXPECT_EQ(deleter->Count(), (i + 1) * 2);
|
|
}
|
|
EXPECT_EQ(0, count);
|
|
EXPECT_DEATH(delete deleter, "");
|
|
// We shouldn't leak memory.
|
|
deleter->FreeExpiredObjects(tx::Transaction::MaxId());
|
|
delete deleter;
|
|
}
|
|
|
|
// Check if deleter frees objects.
|
|
TEST(DeferredDeleter, FreeExpiredObjects) {
|
|
DeferredDeleter<DestrCountRec> deleter;
|
|
std::atomic<int> count{0};
|
|
deleter.AddObject(new DestrCountRec(count), 5);
|
|
deleter.AddObject(new DestrCountRec(count), 5);
|
|
|
|
deleter.FreeExpiredObjects(5);
|
|
EXPECT_EQ(deleter.Count(), 2);
|
|
EXPECT_EQ(count, 0);
|
|
|
|
deleter.FreeExpiredObjects(6);
|
|
EXPECT_EQ(deleter.Count(), 0);
|
|
EXPECT_EQ(count, 2);
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|