Refactor SelfResolvable.

1 files changed, 82 insertions, 0 deletions

diff --git a/test/base/SelfResolvableTest.cpp b/test/base/SelfResolvableTest.cpp
new file mode 100644
index 00000000..17d9309d
--- /dev/null
+++ b/test/base/SelfResolvableTest.cpp
@@ -0,0 +1,82 @@
+#include "cru/base/SelfResolvable.h"
+
+#include <catch2/catch_test_macros.hpp>
+
+#include <memory>
+
+namespace {
+class SelfResolvableTestClassBase {
+ public:
+  virtual ~SelfResolvableTestClassBase() = default;
+};
+
+class SelfResolvableTestClass
+    : public SelfResolvableTestClassBase,
+      public cru::SelfResolvable<SelfResolvableTestClass> {
+ public:
+  SelfResolvableTestClass() : ptr_(new int(123)) {}
+  ~SelfResolvableTestClass() override = default;
+
+ private:
+  std::shared_ptr<int> ptr_;
+};
+}  // namespace
+
+TEST_CASE("SelfResolvable resolver should work.", "[self-resolvable]") {
+  SelfResolvableTestClass test_object;
+
+  auto resolver = test_object.CreateResolver();
+  REQUIRE(resolver.Resolve() == &test_object);
+
+  auto resolver_copy = resolver;
+  REQUIRE(resolver.Resolve() == &test_object);
+  REQUIRE(resolver_copy.Resolve() == &test_object);
+
+  auto resolver_move = std::move(resolver_copy);
+  REQUIRE(resolver.Resolve() == &test_object);
+  REQUIRE(resolver_copy.IsValid() == false);
+  REQUIRE(resolver_move.Resolve() == &test_object);
+}
+
+TEST_CASE("SelfResolvable object destructed should work.",
+          "[self-resolvable]") {
+  SelfResolvableTestClass* test_object = new SelfResolvableTestClass();
+
+  auto resolver = test_object->CreateResolver();
+  auto resolver_copy = resolver;
+
+  delete test_object;
+
+  REQUIRE(resolver.Resolve() == nullptr);
+  REQUIRE(resolver_copy.Resolve() == nullptr);
+
+  auto resolver_copy2 = resolver_copy;
+  REQUIRE(resolver_copy2.Resolve() == nullptr);
+
+  auto resolver_move = std::move(resolver_copy);
+  REQUIRE(resolver_copy.IsValid() == false);
+  REQUIRE(resolver_move.Resolve() == nullptr);
+}
+
+TEST_CASE("SelfResolvable should work for casted type.", "[self-resolvable]") {
+  auto test_object = new SelfResolvableTestClass();
+
+  cru::ObjectResolver<SelfResolvableTestClassBase> base_resolver =
+      test_object->CreateResolver();
+  auto resolver = test_object->CreateResolver();
+
+  REQUIRE(base_resolver.Resolve() == test_object);
+  REQUIRE(resolver.Resolve() == test_object);
+
+  auto base_resolver2 = base_resolver;
+  REQUIRE(base_resolver2.Resolve() == test_object);
+
+  auto base_resolver3 = std::move(base_resolver2);
+  REQUIRE(base_resolver3.Resolve() == test_object);
+
+  delete test_object;
+
+  REQUIRE(base_resolver.Resolve() == nullptr);
+  REQUIRE(base_resolver3.Resolve() == nullptr);
+  REQUIRE(resolver.Resolve() == nullptr);
+}