【m】插件上传
This commit is contained in:
394
Packages/com.unity.renderstreaming@3.1.0-exp.9/Tests/Runtime/SignalingTest.cs
vendored
Normal file
394
Packages/com.unity.renderstreaming@3.1.0-exp.9/Tests/Runtime/SignalingTest.cs
vendored
Normal file
@@ -0,0 +1,394 @@
|
||||
using System;
|
||||
using System.Collections;
|
||||
using System.Diagnostics;
|
||||
using System.Threading;
|
||||
using NUnit.Framework;
|
||||
using Unity.RenderStreaming.RuntimeTest.Signaling;
|
||||
using Unity.RenderStreaming.Signaling;
|
||||
using Unity.WebRTC;
|
||||
using UnityEngine;
|
||||
using UnityEngine.TestTools;
|
||||
using Debug = UnityEngine.Debug;
|
||||
|
||||
namespace Unity.RenderStreaming.RuntimeTest
|
||||
{
|
||||
[TestFixture(typeof(WebSocketSignaling))]
|
||||
[TestFixture(typeof(HttpSignaling))]
|
||||
[TestFixture(typeof(MockSignaling))]
|
||||
[UnityPlatform(exclude = new[] { RuntimePlatform.IPhonePlayer })]
|
||||
[ConditionalIgnore(ConditionalIgnore.IL2CPP, "Process.Start does not implement in IL2CPP.")]
|
||||
class SignalingTest : IPrebuildSetup
|
||||
{
|
||||
private readonly Type m_SignalingType;
|
||||
private Process m_ServerProcess;
|
||||
private RTCSessionDescription m_DescOffer;
|
||||
private RTCSessionDescription m_DescAnswer;
|
||||
private RTCIceCandidate m_candidate;
|
||||
|
||||
private SynchronizationContext m_Context;
|
||||
private ISignaling signaling1;
|
||||
private ISignaling signaling2;
|
||||
|
||||
public SignalingTest()
|
||||
{
|
||||
}
|
||||
|
||||
public SignalingTest(Type type)
|
||||
{
|
||||
m_SignalingType = type;
|
||||
}
|
||||
|
||||
public void Setup()
|
||||
{
|
||||
if (m_SignalingType == typeof(MockSignaling))
|
||||
{
|
||||
return;
|
||||
}
|
||||
#if UNITY_EDITOR
|
||||
string dir = System.IO.Directory.GetCurrentDirectory();
|
||||
string fileName = System.IO.Path.Combine(dir, Editor.WebAppDownloader.GetFileName());
|
||||
if (System.IO.File.Exists(fileName) || System.IO.File.Exists(TestUtility.GetWebAppLocationFromEnv()))
|
||||
{
|
||||
// already exists.
|
||||
return;
|
||||
}
|
||||
|
||||
bool downloadRaised = false;
|
||||
Editor.WebAppDownloader.DownloadCurrentVersionWebApp(dir, success => { downloadRaised = true; });
|
||||
TestUtility.Wait(() => downloadRaised, 10000);
|
||||
#endif
|
||||
}
|
||||
|
||||
[OneTimeSetUp]
|
||||
public void OneTimeSetUp()
|
||||
{
|
||||
if (m_SignalingType == typeof(MockSignaling))
|
||||
{
|
||||
MockSignaling.Reset(false);
|
||||
return;
|
||||
}
|
||||
|
||||
m_ServerProcess = new Process();
|
||||
|
||||
string fileName = TestUtility.GetWebAppLocationFromEnv();
|
||||
if (string.IsNullOrEmpty(fileName))
|
||||
{
|
||||
Debug.Log($"webapp file not found in {fileName}");
|
||||
string dir = System.IO.Directory.GetCurrentDirectory();
|
||||
fileName = System.IO.Path.Combine(dir, TestUtility.GetFileName());
|
||||
}
|
||||
|
||||
Assert.IsTrue(System.IO.File.Exists(fileName), $"webapp file not found in {fileName}");
|
||||
ProcessStartInfo startInfo = new ProcessStartInfo
|
||||
{
|
||||
WindowStyle = ProcessWindowStyle.Hidden,
|
||||
FileName = fileName,
|
||||
UseShellExecute = false
|
||||
};
|
||||
|
||||
string arguments = $"-p {TestUtility.PortNumber}";
|
||||
|
||||
if (m_SignalingType == typeof(HttpSignaling))
|
||||
{
|
||||
arguments += " -t http";
|
||||
}
|
||||
|
||||
startInfo.Arguments = arguments;
|
||||
|
||||
m_ServerProcess.StartInfo = startInfo;
|
||||
m_ServerProcess.OutputDataReceived += (sender, e) => { Debug.Log(e.Data); };
|
||||
bool success = m_ServerProcess.Start();
|
||||
Assert.True(success);
|
||||
}
|
||||
|
||||
[OneTimeTearDown]
|
||||
public void OneTimeTearDown()
|
||||
{
|
||||
if (m_SignalingType == typeof(MockSignaling))
|
||||
{
|
||||
return;
|
||||
}
|
||||
m_ServerProcess?.Kill();
|
||||
m_ServerProcess?.WaitForExit();
|
||||
m_ServerProcess?.Dispose();
|
||||
m_ServerProcess = null;
|
||||
}
|
||||
|
||||
ISignaling CreateSignaling(Type type, SynchronizationContext mainThread)
|
||||
{
|
||||
if (type == typeof(WebSocketSignaling))
|
||||
{
|
||||
var settings = new WebSocketSignalingSettings
|
||||
(
|
||||
url: $"ws://localhost:{TestUtility.PortNumber}"
|
||||
);
|
||||
return new WebSocketSignaling(settings, mainThread);
|
||||
}
|
||||
|
||||
if (type == typeof(HttpSignaling))
|
||||
{
|
||||
var settings = new HttpSignalingSettings
|
||||
(
|
||||
url: $"http://localhost:{TestUtility.PortNumber}",
|
||||
interval: 100
|
||||
);
|
||||
return new HttpSignaling(settings, mainThread);
|
||||
}
|
||||
|
||||
if (type == typeof(MockSignaling))
|
||||
{
|
||||
return new MockSignaling();
|
||||
}
|
||||
throw new ArgumentException();
|
||||
}
|
||||
|
||||
[UnitySetUp, Timeout(1000)]
|
||||
public IEnumerator UnitySetUp()
|
||||
{
|
||||
RTCConfiguration config = default;
|
||||
RTCIceCandidate candidate_ = null;
|
||||
config.iceServers = new[] { new RTCIceServer { urls = new[] { "stun:stun.l.google.com:19302" } } };
|
||||
|
||||
var peer1 = new RTCPeerConnection(ref config);
|
||||
var peer2 = new RTCPeerConnection(ref config);
|
||||
peer1.OnIceCandidate = candidate => { candidate_ = candidate; };
|
||||
|
||||
AudioStreamTrack track = new AudioStreamTrack();
|
||||
peer1.AddTrack(track);
|
||||
|
||||
var op1 = peer1.CreateOffer();
|
||||
yield return op1;
|
||||
m_DescOffer = op1.Desc;
|
||||
var op2 = peer1.SetLocalDescription(ref m_DescOffer);
|
||||
yield return op2;
|
||||
var op3 = peer2.SetRemoteDescription(ref m_DescOffer);
|
||||
yield return op3;
|
||||
|
||||
var op4 = peer2.CreateAnswer();
|
||||
yield return op4;
|
||||
m_DescAnswer = op4.Desc;
|
||||
var op5 = peer2.SetLocalDescription(ref m_DescAnswer);
|
||||
yield return op5;
|
||||
var op6 = peer1.SetRemoteDescription(ref m_DescAnswer);
|
||||
yield return op6;
|
||||
|
||||
yield return new WaitUntil(() => candidate_ != null);
|
||||
m_candidate = candidate_;
|
||||
|
||||
track.Dispose();
|
||||
peer1.Close();
|
||||
peer2.Close();
|
||||
|
||||
m_Context = SynchronizationContext.Current;
|
||||
signaling1 = CreateSignaling(m_SignalingType, m_Context);
|
||||
signaling2 = CreateSignaling(m_SignalingType, m_Context);
|
||||
}
|
||||
|
||||
[TearDown]
|
||||
public void TearDown()
|
||||
{
|
||||
signaling1.Stop();
|
||||
signaling2.Stop();
|
||||
m_Context = null;
|
||||
}
|
||||
|
||||
[UnityTest, Timeout(10000)]
|
||||
public IEnumerator OnConnect()
|
||||
{
|
||||
bool startRaised1 = false;
|
||||
string connectionId1 = null;
|
||||
|
||||
signaling1.OnStart += s => { startRaised1 = true; };
|
||||
signaling1.Start();
|
||||
yield return new WaitUntil(() => startRaised1);
|
||||
|
||||
signaling1.OnCreateConnection += (s, connectionId, polite) => { connectionId1 = connectionId; };
|
||||
signaling1.OpenConnection(Guid.NewGuid().ToString());
|
||||
yield return new WaitUntil(() => !string.IsNullOrEmpty(connectionId1));
|
||||
Assert.IsNotEmpty(connectionId1);
|
||||
}
|
||||
|
||||
[UnityTest, Timeout(10000)]
|
||||
public IEnumerator OnOffer()
|
||||
{
|
||||
bool startRaised1 = false;
|
||||
bool startRaised2 = false;
|
||||
bool offerRaised = false;
|
||||
bool raiseOnDestroy1 = false;
|
||||
bool raiseOnDestroy2 = false;
|
||||
string connectionId1 = null;
|
||||
string connectionId2 = null;
|
||||
const string _connectionId = "12345";
|
||||
|
||||
signaling1.OnStart += s => { startRaised1 = true; };
|
||||
signaling2.OnStart += s => { startRaised2 = true; };
|
||||
signaling1.Start();
|
||||
signaling2.Start();
|
||||
yield return new WaitUntil(() => startRaised1 && startRaised2);
|
||||
|
||||
signaling1.OnCreateConnection += (s, connectionId, polite) =>
|
||||
{
|
||||
connectionId1 = connectionId;
|
||||
};
|
||||
signaling1.OnDestroyConnection += (signaling, id) =>
|
||||
{
|
||||
raiseOnDestroy1 = id == connectionId1;
|
||||
};
|
||||
signaling1.OpenConnection(_connectionId);
|
||||
|
||||
signaling2.OnCreateConnection += (s, connectionId, polite) =>
|
||||
{
|
||||
connectionId2 = connectionId;
|
||||
};
|
||||
signaling1.OnDestroyConnection += (signaling, id) =>
|
||||
{
|
||||
raiseOnDestroy2 = id == connectionId2;
|
||||
};
|
||||
signaling2.OpenConnection(_connectionId);
|
||||
yield return new WaitUntil(() =>
|
||||
!string.IsNullOrEmpty(connectionId1) && !string.IsNullOrEmpty(connectionId2));
|
||||
|
||||
Assert.That(connectionId1, Is.EqualTo(_connectionId));
|
||||
Assert.That(connectionId2, Is.EqualTo(_connectionId));
|
||||
|
||||
signaling2.OnOffer += (s, e) => { offerRaised = true; };
|
||||
signaling1.SendOffer(connectionId1, m_DescOffer);
|
||||
yield return new WaitUntil(() => offerRaised);
|
||||
|
||||
signaling1.CloseConnection(connectionId1);
|
||||
|
||||
yield return new WaitUntil(() => raiseOnDestroy1 && raiseOnDestroy2);
|
||||
Assert.That(raiseOnDestroy1, Is.True);
|
||||
Assert.That(raiseOnDestroy2, Is.True);
|
||||
}
|
||||
|
||||
[UnityTest, Timeout(10000)]
|
||||
public IEnumerator OnAnswer()
|
||||
{
|
||||
bool startRaised1 = false;
|
||||
bool startRaised2 = false;
|
||||
bool offerRaised = false;
|
||||
bool answerRaised = false;
|
||||
string connectionId1 = null;
|
||||
string connectionId2 = null;
|
||||
|
||||
signaling1.OnStart += s => { startRaised1 = true; };
|
||||
signaling2.OnStart += s => { startRaised2 = true; };
|
||||
signaling1.Start();
|
||||
signaling2.Start();
|
||||
yield return new WaitUntil(() => startRaised1 && startRaised2);
|
||||
|
||||
signaling1.OnCreateConnection += (s, connectionId, polite) =>
|
||||
{
|
||||
connectionId1 = connectionId;
|
||||
};
|
||||
signaling1.OpenConnection(Guid.NewGuid().ToString());
|
||||
signaling2.OnCreateConnection += (s, connectionId, polite) =>
|
||||
{
|
||||
connectionId2 = connectionId;
|
||||
};
|
||||
signaling2.OpenConnection(Guid.NewGuid().ToString());
|
||||
yield return new WaitUntil(() =>
|
||||
!string.IsNullOrEmpty(connectionId1) && !string.IsNullOrEmpty(connectionId2));
|
||||
|
||||
signaling2.OnOffer += (s, e) => { offerRaised = true; };
|
||||
signaling1.SendOffer(connectionId1, m_DescOffer);
|
||||
yield return new WaitUntil(() => offerRaised);
|
||||
|
||||
signaling1.OnAnswer += (s, e) => { answerRaised = true; };
|
||||
signaling2.SendAnswer(connectionId1, m_DescAnswer);
|
||||
yield return new WaitUntil(() => answerRaised);
|
||||
}
|
||||
|
||||
[UnityTest, Timeout(10000)]
|
||||
public IEnumerator OnCandidate()
|
||||
{
|
||||
bool startRaised1 = false;
|
||||
bool startRaised2 = false;
|
||||
bool offerRaised = false;
|
||||
bool answerRaised = false;
|
||||
bool candidateRaised1 = false;
|
||||
bool candidateRaised2 = false;
|
||||
string connectionId1 = null;
|
||||
string connectionId2 = null;
|
||||
|
||||
signaling1.OnStart += s => { startRaised1 = true; };
|
||||
signaling2.OnStart += s => { startRaised2 = true; };
|
||||
signaling1.Start();
|
||||
signaling2.Start();
|
||||
yield return new WaitUntil(() => startRaised1 && startRaised2);
|
||||
|
||||
signaling1.OnCreateConnection += (s, connectionId, polite) =>
|
||||
{
|
||||
connectionId1 = connectionId;
|
||||
};
|
||||
signaling1.OpenConnection(Guid.NewGuid().ToString());
|
||||
signaling2.OnCreateConnection += (s, connectionId, polite) =>
|
||||
{
|
||||
connectionId2 = connectionId;
|
||||
};
|
||||
signaling2.OpenConnection(Guid.NewGuid().ToString());
|
||||
yield return new WaitUntil(() =>
|
||||
!string.IsNullOrEmpty(connectionId1) && !string.IsNullOrEmpty(connectionId2));
|
||||
|
||||
signaling2.OnOffer += (s, e) => { offerRaised = true; };
|
||||
signaling1.SendOffer(connectionId1, m_DescOffer);
|
||||
yield return new WaitUntil(() => offerRaised);
|
||||
|
||||
signaling1.OnAnswer += (s, e) => { answerRaised = true; };
|
||||
signaling2.SendAnswer(connectionId1, m_DescAnswer);
|
||||
yield return new WaitUntil(() => answerRaised);
|
||||
|
||||
signaling2.OnIceCandidate += (s, e) => { candidateRaised1 = true; };
|
||||
signaling1.SendCandidate(connectionId1, m_candidate);
|
||||
yield return new WaitUntil(() => candidateRaised1);
|
||||
|
||||
signaling1.OnIceCandidate += (s, e) => { candidateRaised2 = true; };
|
||||
signaling2.SendCandidate(connectionId1, m_candidate);
|
||||
yield return new WaitUntil(() => candidateRaised2);
|
||||
}
|
||||
|
||||
[UnityTest, Timeout(10000)]
|
||||
public IEnumerator OnStop()
|
||||
{
|
||||
bool startRaised1 = false;
|
||||
bool startRaised2 = false;
|
||||
bool offerRaised = false;
|
||||
bool answerRaised = false;
|
||||
bool connectionClosed2 = false;
|
||||
string connectionId1 = null;
|
||||
string connectionId2 = null;
|
||||
|
||||
signaling1.OnStart += s => { startRaised1 = true; };
|
||||
signaling2.OnStart += s => { startRaised2 = true; };
|
||||
signaling1.Start();
|
||||
signaling2.Start();
|
||||
yield return new WaitUntil(() => startRaised1 && startRaised2);
|
||||
|
||||
signaling1.OnCreateConnection += (s, connectionId, polite) =>
|
||||
{
|
||||
connectionId1 = connectionId;
|
||||
};
|
||||
signaling1.OpenConnection(Guid.NewGuid().ToString());
|
||||
signaling2.OnCreateConnection += (s, connectionId, polite) =>
|
||||
{
|
||||
connectionId2 = connectionId;
|
||||
};
|
||||
signaling2.OpenConnection(Guid.NewGuid().ToString());
|
||||
yield return new WaitUntil(() =>
|
||||
!string.IsNullOrEmpty(connectionId1) && !string.IsNullOrEmpty(connectionId2));
|
||||
|
||||
signaling2.OnOffer += (s, e) => { offerRaised = true; };
|
||||
signaling1.SendOffer(connectionId1, m_DescOffer);
|
||||
yield return new WaitUntil(() => offerRaised);
|
||||
|
||||
signaling1.OnAnswer += (s, e) => { answerRaised = true; };
|
||||
signaling2.SendAnswer(connectionId1, m_DescAnswer);
|
||||
yield return new WaitUntil(() => answerRaised);
|
||||
|
||||
signaling2.OnDestroyConnection += (s, e) => { connectionClosed2 = true; };
|
||||
signaling1.Stop();
|
||||
yield return new WaitUntil(() => connectionClosed2);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user