103 lines
3.1 KiB
C#
103 lines
3.1 KiB
C#
using RobotNet10.FleetManager.Data;
|
|
using RobotNet10.FleetManager.Shared.Enums;
|
|
|
|
namespace RobotNet10.RobotManager.Test.Helpers;
|
|
|
|
/// <summary>
|
|
/// Helper methods for creating test data
|
|
/// </summary>
|
|
public static class TestHelpers
|
|
{
|
|
/// <summary>
|
|
/// Creates a test RobotModel entity
|
|
/// </summary>
|
|
public static RobotModel CreateTestRobotModel(
|
|
Guid? id = null,
|
|
string? modelName = null,
|
|
double length = 1.0,
|
|
double width = 0.5,
|
|
int imageWidth = 800,
|
|
int imageHeight = 600,
|
|
double navigationPointX = 0.0,
|
|
double navigationPointY = 0.0,
|
|
NavigationType navigationType = NavigationType.Differential)
|
|
{
|
|
return new RobotModel
|
|
{
|
|
Id = id ?? Guid.NewGuid(),
|
|
ModelName = modelName ?? "TestModel",
|
|
Length = length,
|
|
Width = width,
|
|
ImageWidth = imageWidth,
|
|
ImageHeight = imageHeight,
|
|
NavigationPointX = navigationPointX,
|
|
NavigationPointY = navigationPointY,
|
|
NavigationType = navigationType,
|
|
CreatedDate = DateTime.UtcNow
|
|
};
|
|
}
|
|
|
|
/// <summary>
|
|
/// Creates a test Robot entity
|
|
/// </summary>
|
|
public static Robot CreateTestRobot(
|
|
Guid? id = null,
|
|
string? robotId = null,
|
|
string? name = null,
|
|
Guid? modelId = null,
|
|
Guid? mapId = null)
|
|
{
|
|
return new Robot
|
|
{
|
|
Id = id ?? Guid.NewGuid(),
|
|
RobotId = robotId ?? "ROBOT-001",
|
|
Name = name ?? "Test Robot",
|
|
ModelId = modelId ?? Guid.NewGuid(),
|
|
MapId = mapId,
|
|
CreatedDate = DateTime.UtcNow
|
|
};
|
|
}
|
|
|
|
/// <summary>
|
|
/// Seeds test data into the database context
|
|
/// </summary>
|
|
public static void SeedTestData(ApplicationDbContext context)
|
|
{
|
|
var model1 = CreateTestRobotModel(
|
|
id: Guid.Parse("11111111-1111-1111-1111-111111111111"),
|
|
modelName: "AMR-T800",
|
|
length: 1.2,
|
|
width: 0.8);
|
|
|
|
var model2 = CreateTestRobotModel(
|
|
id: Guid.Parse("22222222-2222-2222-2222-222222222222"),
|
|
modelName: "AMR-F500",
|
|
length: 1.0,
|
|
width: 0.6,
|
|
navigationType: NavigationType.Forklift);
|
|
|
|
context.RobotModels.AddRange(model1, model2);
|
|
|
|
var robot1 = CreateTestRobot(
|
|
id: Guid.Parse("33333333-3333-3333-3333-333333333333"),
|
|
robotId: "ROBOT-001",
|
|
name: "Robot 1",
|
|
modelId: model1.Id);
|
|
|
|
var robot2 = CreateTestRobot(
|
|
id: Guid.Parse("44444444-4444-4444-4444-444444444444"),
|
|
robotId: "ROBOT-002",
|
|
name: "Robot 2",
|
|
modelId: model1.Id);
|
|
|
|
var robot3 = CreateTestRobot(
|
|
id: Guid.Parse("55555555-5555-5555-5555-555555555555"),
|
|
robotId: "ROBOT-003",
|
|
name: "Robot 3",
|
|
modelId: model2.Id);
|
|
|
|
context.Robots.AddRange(robot1, robot2, robot3);
|
|
context.SaveChanges();
|
|
}
|
|
}
|