Initial commit
This commit is contained in:
@@ -0,0 +1,102 @@
|
||||
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();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user