Make TestUnit executable with a GUI button

Author: chocola-mint

Editor/TestUnitDescriptor.cs

@@ -17,7 +17,7 @@ public TestUnitDescriptor(TEvent @event) : base(@event) { }
         // }
         protected override string DefinedSummary()
         {
-            return "Double click to execute this node."
+            return "Click on the Run button to execute this node."
             + "\nWorks in both Edit Mode and Play Mode."
             + "\nNote 1: Coroutine tests are only usable in Play Mode."
             + "\nNote 2: Exceptions in edit mode won't go away on their own. "

Editor/TestUnitWidget.cs

@@ -9,13 +9,26 @@ namespace CHM.VisualScriptingPlus.Editor
     [Widget(typeof(TestUnit))]
     public class TestUnitWidget : UnitWidget<TestUnit>
     {
+        public override bool foregroundRequiresInput => true;
         public TestUnitWidget(FlowCanvas canvas, TestUnit unit) : base(canvas, unit)
         {
         }
-        protected override void OnDoubleClick()
+        
+        public override void DrawForeground()
         {
-            base.OnDoubleClick();
-            Execute();
+            base.DrawForeground();
+            // We implement the button here instead of using a custom button attribute
+            // on TestUnit, because crucially a node needs a GraphReference when triggered.
+            // Units don't have direct access to GraphReferences, but widgets do.
+            Rect buttonPosition = portsBackgroundPosition;
+            Vector2 margin = Vector2.one * 8;
+            buttonPosition.position += margin / 2;
+            buttonPosition.size -= margin;
+            buttonPosition.size -= Vector2.right * 22;
+            if(GUI.Button(buttonPosition, "Run"))
+            {
+                Execute();
+            }
         }
 
         private void Execute()
@@ -27,10 +40,12 @@ private void Execute()
             }
             unit.Trigger(reference, 0);
         }
+
         private void ClearExceptions()
         {
             GraphUtility.ClearExceptionsInActiveWindow();
         }
+
         protected override IEnumerable<DropdownOption> contextOptions
         {
             get

Runtime/TestUnit.cs

@@ -9,6 +9,6 @@ namespace CHM.VisualScriptingPlus
     [UnitCategory("Events\\VSPlus")]//Set the path to find the node in the fuzzy finder as Events > My Events.
     public class TestUnit : EventUnit<int>
     {
-        protected override bool register => true;
+        protected override bool register => false;
     }
 }