BUGFIX Use correct query when searching for items managed by a tree dropdown field #3173

forms/TreeDropdownField.php

@@ -460,15 +460,18 @@ class TreeDropdownField extends FormField {
 				if ($row->ParentID) $parents[$row->ParentID] = true;
 				$this->searchIds[$row->ID] = true;
 			}
+			
+			$sourceObject = $this->sourceObject;
+			
 			while (!empty($parents)) {
-				$res = DB::query('SELECT "ParentID", "ID" FROM "' . $this->sourceObject
+				$items = $sourceObject::get()
-					. '" WHERE "ID" in ('.implode(',',array_keys($parents)).')');
+					->filter("ID",array_keys($parents));
 				$parents = array();
 
-				foreach($res as $row) {
+				foreach($items as $item) {
-					if ($row['ParentID']) $parents[$row['ParentID']] = true;
+					if ($item->ParentID) $parents[$item->ParentID] = true;
-					$this->searchIds[$row['ID']] = true;
+					$this->searchIds[$item->ID] = true;
-					$this->searchExpanded[$row['ID']] = true;
+					$this->searchExpanded[$item->ID] = true;
 				}
 			}
 		}

tests/forms/TreeDropdownFieldTest.php

@@ -0,0 +1,97 @@
+<?php
+/**
+ * @package framework
+ * @subpackage tests
+ */
+class TreeDropdownFieldTest extends SapphireTest {
+
+	protected static $fixture_file = 'TreeDropdownFieldTest.yml';
+
+	public function testTreeSearch(){
+
+		$field = new TreeDropdownField('TestTree', 'Test tree', 'Folder');
+
+		// case insensitive search against keyword 'sub' for folders
+		$request = new SS_HTTPRequest('GET','url',array('search'=>'sub'));
+		$tree = $field->tree($request);
+		
+		$folder1 = $this->objFromFixture('Folder','folder1');
+		$folder1Subfolder1 = $this->objFromFixture('Folder','folder1-subfolder1');
+
+		$parser = new CSSContentParser($tree);
+		$cssPath = 'ul.tree li#selector-TestTree-'.$folder1->ID.' li#selector-TestTree-'.$folder1Subfolder1->ID.' a span.item';
+		$firstResult = $parser->getBySelector($cssPath);
+		$this->assertEquals(
+			(string)$firstResult[0], 
+			$folder1Subfolder1->Name, 
+			'FileTest-folder1-subfolder1 is found, nested under folder1'
+		);
+
+		$subfolder = $this->objFromFixture('Folder','subfolder');
+		$cssPath = 'ul.tree li#selector-TestTree-'.$subfolder->ID.' a span.item';
+		$secondResult = $parser->getBySelector($cssPath);
+		$this->assertEquals(
+			(string)$secondResult[0], 
+			$subfolder->Name, 
+			'FileTest-subfolder is found at root level'
+		);
+
+		// other folders which don't contain the keyword 'sub' are not returned in search results
+		$folder2 = $this->objFromFixture('Folder','folder2');
+		$cssPath = 'ul.tree li#selector-TestTree-'.$folder2->ID.' a span.item';
+		$noResult = $parser->getBySelector($cssPath);
+		$this->assertEquals(
+			$noResult, 
+			array(), 
+			'FileTest-folder2 is not found'
+		);
+
+		$field = new TreeDropdownField('TestTree', 'Test tree', 'File');
+
+		// case insensitive search against keyword 'sub' for files
+		$request = new SS_HTTPRequest('GET','url',array('search'=>'sub'));
+		$tree = $field->tree($request);
+
+		$parser = new CSSContentParser($tree);
+
+		// Even if we used File as the source object, folders are still returned because Folder is a File
+		$cssPath = 'ul.tree li#selector-TestTree-'.$folder1->ID.' li#selector-TestTree-'.$folder1Subfolder1->ID.' a span.item';
+		$firstResult = $parser->getBySelector($cssPath);
+		$this->assertEquals(
+			(string)$firstResult[0], 
+			$folder1Subfolder1->Name, 
+			'FileTest-folder1-subfolder1 is found, nested under folder1'
+		);
+
+		// Looking for two files with 'sub' in their name, both under the same folder
+		$file1 = $this->objFromFixture('File','subfolderfile1');
+		$file2 = $this->objFromFixture('File','subfolderfile2');
+		$cssPath = 'ul.tree li#selector-TestTree-'.$subfolder->ID.' li#selector-TestTree-'.$file1->ID.' a';
+		$firstResult = $parser->getBySelector($cssPath);
+		$this->assertEquals(
+			(string)$firstResult[0], 
+			$file1->Name, 
+			'TestFile1InSubfolder is found nested under subfolder'
+		);
+
+		$cssPath = 'ul.tree li#selector-TestTree-'.$subfolder->ID.' li#selector-TestTree-'.$file2->ID.' a';
+		$secondResult = $parser->getBySelector($cssPath);
+		$this->assertEquals(
+			(string)$secondResult[0], 
+			$file2->Name, 
+			'TestFile2InSubfolder is found nested under subfolder'
+		);
+
+		// other files which don't include 'sub' are not returned in search results
+		$file3 = $this->objFromFixture('File','asdf');
+		$cssPath = 'ul.tree li#selector-TestTree-'.$file3->ID;
+		$noResult = $parser->getBySelector($cssPath);
+		$this->assertEquals(
+			$noResult, 
+			array(), 
+			'FileTest.txt is not found'
+		);
+	}
+
+}
+	

tests/forms/TreeDropdownFieldTest.yml

@@ -0,0 +1,25 @@
+Folder:
+  subfolder:
+    Name: FileTest-subfolder
+  folder1:
+    Name: FileTest-folder1
+  folder2:
+    Name: FileTest-folder2
+  folder1-subfolder1:
+    Name: FileTest-folder1-subfolder1
+    ParentID: =>Folder.folder1
+File:
+  asdf:
+    Filename: assets/FileTest.txt
+  subfolderfile1:
+    Filename: assets/FileTest-subfolder/TestFile1InSubfolder.txt
+    Name: TestFile1InSubfolder
+    ParentID: =>Folder.subfolder
+  subfolderfile2:
+    Filename: assets/FileTest-subfolder/TestFile2InSubfolder.txt
+    Name: TestFile2InSubfolder
+    ParentID: =>Folder.subfolder
+  file1-folder1:
+    Filename: assets/FileTest-folder1/File1.txt
+    Name: File1.txt
+    ParentID: =>Folder.folder1