| | | 1 | | using System; |
| | | 2 | | using System.Reflection; |
| | | 3 | | using NUnit.Framework; |
| | | 4 | | using UnityEngine; |
| | | 5 | | |
| | | 6 | | public abstract class AgentTestBase : TestBase { |
| | | 7 | | protected SimManager _simManager; |
| | | 8 | | |
| | | 9 | | [SetUp] |
| | 0 | 10 | | public virtual void Setup() { |
| | | 11 | | // Initialize the simulation manager. |
| | 0 | 12 | | var simManagerGameObject = new GameObject("SimManager"); |
| | 0 | 13 | | _simManager = simManagerGameObject.AddComponent<SimManager>(); |
| | 0 | 14 | | _simManager.SimulationConfig = new Configs.SimulationConfig(); |
| | 0 | 15 | | SimManager.Instance = _simManager; |
| | 0 | 16 | | } |
| | | 17 | | |
| | | 18 | | [TearDown] |
| | 0 | 19 | | public virtual void Teardown() { |
| | | 20 | | // Clean up the simulation manager. |
| | 0 | 21 | | if (_simManager != null) { |
| | 0 | 22 | | GameObject.DestroyImmediate(_simManager.gameObject); |
| | 0 | 23 | | } |
| | 0 | 24 | | } |
| | | 25 | | |
| | 0 | 26 | | protected Interceptor CreateTestInterceptor(Configs.AgentConfig config) { |
| | 0 | 27 | | Interceptor interceptor = SimManager.Instance.CreateInterceptor( |
| | | 28 | | config, new Simulation.State() { Position = Coordinates3.ToProto(Vector3.zero), |
| | | 29 | | Velocity = Coordinates3.ToProto(Vector3.zero) }); |
| | 0 | 30 | | InvokePrivateMethod(interceptor, "Start"); |
| | 0 | 31 | | InvokePrivateMethod(interceptor.gameObject.GetComponent<Sensor>(), "Start"); |
| | 0 | 32 | | return interceptor; |
| | 0 | 33 | | } |
| | | 34 | | |
| | 0 | 35 | | protected Threat CreateTestThreat(Configs.AgentConfig config) { |
| | 0 | 36 | | Threat threat = SimManager.Instance.CreateThreat(config); |
| | 0 | 37 | | InvokePrivateMethod(threat, "Start"); |
| | 0 | 38 | | InvokePrivateMethod(threat.gameObject.GetComponent<Sensor>(), "Start"); |
| | 0 | 39 | | return threat; |
| | 0 | 40 | | } |
| | | 41 | | } |