!3784 test: add TreeNodeCache unit test

* test: 增加 Reset 单元测试 * test: 增加 FindParentNode 单元测试 * fix: 修复不确定状态节点状态未重置问题 * test: 增加设置初始化状态逻辑单元测试 * refactor: 更新单元测试 * test: 增加内部数据检查 * test: 增加 ToggleCheck 单元测试 * test: 增加 TreeNodeCache 单元测试
2024-12-08 23:18:53 +08:00 · 2023-01-21 14:47:25 +00:00 · 2023-01-21 14:47:25 +00:00 · e4d68935e8
commit e4d68935e8
parent 63b680bb42
2 changed files with 311 additions and 0 deletions
--- a/src/BootstrapBlazor/Misc/TreeNodeCache.cs
+++ b/src/BootstrapBlazor/Misc/TreeNodeCache.cs
@ -190,6 +190,7 @@ public class TreeNodeCache<TNode, TItem> : ExpandableNodeCache<TNode, TItem> whe
    {
        uncheckedNodeCache.Clear();
        checkedNodeCache.Clear();
+        indeterminateNodeCache.Clear();
        expandedNodeCache.Clear();
        collapsedNodeCache.Clear();
    }
--- a/test/UnitTest/Misc/TreeNodeCacheTest.cs
+++ b/test/UnitTest/Misc/TreeNodeCacheTest.cs
@ -0,0 +1,310 @@
+// Copyright (c) Argo Zhang (argo@163.com). All rights reserved.
+// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
+// Website: https://www.blazor.zone or https://argozhang.github.io/
+
+using BootstrapBlazor.Shared;
+
+namespace UnitTest.Misc;
+
+public class TreeNodeCacheTest
+{
+    [Fact]
+    public void ToggleCheck_Ok()
+    {
+        var items = TreeFoo.GetTreeItems();
+        var nodeCache = new TreeNodeCache<TreeViewItem<TreeFoo>, TreeFoo>(Comparer);
+        nodeCache.IsChecked(items);
+
+        // 设置 1010 节点为选中状态
+        var node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1010",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Checked
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1020 节点为未确定状态
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1020",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Indeterminate
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1040 节点为未选中状态
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1040",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.UnChecked
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1100 子节点全部被选中
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1130",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Checked
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1100 子节点全部被选中
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1140",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Checked
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1100 子节点全部被选中
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1150",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Checked
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1100 子节点选中
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1100",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Checked
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1080 节点为选中状态
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1080",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Indeterminate
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1050 节点为选中状态
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1050",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Indeterminate
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1020 节点为选中状态
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1020",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Indeterminate
+        };
+        nodeCache.ToggleCheck(node);
+
+        var count = GetCheckItemCount(nodeCache);
+        Assert.Equal(5, count);
+
+        count = GetUncheckItemCount(nodeCache);
+        Assert.Equal(1, count);
+
+        count = GetIndeterminateItemCount(nodeCache);
+        Assert.Equal(3, count);
+
+        // 开始测试
+        // 设置 1030 节点为选中状态
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1030",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Checked
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1030 节点为未选中状态
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1030",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.UnChecked
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1030 节点为选中状态
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1030",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Checked
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 重给 Items
+        items = TreeFoo.GetTreeItems();
+        nodeCache.IsChecked(items);
+
+        // 检查结果
+        var nodes = items.GetAllItems();
+        var state = nodes.First(i => i.Value?.Id == "1010").CheckedState;
+        Assert.Equal(CheckboxState.Checked, state);
+
+        state = nodes.First(i => i.Value?.Id == "1030").CheckedState;
+        Assert.Equal(CheckboxState.Checked, state);
+
+        state = nodes.First(i => i.Value?.Id == "1020").CheckedState;
+        Assert.Equal(CheckboxState.Indeterminate, state);
+
+        state = nodes.First(i => i.Value?.Id == "1050").CheckedState;
+        Assert.Equal(CheckboxState.Indeterminate, state);
+
+        state = nodes.First(i => i.Value?.Id == "1080").CheckedState;
+        Assert.Equal(CheckboxState.Indeterminate, state);
+
+        state = nodes.First(i => i.Value?.Id == "1100").CheckedState;
+        Assert.Equal(CheckboxState.Checked, state);
+
+        state = nodes.First(i => i.Value?.Id == "1130").CheckedState;
+        Assert.Equal(CheckboxState.Checked, state);
+
+        state = nodes.First(i => i.Value?.Id == "1130").CheckedState;
+        Assert.Equal(CheckboxState.Checked, state);
+
+        state = nodes.First(i => i.Value?.Id == "1130").CheckedState;
+        Assert.Equal(CheckboxState.Checked, state);
+    }
+
+    [Fact]
+    public void FindParentNode_Ok()
+    {
+        var items = TreeFoo.GetTreeItems();
+        var nodeCache = new TreeNodeCache<TreeViewItem<TreeFoo>, TreeFoo>(Comparer);
+        var targetId = nodeCache.FindParentNode(items, new TreeViewItem<TreeFoo>(new TreeFoo() { Id = "1110" }))?.Value?.Id;
+        Assert.Equal("1080", targetId);
+    }
+
+    [Fact]
+    public void Reset_Ok()
+    {
+        var items = TreeFoo.GetTreeItems();
+        var nodeCache = new TreeNodeCache<TreeViewItem<TreeFoo>, TreeFoo>(Comparer);
+
+        // 设置 1070 节点为选中状态
+        var node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1070",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Checked
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1080 节点为选中状态
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1080",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.Indeterminate
+        };
+        nodeCache.ToggleCheck(node);
+
+        // 设置 1090 节点为选中状态
+        node = new TreeViewItem<TreeFoo>(new TreeFoo()
+        {
+            Id = "1090",
+            Text = "Test"
+        })
+        {
+            CheckedState = CheckboxState.UnChecked
+        };
+        nodeCache.ToggleCheck(node);
+
+        var count = GetCheckItemCount(nodeCache);
+        Assert.Equal(1, count);
+
+        count = GetUncheckItemCount(nodeCache);
+        Assert.Equal(1, count);
+
+        count = GetIndeterminateItemCount(nodeCache);
+        Assert.Equal(1, count);
+
+        nodeCache.Reset();
+        count = GetCheckItemCount(nodeCache);
+        Assert.Equal(0, count);
+
+        count = GetUncheckItemCount(nodeCache);
+        Assert.Equal(0, count);
+
+        count = GetIndeterminateItemCount(nodeCache);
+        Assert.Equal(0, count);
+    }
+
+    private bool Comparer(TreeFoo x, TreeFoo y) => x.Id == y.Id;
+
+    private static int GetCheckItemCount(TreeNodeCache<TreeViewItem<TreeFoo>, TreeFoo> treeNodeCache)
+    {
+        var count = 0;
+        var type = treeNodeCache.GetType();
+        var fieldInfo = type.GetField("checkedNodeCache", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance);
+        if (fieldInfo != null && fieldInfo.GetValue(treeNodeCache) is List<TreeFoo> data)
+        {
+            count = data.Count;
+        }
+        return count;
+    }
+
+    private static int GetUncheckItemCount(TreeNodeCache<TreeViewItem<TreeFoo>, TreeFoo> treeNodeCache)
+    {
+        var count = 0;
+        var type = treeNodeCache.GetType();
+        var fieldInfo = type.GetField("uncheckedNodeCache", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance);
+        if (fieldInfo != null && fieldInfo.GetValue(treeNodeCache) is List<TreeFoo> data)
+        {
+            count = data.Count;
+        }
+        return count;
+    }
+
+    private static int GetIndeterminateItemCount(TreeNodeCache<TreeViewItem<TreeFoo>, TreeFoo> treeNodeCache)
+    {
+        var count = 0;
+        var type = treeNodeCache.GetType();
+        var fieldInfo = type.GetField("indeterminateNodeCache", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Instance);
+        if (fieldInfo != null && fieldInfo.GetValue(treeNodeCache) is List<TreeFoo> data)
+        {
+            count = data.Count;
+        }
+        return count;
+    }
+}