GHZMRNavigatorProXR/com.unity.renderstreaming/Tests/Runtime/RenderStreamingTest.cs

using System.Linq;
using NUnit.Framework;
using Unity.RenderStreaming.RuntimeTest.Signaling;
using Unity.RenderStreaming.Signaling;
using UnityEngine;
using UnityEngine.TestTools;

#if UNITY_EDITOR
using UnityEditor;
#endif

namespace Unity.RenderStreaming.RuntimeTest
{
    class RenderStreamingTest : IPrebuildSetup, IPostBuildCleanup
    {
        private RenderStreamingSettings temp;

        [SetUp]
        public void SetUpTest()
        {
            temp = RenderStreaming.Settings;
        }

        [TearDown]
        public void TearDown()
        {
            if (temp != null)
            {
                RenderStreaming.Settings = temp;
            }
        }

        void IPrebuildSetup.Setup()
        {
#if UNITY_EDITOR
            var defaultSettings = RenderStreaming.Settings;
            RenderStreaming.Settings =
                AssetDatabase.LoadAssetAtPath<RenderStreamingSettings>(RenderStreaming.DefaultRenderStreamingSettingsPath);
            if (defaultSettings != null)
            {
                EditorBuildSettings.AddConfigObject(RenderStreaming.EditorBuildSettingsConfigKey, defaultSettings, true);
            }
#endif
        }

        void IPostBuildCleanup.Cleanup()
        {
#if UNITY_EDITOR
            if (EditorBuildSettings.TryGetConfigObject(RenderStreaming.EditorBuildSettingsConfigKey, out RenderStreamingSettings settingsAsset))
            {
                RenderStreaming.Settings = settingsAsset;
            }
            else
            {
                RenderStreaming.Settings =
                    AssetDatabase.LoadAssetAtPath<RenderStreamingSettings>(RenderStreaming.DefaultRenderStreamingSettingsPath);
            }
#endif
        }

        [Test]
        public void Settings()
        {
            Assert.That(() => RenderStreaming.Settings = null, Throws.ArgumentNullException);

            var settings = ScriptableObject.CreateInstance<RenderStreamingSettings>();
            settings.signalingSettings = new MockSignalingSettings();

            RenderStreaming.Settings = settings;
            Assert.That(RenderStreaming.Settings.automaticStreaming, Is.EqualTo(settings.automaticStreaming));
            Assert.That(RenderStreaming.Settings.signalingSettings, Is.EqualTo(settings.signalingSettings));

            Object.DestroyImmediate(settings);
        }

        [Test]
        public void AutomaticStreaming()
        {
            var settings = ScriptableObject.CreateInstance<RenderStreamingSettings>();
            settings.automaticStreaming = false;
            settings.signalingSettings = new WebSocketSignalingSettings();
            RenderStreaming.Settings = settings;

            RenderStreaming.AutomaticStreaming = true;
            var automaticStreaming = Object.FindObjectOfType<AutomaticStreaming>();
            Assert.That(automaticStreaming, Is.Not.Null);

            RenderStreaming.AutomaticStreaming = false;
            automaticStreaming = Object.FindObjectOfType<AutomaticStreaming>();
            Assert.That(automaticStreaming, Is.Null);

            Object.DestroyImmediate(settings);
        }
    }
}