.NET: Enhance VisualCheckOptions

visual-dotnet/SauceLabs.Visual/Utils/DiffingOptionsInHelper.cs

@@ -7,7 +7,7 @@ namespace SauceLabs.Visual.Utils
 {
     public static class DiffingOptionsInHelper
     {
-        private static DiffingOptionsIn SetOptions(DiffingOptionsIn opts, DiffingOption flags, bool value)
+        internal static DiffingOptionsIn SetOptions(DiffingOptionsIn opts, DiffingOption flags, bool value)
         {
             if (flags.HasFlag(DiffingOption.Content))
             {

visual-dotnet/SauceLabs.Visual/VisualCheckDiffingOptions.cs

@@ -0,0 +1,59 @@
+using System.Diagnostics;
+using System.Linq;
+using OpenQA.Selenium;
+using SauceLabs.Visual.GraphQL;
+using SauceLabs.Visual.Models;
+using SauceLabs.Visual.Utils;
+
+namespace SauceLabs.Visual
+{
+    public class VisualCheckDiffingOptions
+    {
+        private enum DiffingOptionMode
+        {
+            EnableOnly,
+            DisableOnly
+        }
+
+        private readonly DiffingOptionMode _mode;
+        private readonly DiffingOption _flags;
+
+        private VisualCheckDiffingOptions(DiffingOptionMode mode, DiffingOption flags)
+        {
+            _mode = mode;
+            _flags = flags;
+        }
+
+        /// <summary>
+        /// <c>EnableOnly</c> sets which change types to check for.
+        /// Compatible only with <c>DiffingMethod.Balanced</c>
+        /// </summary>
+        public static VisualCheckDiffingOptions EnableOnly(DiffingOption flags)
+        {
+            return new VisualCheckDiffingOptions(DiffingOptionMode.EnableOnly, flags);
+        }
+
+        /// <summary>
+        /// <c>DisableOnly</c> sets which change types to ignore.
+        /// Compatible only with <c>DiffingMethod.Balanced</c>
+        /// </summary>
+        public static VisualCheckDiffingOptions DisableOnly(DiffingOption flags)
+        {
+            return new VisualCheckDiffingOptions(DiffingOptionMode.DisableOnly, flags);
+        }
+
+        internal DiffingOptionsIn ToDiffingOptionsIn()
+        {
+            DiffingOptionsIn options;
+            if (_mode == DiffingOptionMode.EnableOnly)
+            {
+                options = DiffingOptionsInHelper.SetOptions(new DiffingOptionsIn(false), _flags, true);
+            }
+            else
+            {
+                options = DiffingOptionsInHelper.SetOptions(new DiffingOptionsIn(true), _flags, false);
+            }
+            return options;
+        }
+    }
+}

visual-dotnet/SauceLabs.Visual/VisualCheckOptions.cs

@@ -20,19 +20,14 @@
         public string? ClipSelector { get; set; }
 
         /// <summary>
-        /// <c>EnableOnly</c> allows to specify which changes to consider globaly.
+        /// <c>DiffingOptions</c> set which kind of changes should be considered.
         /// </summary>
-        public DiffingOption? EnableOnly { get; set; }
-
-        /// <summary>
-        /// <c>Disable</c> allows to specify which changes to ignore globally.
-        /// </summary>
-        public DiffingOption? DisableOnly { get; set; }
+        public VisualCheckDiffingOptions? DiffingOptions { get; set; }
 
         /// <summary>
         /// <c>Regions</c> allows to specify what kind of checks needs to be done in a specific region.
         /// </summary>
         public SelectiveRegion[]? Regions { get; set; }
 
         /// <summary>
         /// <c>SuiteName</c> manually set the SuiteName of the Test.

visual-dotnet/SauceLabs.Visual/VisualClient.cs

@@ -92,7 +92,7 @@
         /// <param name="region">the Sauce Labs region to connect to</param>
         /// <param name="username">the Sauce Labs username</param>
         /// <param name="accessKey">the Sauce Labs access key</param>
         private VisualClient(WebDriver wd, Region region, string username, string accessKey)
         {
             if (StringUtils.IsNullOrEmpty(username) || StringUtils.IsNullOrEmpty(accessKey))
             {
@@ -165,7 +165,7 @@
                 suiteName: options.SuiteName,
                 testName: options.TestName,
                 fullPageConfig: fullPageConfigIn,
-                diffingOptions: DiffingOptionsInHelper.CreateFromEnableOnlyDisable(options.EnableOnly, options.DisableOnly)
+                diffingOptions: options.DiffingOptions?.ToDiffingOptionsIn()
             ))).EnsureValidResponse();
             result.Result.Diffs.Nodes.ToList().ForEach(d => _screenshotIds.Add(d.Id));
             return result.Result.Id;