#698 Add page snapshot functionality

atata-framework · Nov 16, 2022 · 6e1f9ec · 6e1f9ec
1 parent 9215071
commit 6e1f9ec
Show file tree

Hide file tree

Showing 2 changed files with 59 additions and 17 deletions.
diff --git a/src/Atata/Logging/Snapshots/PageSnapshotTaker.cs b/src/Atata/Logging/Snapshots/PageSnapshotTaker.cs
@@ -1,4 +1,5 @@
-using System.Collections.Generic;
+using System;
+using System.Collections.Generic;
 
 namespace Atata
 {
@@ -29,16 +30,23 @@ public void TakeSnapshot(string title = null)
 
             _snapshotNumber++;
 
-            _context.Log.ExecuteSection(
-                new TakePageSnapshotLogSection(_snapshotNumber, title),
-                () =>
-                {
-                    FileContentWithExtension fileContent = _snapshotStrategy.TakeSnapshot(_context);
-                    string filePath = FormatFilePath(title);
+            try
+            {
+                _context.Log.ExecuteSection(
+                    new TakePageSnapshotLogSection(_snapshotNumber, title),
+                    () =>
+                    {
+                        FileContentWithExtension fileContent = _snapshotStrategy.TakeSnapshot(_context);
+                        string filePath = FormatFilePath(title);
 
-                    _context.AddArtifact(filePath, fileContent, ArtifactTypes.PageSnapshot);
-                    return filePath + fileContent.Extension;
-                });
+                        _context.AddArtifact(filePath, fileContent, ArtifactTypes.PageSnapshot);
+                        return filePath + fileContent.Extension;
+                    });
+            }
+            catch (Exception e)
+            {
+                _context.Log.Error("Page snapshot failed", e);
+            }
         }
 
         private string FormatFilePath(string title)

diff --git a/test/Atata.IntegrationTests/Context/AtataContextTests.cs b/test/Atata.IntegrationTests/Context/AtataContextTests.cs
@@ -196,15 +196,12 @@ public class FillTemplateString : UITestFixtureBase
 
     public class TakeScreenshot : UITestFixtureBase
     {
-        [SetUp]
-        public void SetUp() =>
-            ConfigureBaseAtataContext()
-                .ScreenshotConsumers.AddFile()
-                .Build();
-
         [Test]
         public void WhenNavigated()
         {
+            ConfigureBaseAtataContext()
+                .ScreenshotConsumers.AddFile()
+                .Build();
             Go.To<InputPage>();
 
             AtataContext.Current.TakeScreenshot();
@@ -215,17 +212,37 @@ public void WhenNavigated()
         [Test]
         public void WhenNoNavigation()
         {
+            ConfigureBaseAtataContext()
+                .ScreenshotConsumers.AddFile()
+                .Build();
+
             AtataContext.Current.TakeScreenshot();
 
             AtataContext.Current.Artifacts.Should.ContainFile("01.png");
         }
+
+        [Test]
+        public void WhenThrows()
+        {
+            ConfigureBaseAtataContext()
+                .ScreenshotConsumers.Add(Mock.Of<IScreenshotConsumer>(MockBehavior.Strict))
+                .Build();
+            Go.To<InputPage>();
+
+            AtataContext.Current.TakeScreenshot();
+
+            VerifyLastLogMessagesContain(LogLevel.Error, "Screenshot failed");
+            AtataContext.Current.Artifacts.Should.Not.Exist();
+        }
     }
 
-    public class TakePageSnapshot : UITestFixture
+    public class TakePageSnapshot : UITestFixtureBase
     {
         [Test]
         public void WhenNavigated()
         {
+            ConfigureBaseAtataContext()
+                .Build();
             Go.To<InputPage>();
 
             AtataContext.Current.TakePageSnapshot();
@@ -236,9 +253,26 @@ public void WhenNavigated()
         [Test]
         public void WhenNoNavigation()
         {
+            ConfigureBaseAtataContext()
+                .Build();
+
             AtataContext.Current.TakePageSnapshot();
 
             AtataContext.Current.Artifacts.Should.ContainFile("01.mhtml");
         }
+
+        [Test]
+        public void WhenThrows()
+        {
+            ConfigureBaseAtataContext()
+                .PageSnapshots.UseStrategy(Mock.Of<IPageSnapshotStrategy>(MockBehavior.Strict))
+                .Build();
+            Go.To<InputPage>();
+
+            AtataContext.Current.TakePageSnapshot();
+
+            VerifyLastLogMessagesContain(LogLevel.Error, "Page snapshot failed");
+            AtataContext.Current.Artifacts.Should.Not.Exist();
+        }
     }
 }