1 files changed, 119 insertions, 0 deletions
diff --git a/tests/tree/nestedset_forum_get_data_test.php b/tests/tree/nestedset_forum_get_data_test.php
new file mode 100644
index 0000000000..ca1863e55e
--- /dev/null
+++ b/tests/tree/nestedset_forum_get_data_test.php
@@ -0,0 +1,119 @@
+<?php
+/**
+*
+* @package tree
+* @copyright (c) 2013 phpBB Group
+* @license http://opensource.org/licenses/gpl-2.0.php GNU General Public License v2
+*
+*/
+
+require_once dirname(__FILE__) . '/nestedset_forum_base.php';
+
+class phpbb_tests_tree_nestedset_forum_get_data_test extends phpbb_tests_tree_nestedset_forum_base
+{
+	public function get_path_and_subtree_data_data()
+	{
+		return array(
+			array(1, true, true, array(1, 2, 3)),
+			array(1, true, false, array(2, 3)),
+			array(1, false, true, array(3, 2, 1)),
+			array(1, false, false, array(3, 2)),
+
+			array(2, true, true, array(1, 2)),
+			array(2, true, false, array(1)),
+			array(2, false, true, array(2, 1)),
+			array(2, false, false, array(1)),
+
+			array(5, true, true, array(4, 5, 6)),
+			array(5, true, false, array(4, 6)),
+			array(5, false, true, array(6, 5, 4)),
+			array(5, false, false, array(6, 4)),
+		);
+	}
+
+	/**
+	* @dataProvider get_path_and_subtree_data_data
+	*/
+	public function test_get_path_and_subtree_data($forum_id, $order_asc, $include_item, $expected)
+	{
+		$this->assertEquals($expected, array_keys($this->set->get_path_and_subtree_data($forum_id, $order_asc, $include_item)));
+	}
+
+	public function get_path_data_data()
+	{
+		return array(
+			array(1, true, true, array(1)),
+			array(1, true, false, array()),
+			array(1, false, true, array(1)),
+			array(1, false, false, array()),
+
+			array(2, true, true, array(1, 2)),
+			array(2, true, false, array(1)),
+			array(2, false, true, array(2, 1)),
+			array(2, false, false, array(1)),
+
+			array(5, true, true, array(4, 5)),
+			array(5, true, false, array(4)),
+			array(5, false, true, array(5, 4)),
+			array(5, false, false, array(4)),
+		);
+	}
+
+	/**
+	* @dataProvider get_path_data_data
+	*/
+	public function test_get_path_data($forum_id, $order_asc, $include_item, $expected)
+	{
+		$this->assertEquals($expected, array_keys($this->set->get_path_data($forum_id, $order_asc, $include_item)));
+	}
+
+	public function get_subtree_data_data()
+	{
+		return array(
+			array(1, true, true, array(1, 2, 3)),
+			array(1, true, false, array(2, 3)),
+			array(1, false, true, array(3, 2, 1)),
+			array(1, false, false, array(3, 2)),
+
+			array(2, true, true, array(2)),
+			array(2, true, false, array()),
+			array(2, false, true, array(2)),
+			array(2, false, false, array()),
+
+			array(5, true, true, array(5, 6)),
+			array(5, true, false, array(6)),
+			array(5, false, true, array(6, 5)),
+			array(5, false, false, array(6)),
+		);
+	}
+
+	/**
+	* @dataProvider get_subtree_data_data
+	*/
+	public function test_get_subtree_data($forum_id, $order_asc, $include_item, $expected)
+	{
+		$this->assertEquals($expected, array_keys($this->set->get_subtree_data($forum_id, $order_asc, $include_item)));
+	}
+
+	public function get_path_basic_data_data()
+	{
+		return array(
+			array(1, '', array()),
+			array(1, serialize(array()), array()),
+			array(2, '', array(1)),
+			array(2, serialize(array(1 => array())), array(1)),
+			array(10, '', array(7, 9)),
+			array(10, serialize(array(7 => array(), 9 => array())), array(7, 9)),
+		);
+	}
+
+	/**
+	* @dataProvider get_path_basic_data_data
+	*/
+	public function test_get_path_basic_data($forum_id, $forum_parents, $expected)
+	{
+		$forum_data = $this->forum_data[$forum_id];
+		$forum_data['forum_parents'] = $forum_parents;
+		$this->assertEquals($expected, array_keys($this->set->get_path_basic_data($forum_data)));
+	}
+}