unity-application/Assets/Common/Tests/PlayMode/MainMenuScreenTests.cs

using NUnit.Framework;
using System.Collections;
using System.IO;
using UnityEditor;
using UnityEngine;
using UnityEngine.SceneManagement;
using UnityEngine.TestTools;

/// <summary>
/// Test the MainMenuScreen class
/// </summary>
public class MainMenuScreenTests
{
    /// <summary>
    /// Setup function to prepare the test environment before each test.
    /// </summary>
    [UnitySetUp]
    public IEnumerator SetupFunction()
    {
        string path = $"{Application.persistentDataPath}/wesign_unit_test.json";
        string oneUser = $"{{\"version\":1537,\"users\":[{{\"entries\":[],\"username\":\"TEST\",\"avatarIndex\":0,\"playtime\":0.0,\"minigames\":[],\"courses\":[]}}],\"currentUser\":0,\"currentMinigame\":0,\"currentCourse\":0,\"currentTheme\":0}}";

        File.WriteAllText(path, oneUser);
        PersistentDataController.PATH = path;
        PersistentDataController.GetInstance().Load();
        AssetDatabase.LoadAssetAtPath<UserAvatarList>("Assets/Accounts/ScriptableObjects/UserAvatarList.asset").Awake();
        SceneManager.LoadScene("Common/Scenes/MainMenuScreen");
        yield return new WaitForSeconds(0.2f);
    }

    /// <summary>
    /// Cleanup after testing.
    /// </summary>
    [TearDown]
    public void TearDown_MainMenuScreenTests()
    {
        PersistentDataController.PATH = null;
    }

    /// <summary>
    /// Test to verify the transition to the CourseMenuScreen.
    /// </summary>
    [UnityTest]
    public IEnumerator GotoCoursesTest()
    {
        var mainMenuScreen = GameObject.FindObjectOfType<MainMenuScreen>();
        mainMenuScreen.GotoCourses();
        yield return new WaitForSeconds(0.2f);

        var courseMenuScreen = GameObject.FindObjectOfType<CourseMenuScreen>();

        Assert.IsNotNull(courseMenuScreen);
    }

    /// <summary>
    /// Test to verify the transition to the ListMinigamesScreen.
    /// </summary>
    [UnityTest]
    public IEnumerator GotoMiniGamesTest()
    {
        var mainMenuScreen = GameObject.FindObjectOfType<MainMenuScreen>();
        mainMenuScreen.GotoMinigames();
        yield return new WaitForSeconds(0.2f);

        var listMinigamesScreen = GameObject.FindObjectOfType<ListMinigamesScreen>();
        Assert.IsNotNull(listMinigamesScreen);
    }

    /// <summary>
    /// Test to verify the behavior when the user data is deleted.
    /// </summary>
    [UnityTest]
    public IEnumerator DeletedUserTest()
    {
        // Remove the user data file
        File.Delete(PersistentDataController.PATH);

        SceneManager.LoadScene("Common/Scenes/MainMenuScreen");
        yield return new WaitForSeconds(0.2f);

        var userCreationScreen = GameObject.FindObjectOfType<UserCreationScreen>();
        Assert.IsNotNull(userCreationScreen);
    }

    /// <summary>
    /// Test to verify the behavior when starting the scene from the Boot scene.
    /// </summary>
    [UnityTest]
    public IEnumerator StartFromBootTest()
    {
        SystemController.GetInstance().previousScene = SystemController.GetSceneIndex("Common/Scenes/Boot");

        SceneManager.LoadScene("Common/Scenes/MainMenuScreen");
        yield return new WaitForSeconds(0.2f);

        var mainMenuScreen = GameObject.FindObjectOfType<MainMenuScreen>();
        Assert.IsTrue(mainMenuScreen.directorEnterFromBoot.state == UnityEngine.Playables.PlayState.Playing);
    }

    /// <summary>
    /// Test to verify the transition to the SettingsScreen.
    /// </summary>
    [UnityTest]
    public IEnumerator GotoSettingsTest()
    {
        var mainMenuScreen = GameObject.FindObjectOfType<MainMenuScreen>();
        mainMenuScreen.GotoSettings();
        yield return new WaitForSeconds(0.2f);

        var settingsScreen = GameObject.FindObjectOfType<SettingsScreen>();
        Assert.IsNotNull(settingsScreen);

        GameObject backbutton = GameObject.Find("ButtonBack");
        Assert.IsNotNull(backbutton);
        backbutton.GetComponent<UnityEngine.UI.Button>().onClick.Invoke();
    }

    /// <summary>
    /// Tests the QuicApplication function of the MainMenuScreen
    /// </summary>
    [UnityTest]
    public IEnumerator QuitAppTest()
    {
        var mainMenuScreen = GameObject.FindObjectOfType<MainMenuScreen>();
        mainMenuScreen.QuitApplication();
        yield return new WaitForSeconds(0.2f);
    }
}