space-wizards/RobustToolbox
1
0
Fork 0
mirror of https://github.com/space-wizards/RobustToolbox.git synced 2026-02-15 03:30:53 +01:00
Code Issues Packages Projects Releases Wiki Activity

Assert that messages have been registerd in integration tests.

Browse Source
This commit is contained in:
Pieter-Jan Briers
2020-07-06 21:23:56 +02:00
parent e15b5ad3d1
commit 9e5374ac86
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
Robust.UnitTesting/RobustIntegrationTest.NetManager.cs
View File

@@ -37,6 +37,7 @@ namespace Robust.UnitTesting
public IReadOnlyDictionary<Type, long> MessageBandwidthUsage { get; } = new Dictionary<Type, long>();
private readonly Dictionary<Type, ProcessMessage> _callbacks = new Dictionary<Type, ProcessMessage>();
private readonly HashSet<Type> _registeredMessages = new HashSet<Type>();
/// <summary>
/// The channel we will connect to when <see cref="ClientConnect"/> is called.
@@ -227,12 +228,18 @@ namespace Robust.UnitTesting
public void RegisterNetMessage<T>(string name, ProcessMessage<T>? rxCallback = null) where T : NetMessage
{
_registeredMessages.Add(typeof(T));
if (rxCallback != null)
_callbacks.Add(typeof(T), msg => rxCallback((T) msg));
}
public T CreateNetMessage<T>() where T : NetMessage
{
if (!_registeredMessages.Contains(typeof(T)))
{
throw new ArgumentException("Net message type is not registered.");
}
return (T) Activator.CreateInstance(typeof(T), (INetChannel?) null)!;
}