-
Notifications
You must be signed in to change notification settings - Fork 632
/
CustomNodeTests.cs
115 lines (101 loc) · 4.88 KB
/
CustomNodeTests.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
using System;
using System.IO;
using System.Linq;
using CoreNodeModels;
using CoreNodeModels.Input;
using Dynamo.Graph.Annotations;
using Dynamo.Graph.Nodes;
using Dynamo.Graph.Notes;
using Dynamo.Models;
using Dynamo.Selection;
using Dynamo.Tests;
using NUnit.Framework;
namespace DynamoCoreWpfTests
{
[TestFixture]
class CustomNodeTests : DynamoViewModelUnitTest
{
[Test]
public void TestNewNodeFromSelectionCommandState()
{
Assert.IsFalse(ViewModel.CurrentSpaceViewModel.NodeFromSelectionCommand.CanExecute(null));
var node = new DoubleInput();
ViewModel.Model.CurrentWorkspace.AddAndRegisterNode(node, false);
ViewModel.Model.AddToSelection(node);
Assert.IsTrue(ViewModel.CurrentSpaceViewModel.NodeFromSelectionCommand.CanExecute(null));
var ws = ViewModel.Model.CustomNodeManager.Collapse(
DynamoSelection.Instance.Selection.OfType<NodeModel>(),
Enumerable.Empty<NoteModel>(),
ViewModel.Model.CurrentWorkspace,
true,
new FunctionNamePromptEventArgs
{
Category = "Testing",
Description = "",
Name = "__CollapseTest2__",
Success = true
});
Assert.IsFalse(ViewModel.CurrentSpaceViewModel.NodeFromSelectionCommand.CanExecute(null));
}
[Test]
public void TestNestedGroupsOnCustomNodes()
{
string openPath = Path.Combine(TestDirectory, @"core\collapse\collapse-group.dyn");
OpenModel(openPath);
foreach (var node in ViewModel.CurrentSpaceViewModel.Nodes)
{
DynamoSelection.Instance.Selection.Add(node.NodeModel);
}
foreach (var note in ViewModel.CurrentSpaceViewModel.Notes)
{
DynamoSelection.Instance.Selection.Add(note.Model);
}
foreach (var grp in ViewModel.CurrentSpaceViewModel.Annotations)
{
DynamoSelection.Instance.Selection.Add(grp.AnnotationModel);
}
//assert that everything is selected
var groupsSelected = DynamoSelection.Instance.Selection.OfType<AnnotationModel>().Count();
var nodesSelected = DynamoSelection.Instance.Selection.OfType<NodeModel>().Count();
var notesSelected = DynamoSelection.Instance.Selection.OfType<NoteModel>().Count();
Assert.AreEqual(ViewModel.CurrentSpaceViewModel.Annotations.Count(), groupsSelected); // 3 groups
Assert.AreEqual(ViewModel.CurrentSpaceViewModel.Nodes.Count(), nodesSelected); //6 nodes
Assert.AreEqual(ViewModel.CurrentSpaceViewModel.Notes.Count(), notesSelected); // 2 notes
var ws = ViewModel.Model.CustomNodeManager.Collapse(
DynamoSelection.Instance.Selection.OfType<NodeModel>(),
DynamoSelection.Instance.Selection.OfType<NoteModel>(),
ViewModel.Model.CurrentWorkspace,
true,
new FunctionNamePromptEventArgs
{
Category = "Testing",
Description = "",
Name = "__CollapseTest3__",
Success = true
});
ViewModel.Model.AddCustomNodeWorkspace(ws);
ViewModel.Model.OpenCustomNodeWorkspace(ws.CustomNodeId);
var customNodeWorkspace = ViewModel.Model.CurrentWorkspace;
Assert.AreEqual(groupsSelected, customNodeWorkspace.Annotations.Count());
Assert.AreEqual(nodesSelected, customNodeWorkspace.Nodes.Count() - 1); //Subtract 1 for the Output Node
Assert.AreEqual(notesSelected, customNodeWorkspace.Notes.Count());
//assert that note and groups are still associated
var maingroup = customNodeWorkspace.Annotations.First(x => x.AnnotationText.Equals("Test"));
Assert.AreEqual(2, maingroup.Nodes.OfType<AnnotationModel>().Count()); // 2 groups inside the main group
Assert.AreEqual(1, maingroup.Nodes.OfType<NoteModel>().Count()); // 1 note inside the main group
}
[Test]
public void TestDataInputInitializationTest()
{
var node = new DefineData();
ViewModel.Model.CurrentWorkspace.AddAndRegisterNode(node, false);
var supportedDynamoTypesList = DSCore.Data.DataNodeDynamoTypeList;
// Assert - default node values
Assert.AreEqual(node.SelectedString, supportedDynamoTypesList.First().Name);
Assert.AreEqual(node.DisplayValue, CoreNodeModels.Properties.Resources.DefineDataDisplayValueMessage);
Assert.AreEqual(node.Value, string.Empty);
Assert.IsTrue(node.IsAutoMode); // default value of automode is now true
Assert.IsFalse(node.IsList);
}
}
}