Added object set

(bzr r14954.1.1)

1 files changed, 56 insertions, 0 deletions

diff --git a/test/src/object-test.cpp b/test/src/object-test.cpp
new file mode 100644
index 000000000..83e90b2ec
--- /dev/null
+++ b/test/src/object-test.cpp
@@ -0,0 +1,56 @@
+#include "gtest/gtest.h"
+#include "object.h"
+
+class ObjectTest: public testing::Test {
+public:
+    ObjectTest() {
+        object = new Object("parent");
+        childOfChild = new Object("childOfChild");
+        child = new Object("child");
+        child2 = new Object("child2");
+        object2 = new Object("object2");
+        child3 = new Object("child3");
+    }
+    ~ObjectTest() {
+        delete child3;
+        delete child2;
+        delete childOfChild;
+        delete child;
+        delete object2;
+        delete object;
+    }
+    Object* object;
+    Object* object2;
+    Object* child;
+    Object* childOfChild;
+    Object* child2;
+    Object* child3;
+};
+
+TEST_F(ObjectTest, AddChild) {
+    EXPECT_EQ(0, object->getChildren().size());
+    object->addChild(child);
+    EXPECT_EQ(1, object->getChildren().size());
+    EXPECT_EQ(child, &(object->getChildren().front()));
+    EXPECT_EQ(object, child->getParent());
+}
+
+TEST_F(ObjectTest, IsDescendantOf) {
+    object->addChild(child);
+    child->addChild(childOfChild);
+    object->addChild(child2);
+    object2->addChild(child3);
+    EXPECT_TRUE(child->isDescendantOf(object));
+    EXPECT_TRUE(childOfChild->isDescendantOf(child));
+    EXPECT_TRUE(childOfChild->isDescendantOf(object));
+    EXPECT_TRUE(child2->isDescendantOf(object));
+    EXPECT_TRUE(child3->isDescendantOf(object2));
+    EXPECT_FALSE(child->isDescendantOf(child2));
+    EXPECT_FALSE(child2->isDescendantOf(child));
+    EXPECT_FALSE(object->isDescendantOf(childOfChild));
+    EXPECT_FALSE(object->isDescendantOf(child));
+    EXPECT_FALSE(object->isDescendantOf(object2));
+    EXPECT_FALSE(object2->isDescendantOf(object));
+    EXPECT_FALSE(child2->isDescendantOf(child3));
+    EXPECT_FALSE(child3->isDescendantOf(child2));
+}