103 lines
4.1 KiB
C#
103 lines
4.1 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using CounterDrone.Core;
|
|
using CounterDrone.Core.Models;
|
|
using SQLite;
|
|
|
|
namespace CounterDrone.Core.Tests
|
|
{
|
|
/// <summary>测试辅助：从 seeded 数据库读取（与生产环境同源）</summary>
|
|
internal static class TestData
|
|
{
|
|
private static readonly string _tempDir;
|
|
private static readonly SQLiteConnection _db;
|
|
|
|
static TestData()
|
|
{
|
|
_tempDir = Path.Combine(Path.GetTempPath(), $"cd_td_{Guid.NewGuid():N}");
|
|
var paths = new TestPathProvider(_tempDir);
|
|
var dbm = new DatabaseManager(paths);
|
|
_db = dbm.OpenMainDb();
|
|
}
|
|
|
|
private static List<AmmunitionSpec>? _ammo;
|
|
public static List<AmmunitionSpec> Ammo =>
|
|
_ammo ??= new List<AmmunitionSpec>(_db.Table<AmmunitionSpec>());
|
|
|
|
private static DroneSpec? _droneSpec;
|
|
public static DroneSpec DefaultDroneSpec => _droneSpec ??= _db.Table<DroneSpec>().First(d => d.Id == "shahed");
|
|
|
|
private static FireUnitSpec? _groundSpec;
|
|
public static FireUnitSpec GroundSpec => _groundSpec ??= _db.Table<FireUnitSpec>().First(f => f.Id == "ground-light");
|
|
|
|
public static ScenarioDrone CreateScenarioDrone(string waveId = "w1", int quantity = 1)
|
|
=> new() { DroneSpecId = DefaultDroneSpec.Id, WaveId = waveId, Quantity = quantity };
|
|
|
|
public static ScenarioDrone CreateScenarioDroneWithId(string scenarioId, string waveId = "w1", int quantity = 1)
|
|
=> new() { ScenarioId = scenarioId, DroneSpecId = DefaultDroneSpec.Id, WaveId = waveId, Quantity = quantity };
|
|
|
|
public static ScenarioUnit CreateScenarioUnit(double px, double py, double pz,
|
|
int ammoType = 0, int? munitionCount = null)
|
|
=> new() { FireUnitSpecId = "ground-light", PositionX = px, PositionY = py, PositionZ = pz,
|
|
AerosolType = ammoType, MunitionCount = munitionCount ?? 8, Quantity = 1 };
|
|
|
|
private static FireUnitSpec? _defaultFuSpec;
|
|
public static FireUnitSpec DefaultFuSpec => _defaultFuSpec ??= new FireUnitSpec
|
|
{
|
|
Id = "test-fu", PlatformType = (int)PlatformType.GroundBased,
|
|
GunCount = 1, ChannelsPerGun = 8, ChannelInterval = 1,
|
|
MuzzleVelocity = 800, Cooldown = 5, AmmoChangeTime = 30,
|
|
RadarRange = 5000,
|
|
};
|
|
|
|
private static SensorSpec? _sensorSpec;
|
|
public static SensorSpec DefaultSensorSpec => _sensorSpec ??= new SensorSpec
|
|
{
|
|
Id = "test-sensor", RadarRange = 3000, Accuracy = 50,
|
|
};
|
|
|
|
public static ScenarioUnit CreateDetectionUnit(double px, double py, double pz)
|
|
=> new()
|
|
{
|
|
SensorSpecId = "radar-mr",
|
|
EquipmentRole = (int)EquipmentRole.Detection,
|
|
PositionX = px, PositionY = py, PositionZ = pz, Quantity = 1
|
|
};
|
|
|
|
public static Dictionary<string, DroneSpec> EmptyDroneSpecs => new();
|
|
public static Dictionary<string, FireUnitSpec> EmptyFuSpecs => new();
|
|
public static Dictionary<string, SensorSpec> EmptySensorSpecs => new();
|
|
|
|
private static DefaultData? _all;
|
|
public static DefaultData All
|
|
{
|
|
get
|
|
{
|
|
if (_all == null)
|
|
{
|
|
var dir = TestPathProvider.FindProjectDataDirectory()!;
|
|
_all = DefaultData.Load(new DirPathProvider(dir));
|
|
}
|
|
return _all;
|
|
}
|
|
}
|
|
|
|
private class DirPathProvider : IPathProvider
|
|
{
|
|
private readonly string _dir;
|
|
public DirPathProvider(string dir) => _dir = dir;
|
|
public string GetDataRoot() => _dir;
|
|
public string GetMainDbPath() => throw new NotImplementedException();
|
|
public string GetFramesDir() => throw new NotImplementedException();
|
|
public string GetModelsDir() => throw new NotImplementedException();
|
|
}
|
|
|
|
public static void Cleanup()
|
|
{
|
|
_db?.Close();
|
|
if (Directory.Exists(_tempDir)) Directory.Delete(_tempDir, true);
|
|
}
|
|
}
|
|
}
|