Merge pull request #350 from TestStack/HEskandari-master

Expose Capture on IUIItem
TestStack · Dec 23, 2015 · a214239 · a214239
2 parents c7057c3 + a6d19da
commit a214239
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/src/TestStack.White.UITests/WhiteUITestBase.cs b/src/TestStack.White.UITests/WhiteUITestBase.cs
@@ -67,7 +67,7 @@ private string TakeScreenshot(string screenshotName)
             var imagePath = Path.Combine(screenshotDir, imagename);
             try
             {
-                new ScreenCapture().CaptureScreenShot().Save(imagePath, ImageFormat.Png);
+                Desktop.CaptureScreenshot().Save(imagePath, ImageFormat.Png);
                 Trace.WriteLine(String.Format("Screenshot taken: {0}", imagePath));
             }
             catch (Exception)

diff --git a/src/TestStack.White/UIItems/IUIItem.cs b/src/TestStack.White/UIItems/IUIItem.cs
@@ -101,5 +101,10 @@ public interface IUIItem : ActionListener
 
         void DrawHighlight();
         void DrawHighlight(Color color);
+
+        /// <summary>
+        /// Captures an image of the element
+        /// </summary>
+        Bitmap Capture();
     }
 }
diff --git a/src/TestStack.White/UIItems/UIItem.cs b/src/TestStack.White/UIItems/UIItem.cs
@@ -449,5 +449,10 @@ public virtual void DrawHighlight(Color color)
                 new Drawing.FrameRectangle(color, rectangle).Highlight();
             }
         }
+
+        public virtual Bitmap Capture()
+        {
+            return Desktop.CaptureScreenshot(Bounds);
+        }
     }
 }