closes #18 - provide ability to remove a test

anshooarora · Mar 28, 2017 · dce7875 · dce7875
1 parent 5a1f613
commit dce7875
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/ExtentReports/ExtentReports/ExtentReports.cs b/ExtentReports/ExtentReports/ExtentReports.cs
@@ -121,6 +121,15 @@ public ExtentTest CreateTest(GherkinKeyword gherkinKeyword, string name, string
             return extentTest;
         }
 
+        /// <summary>
+        /// Allows removing a test from the list
+        /// </summary>
+        /// <param name="test"><see cref="ExtentTest"/> An ExtentTest object</param>
+        public void RemoveTest(ExtentTest test)
+        {
+            base.RemoveTest(test.GetModel());
+        }
+
         /// <summary>
         /// Writes test information from the started reporters to their output view
         /// </summary>

diff --git a/ExtentReports/ExtentReports/Model/AbstractStructure.cs b/ExtentReports/ExtentReports/Model/AbstractStructure.cs
@@ -26,6 +26,11 @@ public void Add(T t)
             _list.Add(t);
         }
 
+        public void Remove(T t)
+        {
+            _list.Remove(t);
+        }
+
         public T Get(int index)
         {
             return _list[index];

diff --git a/ExtentReports/ExtentReports/Model/Report.cs b/ExtentReports/ExtentReports/Model/Report.cs
@@ -62,6 +62,31 @@ protected void CreateTest(Test test)
             _reporterCollection.ForEach(x => x.OnTestStarted(test));
         }
 
+        [MethodImpl(MethodImplOptions.Synchronized)]
+        protected void RemoveTest(Test test)
+        {
+            foreach (var t in _testCollection)
+            {
+                if (t.TestId == test.TestId)
+                {
+                    _testCollection.Remove(t);
+                    return;
+                }
+
+                if (t.HasChildren())
+                {
+                    foreach (var n in t.NodeContext().GetAllItems())
+                    {
+                        if (n.TestId == test.TestId)
+                        {
+                            t.NodeContext().Remove(n);
+                            return;
+                        }
+                    }
+                }
+            }
+        }
+
         [MethodImpl(MethodImplOptions.Synchronized)]
         internal void AddNode(Test node)
         {