57 lines
1.5 KiB
C++
57 lines
1.5 KiB
C++
#include <stdlib.h>
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "storage/unique_object_store.hpp"
|
|
#include "utils/total_ordering.hpp"
|
|
|
|
/**
|
|
* Wraps an int and implements total ordering. Used for testing the
|
|
* UniqueObjectStore with custom classes.
|
|
*/
|
|
class IntWrapper : TotalOrdering<IntWrapper>, TotalOrdering<IntWrapper, int> {
|
|
public:
|
|
IntWrapper(const int i) : i_(i) {}
|
|
|
|
friend bool operator==(const IntWrapper &lhs, const IntWrapper &rhs) {
|
|
return lhs.i_ == rhs.i_;
|
|
}
|
|
|
|
friend bool operator<(const IntWrapper &lhs, const IntWrapper &rhs) {
|
|
return lhs.i_ < rhs.i_;
|
|
}
|
|
|
|
private:
|
|
const int i_;
|
|
};
|
|
|
|
TEST(UniqueObjectStoreTest, GetKey) {
|
|
UniqueObjectStore<std::string> store;
|
|
EXPECT_EQ(store.GetKey("name1"), store.GetKey("name1"));
|
|
EXPECT_NE(store.GetKey("name1"), store.GetKey("name2"));
|
|
EXPECT_EQ(store.GetKey("name2"), store.GetKey("name2"));
|
|
}
|
|
|
|
TEST(UniqueObjectStoreTest, CustomClass) {
|
|
UniqueObjectStore<IntWrapper> store;
|
|
EXPECT_EQ(store.GetKey(42), store.GetKey(IntWrapper(42)));
|
|
EXPECT_EQ(store.GetKey(1000), store.GetKey(IntWrapper(1000)));
|
|
EXPECT_NE(store.GetKey(IntWrapper(1000)), store.GetKey(IntWrapper(42)));
|
|
}
|
|
|
|
TEST(UniqueObjectStoreTest, RandomAccess) {
|
|
UniqueObjectStore<int> store;
|
|
for (int i = 0; i < 50 * 50 * 4; i++) {
|
|
int rand1 = rand() % 50;
|
|
int rand2 = rand() % 50;
|
|
EXPECT_EQ(store.GetKey(rand1) == store.GetKey(rand2), rand1 == rand2);
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|