Merge pull request #1329 from chuacw/master

Added tests for AddColumn, EditNode pass and failure

Author: joachimmarder

Tests/VTOnEditCancelledTests.pas

@@ -22,14 +22,26 @@   TVTOnEditCancelledTests = class
     [TearDown]
     procedure TearDown;
 
+    [Test]
+    procedure TestAddColumn;
+
+    [Test]
+    procedure TestEditNodeFail;
+
+    [Test]
+    procedure TestEditNode;
+
+    [Test]
+    procedure TestEditNodeReadOnly;
+
     [Test]
     procedure TestOnEditCancelled;
   end;
 
 implementation
 
 uses
-  System.SysUtils;
+  System.SysUtils, VirtualTrees.Types;
 
 procedure TVTOnEditCancelledTests.Setup;
 begin
@@ -42,6 +54,49 @@ procedure TVTOnEditCancelledTests.TearDown;
   FreeAndNil(fForm);
 end;
 
+procedure TVTOnEditCancelledTests.TestAddColumn;
+begin
+  var LBeforeColumnCount := fTree.Header.Columns.Count;
+  fTree.Header.Columns.Add;
+  var LAfterColumnCount := fTree.Header.Columns.Count;
+  Assert.AreEqual<Integer>(LAfterColumnCount - LBeforeColumnCount, 1);
+end;
+
+procedure TVTOnEditCancelledTests.TestEditNode;
+begin
+  fForm.Show;
+  fTree.TreeOptions.MiscOptions := fTree.TreeOptions.MiscOptions + [toEditable];
+  fTree.Parent := fForm;
+  fTree.Header.Columns.Add;
+  var LNode := fTree.AddChild(fTree.RootNode);
+  var LEditNodeResult := fTree.EditNode(LNode, 0);
+  var LAfterStates := fTree.TreeStates;
+  Assert.AreEqual<TVirtualTreeStates>(LAfterStates * [tsEditing], [tsEditing]);
+  Assert.IsTrue(LEditNodeResult);
+end;
+
+procedure TVTOnEditCancelledTests.TestEditNodeFail;
+begin
+  fForm.Show;
+  fTree.TreeOptions.MiscOptions := fTree.TreeOptions.MiscOptions - [toEditable];
+  fTree.Parent := fForm;
+  fTree.Header.Columns.Add;
+  var LNode := fTree.AddChild(fTree.RootNode);
+  var LEditNodeResult := fTree.EditNode(LNode, 0);
+  Assert.IsFalse(LEditNodeResult);
+end;
+
+procedure TVTOnEditCancelledTests.TestEditNodeReadOnly;
+begin
+  fForm.Show;
+  fTree.Parent := fForm;
+  fTree.Header.Columns.Add;
+  var LNode := fTree.AddChild(fTree.RootNode);
+  fTree.TreeOptions.MiscOptions := fTree.TreeOptions.MiscOptions + [toReadOnly];
+  var LEditNodeResult := fTree.EditNode(LNode, 0);
+  Assert.IsFalse(LEditNodeResult);
+end;
+
 procedure TVTOnEditCancelledTests.TestOnEditCancelled;
 begin
   fForm.Show;