public void QueryVipRegions()
{
List <DensityDevice> devices = new List <DensityDevice>();
int deviceCount = 1;
int channelCount = 1;
int regionCount = 12;
HashSet <string> vips = new HashSet <string>();
//随机创建重点区域
Random random = new Random();
using (IServiceScope serviceScope = TestInit.ServiceProvider.CreateScope())
{
using (DensityContext context = serviceScope.ServiceProvider.GetRequiredService <DensityContext>())
{
context.Database.EnsureDeleted();
context.Database.EnsureCreated();
int deviceId = 1;
int crossingId = 1;
int regionId = 1;
int channelId = 1;
for (int i = 0; i < deviceCount; ++i)
{
DensityDevice densityDevice = new DensityDevice
{
DeviceId = deviceId++,
Ip = "192.168.200.204",
Port = 18000 + i
};
densityDevice.DeviceName = "设备" + densityDevice.Port;
densityDevice.DensityDevice_DensityChannels = new List <DensityDevice_DensityChannel>();
for (int j = 0; j < channelCount; ++j)
{
RoadCrossing roadCrossing = new RoadCrossing
{
CrossingId = crossingId,
CrossingName = "路口" + crossingId
};
DensityChannel channel = new DensityChannel()
{
ChannelId = channelId.ToString(),
ChannelName = $"通道 {densityDevice.DeviceId} {j+1}",
ChannelIndex = j + 1,
CrossingId = crossingId,
Regions = new List <TrafficRegion>(),
RoadCrossing = roadCrossing
};
DensityDevice_DensityChannel relation = new DensityDevice_DensityChannel
{
ChannelId = channel.ChannelId,
DeviceId = densityDevice.DeviceId,
Channel = channel
};
channelId++;
crossingId++;
densityDevice.DensityDevice_DensityChannels.Add(relation);
for (int k = 0; k < regionCount; ++k)
{
int value = random.Next(1, 2);
TrafficRegion region = new TrafficRegion
{
ChannelId = channel.ChannelId,
Channel = channel,
Region = "1",
RegionIndex = k + 1,
RegionName = "区域" + regionId++,
IsVip = value == 1
};
if (value == 1)
{
vips.Add(region.DataId);
}
channel.Regions.Add(region);
}
}
context.Devices.Add(densityDevice);
devices.Add(densityDevice);
}
context.SaveChanges();
}
DensityDbSimulator.CreateData(TestInit.ServiceProvider, devices, DataCreateMode.Fixed, DateTime.Today, DateTime.Today);
TestInit.RefreshDensityCache(devices);
DensitiesManager manager = TestInit.ServiceProvider.GetRequiredService <DensitiesManager>();
var v = manager.QueryVipRegions();
foreach (TrafficDensity density in v)
{
Assert.IsTrue(vips.Contains(density.DataId));
}
}
}
public List <TrafficDensity> QueryVipRegions()
{
return(_manager.QueryVipRegions());
}
