MDL-64820 forum: fix up issues from review
[moodle.git] / mod / forum / tests / entities_sorter_test.php
1 <?php
2 // This file is part of Moodle - http://moodle.org/
3 //
4 // Moodle is free software: you can redistribute it and/or modify
5 // it under the terms of the GNU General Public License as published by
6 // the Free Software Foundation, either version 3 of the License, or
7 // (at your option) any later version.
8 //
9 // Moodle is distributed in the hope that it will be useful,
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 // GNU General Public License for more details.
13 //
14 // You should have received a copy of the GNU General Public License
15 // along with Moodle.  If not, see <http://www.gnu.org/licenses/>.
17 /**
18  * The discussion entity tests.
19  *
20  * @package    mod_forum
21  * @copyright  2019 Ryan Wyllie <ryan@moodle.com>
22  * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
23  */
25 defined('MOODLE_INTERNAL') || die();
27 use mod_forum\local\entities\sorter as sorter_entity;
29 /**
30  * The discussion entity tests.
31  *
32  * @package    mod_forum
33  * @copyright  2019 Ryan Wyllie <ryan@moodle.com>
34  * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
35  */
36 class mod_forum_entities_sorter_testcase extends advanced_testcase {
37     /**
38      * Test the entity returns expected values.
39      */
40     public function test_entity_sort_into_children() {
41         $this->resetAfterTest();
42         $sorter = new sorter_entity(
43             function($entity) {
44                 return $entity['id'];
45             },
46             function($entity) {
47                 return $entity['parent'];
48             }
49         );
51         $a = ['id' => 1, 'parent' => 0];
52         $b = ['id' => 2, 'parent' => 1];
53         $c = ['id' => 3, 'parent' => 1];
54         $d = ['id' => 4, 'parent' => 2];
55         $e = ['id' => 5, 'parent' => 0];
57         $expected = [
58             [$e, []],
59             [$a, [[$b, [[$d, []]]], [$c, []]]],
60         ];
62         $actual = $sorter->sort_into_children([$d, $b, $e, $a, $c]);
64         $this->assertEquals($expected, $actual);
65     }
67     /**
68      * Test the entity returns expected values.
69      */
70     public function test_entity_flatten_children() {
71         $this->resetAfterTest();
72         $sorter = new sorter_entity(
73             function($entity) {
74                 return $entity['id'];
75             },
76             function($entity) {
77                 return $entity['parent'];
78             }
79         );
81         $a = ['id' => 1, 'parent' => 0];
82         $b = ['id' => 2, 'parent' => 1];
83         $c = ['id' => 3, 'parent' => 1];
84         $d = ['id' => 4, 'parent' => 3];
86         $sorted = [
87             [$a, [[$b, [[$d, []]]], [$c, []]]]
88         ];
90         $expected = [$a, $b, $d, $c];
91         $actual = $sorter->flatten_children($sorted);
93         $this->assertEquals($expected, $actual);
94     }
95 }