Use multiple resolutions for Percy screenshots

Author: 12joan

Assets/test/TestHelpers.cs

@@ -1,18 +1,29 @@
 #if UNITY_EDITOR
+using System;
 using System.Collections;
 using System.Collections.Generic;
 using System.Linq;
 using System.Text.RegularExpressions;
+using System.Reflection;
 using NUnit.Framework;
 using UnityEngine;
 using UnityEngine.UI;
 using UnityEngine.EventSystems;
 using UnityEngine.TestTools;
 using UnityEngine.SceneManagement;
+using UnityEditor;
 using UnityEditor.SceneManagement;
+
 using TMPro;
 
 public abstract class ABaseTest {
+  private const string DefaultTestResolution = "16:9";
+  private static readonly string[] ScreenshotResolutions = {
+    "4:3",
+    "16:9",
+    "32:9",
+  };
+
   [UnitySetUp]
   public IEnumerator CommonSetUp() {
     string runId = System.Guid.NewGuid().ToString();
@@ -22,6 +33,7 @@ public IEnumerator CommonSetUp() {
     GameData.Current.Clear();
     ConfigData.Current.Clear();
     CheckpointData.Reset();
+    SetResolution(DefaultTestResolution);
     yield return null;
   }
 
@@ -30,6 +42,7 @@ public IEnumerator CommonTearDown() {
     StopMoving();
     StopZooming();
     ClearMousePosition();
+    SetResolution("Free Aspect");
     yield return null;
   }
 
@@ -517,24 +530,86 @@ protected IEnumerator TakePercyScreenshot(string name) {
     backgroundColor.a = 1;
     camera.backgroundColor = backgroundColor;
 
-    RenderTexture screenTexture = new RenderTexture(Screen.width, Screen.height, 16);
-    RenderTexture previousTargetTexture = camera.targetTexture;
+    foreach (string resolution in ScreenshotResolutions) {
+      SetResolution(resolution);
+      yield return null;
+
+      RenderTexture screenTexture = new RenderTexture(Screen.width, Screen.height, 16);
+      RenderTexture previousTargetTexture = camera.targetTexture;
 
-    camera.targetTexture = screenTexture;
-    RenderTexture.active = screenTexture;
-    camera.Render();
+      camera.targetTexture = screenTexture;
+      RenderTexture.active = screenTexture;
+      camera.Render();
 
-    Texture2D renderedTexture = new Texture2D(Screen.width, Screen.height);
-    renderedTexture.ReadPixels(new Rect(0, 0, Screen.width, Screen.height), 0, 0);
+      Texture2D renderedTexture = new Texture2D(Screen.width, Screen.height);
+      renderedTexture.ReadPixels(new Rect(0, 0, Screen.width, Screen.height), 0, 0);
 
-    // Clean up
-    camera.targetTexture = previousTargetTexture;
-    RenderTexture.active = null;
+      // Clean up
+      camera.targetTexture = previousTargetTexture;
+      RenderTexture.active = null;
 
-    byte[] byteArray = renderedTexture.EncodeToPNG();
-    System.IO.File.WriteAllBytes("./Percy/" + name + ".png", byteArray);
+      string path = String.Format(
+        "./Percy/{0}-{1}.png",
+        name,
+        resolution.Replace(':', 'x')
+      );
+
+      byte[] byteArray = renderedTexture.EncodeToPNG();
+      System.IO.File.WriteAllBytes(path, byteArray);
+    }
+
+    SetResolution(DefaultTestResolution);
 
     yield return null;
   }
+
+  private void SetResolution(string key) {
+    var GameViewSizes = typeof(Editor)
+      .Assembly
+      .GetType("UnityEditor.GameViewSizes");
+
+    var gameViewSizes = typeof(ScriptableSingleton<>)
+      .MakeGenericType(GameViewSizes)
+      .GetProperty("instance")
+      .GetValue(null, null);
+
+    var group = GameViewSizes.GetMethod("GetGroup").Invoke(
+      gameViewSizes,
+      new object[] { (int)GameViewSizeGroupType.Standalone }
+    );
+
+    string[] displayTexts = group
+      .GetType()
+      .GetMethod("GetDisplayTexts")
+      .Invoke(group, null) as string[];
+
+    int index = Array.FindIndex(displayTexts, (displayText) =>
+      displayText == key ||
+      displayText.StartsWith("Test " + key)
+    );
+
+    if (index == -1) {
+      throw new System.Exception(
+        String.Format(
+          "Resolution not found: {0}. Available resolutions: {1}",
+          key,
+          String.Join(", ", displayTexts)
+        )
+      );
+    }
+
+    var GameView = typeof(Editor)
+      .Assembly
+      .GetType("UnityEditor.GameView");
+
+    var window = EditorWindow.GetWindow(GameView);
+
+    GameView
+      .GetProperty(
+        "selectedSizeIndex",
+        BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic
+      )
+      .SetValue(window, index, null);
+  }
 }
 #endif