123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259 |
- using System;
- using UnityEditor.Callbacks;
- using UnityEditor.TestTools.TestRunner.Api;
- using UnityEditor.TestTools.TestRunner.GUI;
- using UnityEngine;
- namespace UnityEditor.TestTools.TestRunner
- {
- [Serializable]
- internal class TestRunnerWindow : EditorWindow, IHasCustomMenu
- {
- internal static class Styles
- {
- public static GUIStyle info;
- public static GUIStyle testList;
- static Styles()
- {
- info = new GUIStyle(EditorStyles.wordWrappedLabel);
- info.wordWrap = false;
- info.stretchHeight = true;
- info.margin.right = 15;
- testList = new GUIStyle("CN Box");
- testList.margin.top = 0;
- testList.padding.left = 3;
- }
- }
- private readonly GUIContent m_GUIHorizontalSplit = EditorGUIUtility.TrTextContent("Horizontal layout");
- private readonly GUIContent m_GUIVerticalSplit = EditorGUIUtility.TrTextContent("Vertical layout");
- private readonly GUIContent m_GUIEnableaPlaymodeTestsRunner = EditorGUIUtility.TrTextContent("Enable playmode tests for all assemblies");
- private readonly GUIContent m_GUIDisablePlaymodeTestsRunner = EditorGUIUtility.TrTextContent("Disable playmode tests for all assemblies");
- private readonly GUIContent m_GUIRunPlayModeTestAsEditModeTests = EditorGUIUtility.TrTextContent("Run playmode tests as editmode tests");
- internal static TestRunnerWindow s_Instance;
- private bool m_IsBuilding;
- [NonSerialized]
- private bool m_Enabled;
- public TestFilterSettings filterSettings;
- [SerializeField]
- private SplitterState m_Spl = new SplitterState(new float[] { 75, 25 }, new[] { 32, 32 }, null);
- private TestRunnerWindowSettings m_Settings;
- private enum TestRunnerMenuLabels
- {
- PlayMode = 0,
- EditMode = 1
- }
- [SerializeField]
- private int m_TestTypeToolbarIndex = (int)TestRunnerMenuLabels.EditMode;
- [SerializeField]
- private PlayModeTestListGUI m_PlayModeTestListGUI;
- [SerializeField]
- private EditModeTestListGUI m_EditModeTestListGUI;
- internal TestListGUI m_SelectedTestTypes;
- private ITestRunnerApi m_testRunnerApi;
- private WindowResultUpdater m_WindowResultUpdater;
- [MenuItem("Window/General/Test Runner", false, 201, false)]
- public static void ShowPlaymodeTestsRunnerWindowCodeBased()
- {
- s_Instance = GetWindow<TestRunnerWindow>("Test Runner");
- s_Instance.Show();
- }
- static TestRunnerWindow()
- {
- InitBackgroundRunners();
- }
- private static void InitBackgroundRunners()
- {
- EditorApplication.playModeStateChanged -= OnPlayModeStateChanged;
- EditorApplication.playModeStateChanged += OnPlayModeStateChanged;
- }
- [DidReloadScripts]
- private static void CompilationCallback()
- {
- UpdateWindow();
- }
- private static void OnPlayModeStateChanged(PlayModeStateChange state)
- {
- if (s_Instance && state == PlayModeStateChange.EnteredEditMode && s_Instance.m_SelectedTestTypes.HasTreeData())
- {
- //repaint message details after exit playmode
- s_Instance.m_SelectedTestTypes.TestSelectionCallback(s_Instance.m_SelectedTestTypes.m_TestListState.selectedIDs.ToArray());
- s_Instance.Repaint();
- }
- }
- public void OnDestroy()
- {
- EditorApplication.playModeStateChanged -= OnPlayModeStateChanged;
- }
- private void OnEnable()
- {
- s_Instance = this;
- SelectTestListGUI(m_TestTypeToolbarIndex);
- m_testRunnerApi = ScriptableObject.CreateInstance<TestRunnerApi>();
- m_WindowResultUpdater = new WindowResultUpdater();
- m_testRunnerApi.RegisterCallbacks(m_WindowResultUpdater);
- }
- private void Enable()
- {
- m_Settings = new TestRunnerWindowSettings("UnityEditor.PlaymodeTestsRunnerWindow");
- filterSettings = new TestFilterSettings("UnityTest.IntegrationTestsRunnerWindow");
- if (m_SelectedTestTypes == null)
- {
- SelectTestListGUI(m_TestTypeToolbarIndex);
- }
- StartRetrieveTestList();
- m_SelectedTestTypes.Reload();
- m_Enabled = true;
- }
- private void SelectTestListGUI(int testTypeToolbarIndex)
- {
- if (testTypeToolbarIndex == (int)TestRunnerMenuLabels.PlayMode)
- {
- if (m_PlayModeTestListGUI == null)
- {
- m_PlayModeTestListGUI = new PlayModeTestListGUI();
- }
- m_SelectedTestTypes = m_PlayModeTestListGUI;
- }
- else if (testTypeToolbarIndex == (int)TestRunnerMenuLabels.EditMode)
- {
- if (m_EditModeTestListGUI == null)
- {
- m_EditModeTestListGUI = new EditModeTestListGUI();
- }
- m_SelectedTestTypes = m_EditModeTestListGUI;
- }
- }
- private void StartRetrieveTestList()
- {
- if (!m_SelectedTestTypes.HasTreeData())
- {
- m_testRunnerApi.RetrieveTestList(m_SelectedTestTypes.TestMode, (rootTest) =>
- {
- m_SelectedTestTypes.Init(this, rootTest);
- m_SelectedTestTypes.Reload();
- });
- }
- }
- public void OnGUI()
- {
- if (!m_Enabled)
- {
- Enable();
- }
- if (BuildPipeline.isBuildingPlayer)
- {
- m_IsBuilding = true;
- }
- else if (m_IsBuilding)
- {
- m_IsBuilding = false;
- Repaint();
- }
- EditorGUILayout.BeginHorizontal();
- GUILayout.FlexibleSpace();
- var selectedIndex = m_TestTypeToolbarIndex;
- m_TestTypeToolbarIndex = GUILayout.Toolbar(m_TestTypeToolbarIndex, Enum.GetNames(typeof(TestRunnerMenuLabels)), "LargeButton", UnityEngine.GUI.ToolbarButtonSize.FitToContents);
- GUILayout.FlexibleSpace();
- EditorGUILayout.EndHorizontal();
- if (selectedIndex != m_TestTypeToolbarIndex)
- {
- SelectTestListGUI(m_TestTypeToolbarIndex);
- StartRetrieveTestList();
- }
- EditorGUILayout.BeginVertical();
- using (new EditorGUI.DisabledScope(EditorApplication.isPlayingOrWillChangePlaymode))
- {
- m_SelectedTestTypes.PrintHeadPanel();
- }
- EditorGUILayout.EndVertical();
- if (m_Settings.verticalSplit)
- SplitterGUILayout.BeginVerticalSplit(m_Spl);
- else
- SplitterGUILayout.BeginHorizontalSplit(m_Spl);
- EditorGUILayout.BeginVertical();
- EditorGUILayout.BeginVertical(Styles.testList);
- m_SelectedTestTypes.RenderTestList();
- EditorGUILayout.EndVertical();
- EditorGUILayout.EndVertical();
- m_SelectedTestTypes.RenderDetails();
- if (m_Settings.verticalSplit)
- SplitterGUILayout.EndVerticalSplit();
- else
- SplitterGUILayout.EndHorizontalSplit();
- }
- public void AddItemsToMenu(GenericMenu menu)
- {
- menu.AddItem(m_GUIVerticalSplit, m_Settings.verticalSplit, m_Settings.ToggleVerticalSplit);
- menu.AddItem(m_GUIHorizontalSplit, !m_Settings.verticalSplit, m_Settings.ToggleVerticalSplit);
- menu.AddSeparator(null);
- var playModeTestRunnerEnabled = PlayerSettings.playModeTestRunnerEnabled;
- var currentActive = playModeTestRunnerEnabled ? m_GUIDisablePlaymodeTestsRunner : m_GUIEnableaPlaymodeTestsRunner;
- if (EditorPrefs.GetBool("InternalMode", false))
- {
- menu.AddItem(m_GUIRunPlayModeTestAsEditModeTests, PlayerSettings.runPlayModeTestAsEditModeTest, () =>
- {
- PlayerSettings.runPlayModeTestAsEditModeTest = !PlayerSettings.runPlayModeTestAsEditModeTest;
- });
- }
- menu.AddItem(currentActive, false, () =>
- {
- PlayerSettings.playModeTestRunnerEnabled = !playModeTestRunnerEnabled;
- EditorUtility.DisplayDialog(currentActive.text, "You need to restart the editor now", "Ok");
- });
- }
- public void RebuildUIFilter()
- {
- if (m_SelectedTestTypes != null && m_SelectedTestTypes.HasTreeData())
- {
- m_SelectedTestTypes.RebuildUIFilter();
- }
- }
- public static void UpdateWindow()
- {
- if (s_Instance != null && s_Instance.m_SelectedTestTypes != null)
- {
- s_Instance.m_SelectedTestTypes.Repaint();
- s_Instance.Repaint();
- }
- }
- }
- }
|