Files
tbd-station-14/Content.Tests/ContentUnitTest.cs
2022-05-13 17:59:03 +10:00

53 lines
1.3 KiB
C#

using System.Collections.Generic;
using System.Reflection;
using Content.Client.IoC;
using Content.Server.IoC;
using Content.Shared.IoC;
using Robust.Shared.Analyzers;
using Robust.UnitTesting;
using EntryPoint = Content.Server.Entry.EntryPoint;
namespace Content.Tests
{
[Virtual]
public class ContentUnitTest : RobustUnitTest
{
protected override void OverrideIoC()
{
base.OverrideIoC();
SharedContentIoC.Register();
if (Project == UnitTestProject.Server)
{
ServerContentIoC.Register();
}
else if (Project == UnitTestProject.Client)
{
ClientContentIoC.Register();
}
}
protected override Assembly[] GetContentAssemblies()
{
var l = new List<Assembly>
{
typeof(Content.Shared.Entry.EntryPoint).Assembly
};
if (Project == UnitTestProject.Server)
{
l.Add(typeof(EntryPoint).Assembly);
}
else if (Project == UnitTestProject.Client)
{
l.Add(typeof(Content.Client.Entry.EntryPoint).Assembly);
}
l.Add(typeof(ContentUnitTest).Assembly);
return l.ToArray();
}
}
}