Test coverage of contextual generators list

Author: elsassph

External/Plugins/ASCompletion/Completion/ASGenerator.cs

@@ -34,11 +34,11 @@ public class ASGenerator
         static private Regex reModifier = new Regex("(public |private |protected )", RegexOptions.Compiled);
         static private Regex reSuperCall = new Regex("^super\\s*\\(", RegexOptions.Compiled);
 
-        static private string contextToken;
-        static private string contextParam;
-        static private Match contextMatch;
-        static private ASResult contextResolved;
-        static private MemberModel contextMember;
+        static internal string contextToken;
+        static internal string contextParam;
+        static internal Match contextMatch;
+        static internal ASResult contextResolved;
+        static internal MemberModel contextMember;
         static private bool firstVar;
 
         static private bool IsHaxe
@@ -795,7 +795,7 @@ private static void ShowDelegateList(FoundDeclaration found, List<ICompletionLis
             options.Add(new GeneratorItem(label, GeneratorJobType.Delegate, found.member, found.inClass));
         }
 
-        private static void ShowEventList(FoundDeclaration found, List<ICompletionListItem> options)
+        internal static void ShowEventList(FoundDeclaration found, List<ICompletionListItem> options)
         {
             string tmp = TextHelper.GetString("ASCompletion.Label.GenerateHandler");
             string labelEvent = String.Format(tmp, "Event");
@@ -4611,7 +4611,7 @@ public Object Data
         }
     }
 
-    class FoundDeclaration
+    internal class FoundDeclaration
     {
         public MemberModel member;
         public ClassModel inClass;

Tests/External/Plugins/ASCompletion.Tests/Completion/ASGeneratorTests.cs

@@ -13,6 +13,7 @@
 using PluginCore;
 using ScintillaNet;
 using ScintillaNet.Enums;
+using System.Text.RegularExpressions;
 
 namespace ASCompletion.Completion
 {
@@ -103,6 +104,90 @@ public void Common(string text, int lineStart, int lineEnd, string resultText, i
             }
         }
 
+        [TestFixture]
+        public class ContextualActions : ASGeneratorTests
+        {
+            [TestFixtureSetUp]
+            public void ContextualActionsSetup()
+            {
+                var pluginMain = Substitute.For<PluginMain>();
+                var pluginUiMock = new PluginUIMock(pluginMain);
+                pluginMain.MenuItems.Returns(new List<System.Windows.Forms.ToolStripItem>());
+                pluginMain.Settings.Returns(new GeneralSettings());
+                pluginMain.Panel.Returns(pluginUiMock);
+                ASContext.GlobalInit(pluginMain);
+                ASContext.Context = Substitute.For<IASContext>();
+            }
+
+            [TestFixture]
+            class ShowEventsList : ContextualActions
+            {
+                ClassModel dataEventModel;
+                FoundDeclaration found;
+
+                [TestFixtureSetUp]
+                public void ShowEventsListSetup()
+                {
+                    ASContext.Context.SetAs3Features();
+                    ASContext.Context.CurrentModel.Returns(new FileModel());
+                    dataEventModel = CreateDataEventModel();
+                    found = new FoundDeclaration
+                    {
+                        inClass = new ClassModel(),
+                        member = new MemberModel()
+                    };
+                }
+
+                [Test]
+                public void ShowEventsList_EventWithDataEvent()
+                {
+                    ASContext.Context.ResolveType(null, null).ReturnsForAnyArgs(dataEventModel);
+                    ASGenerator.contextParam = "Event";
+                    var options = new List<ICompletionListItem>();
+                    ASGenerator.ShowEventList(found, options);
+                    Assert.AreEqual(2, options.Count);
+                    Assert.IsTrue(Regex.IsMatch(options[0].Label, "\\bEvent\\b"));
+                    Assert.IsTrue(Regex.IsMatch(options[1].Label, "\\bDataEvent\\b"));
+                }
+
+                [Test]
+                public void ShowEventsList_EventWithoutDataEvent()
+                {
+                    ASContext.Context.ResolveType(null, null).ReturnsForAnyArgs(ClassModel.VoidClass);
+                    var options = new List<ICompletionListItem>();
+                    ASGenerator.contextParam = "Event";
+                    ASGenerator.ShowEventList(found, options);
+                    Assert.AreEqual(1, options.Count);
+                    Assert.IsTrue(Regex.IsMatch(options[0].Label, "\\bEvent\\b"));
+                }
+
+                [Test]
+                public void ShowEventsList_CustomEventWithDataEvent()
+                {
+                    ASContext.Context.ResolveType(null, null).ReturnsForAnyArgs(dataEventModel);
+                    var options = new List<ICompletionListItem>();
+                    ASGenerator.contextParam = "CustomEvent";
+                    ASGenerator.ShowEventList(found, options);
+                    Assert.AreEqual(2, options.Count);
+                    Assert.IsTrue(Regex.IsMatch(options[0].Label, "\\bCustomEvent\\b"));
+                    Assert.IsTrue(Regex.IsMatch(options[1].Label, "\\bEvent\\b"));
+                }
+
+                private ClassModel CreateDataEventModel()
+                {
+                    var dataEventFile = new FileModel();
+                    var dataEventModel = new ClassModel
+                    {
+                        Name = "DataEvent",
+                        InFile = dataEventFile
+                    };
+                    dataEventFile.Classes.Add(dataEventModel);
+                    return dataEventModel;
+                }
+            }
+
+        }
+
         [TestFixture]
         public class GenerateJob : ASGeneratorTests
         {