shoc-backend/SeaHavenIndustries.Tests/WorkOrderBoardUpdateServiceTests.cs

using Data.SeaHavenIndustries;
using Data.SeaHavenIndustries.Enums;
using Microsoft.EntityFrameworkCore;
using SeaHaven.DataServices.Implementation;
using SeaHaven.Services.DTOs;
using SeaHaven.Services.Exceptions;
using SeaHaven.Services.Helpers;
using SeaHaven.Services.Implementation;

namespace SeaHavenIndustries.Tests;

public class WorkOrderBoardUpdateServiceTests
{
    private static (ApplicationDbContext Context, WorkOrderBoardUpdateService Service) CreateSut()
    {
        var options = new DbContextOptionsBuilder<ApplicationDbContext>()
            .UseInMemoryDatabase(Guid.NewGuid().ToString())
            .Options;
        var context = new ApplicationDbContext(options);
        var boardData = new WorkOrderBoardDataService(context);
        var fieldLocks = new WorkOrderFieldLockService(context);
        var audit = new WorkOrderAuditService(context, fieldLocks);
        var service = new WorkOrderBoardUpdateService(context, boardData, audit);
        return (context, service);
    }

    private static string ToVersion(WorkOrder workOrder)
        => Convert.ToBase64String(workOrder.RowVersion ?? new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 });

    private static string ToVersion(Dispatch dispatch)
        => Convert.ToBase64String(dispatch.RowVersion ?? new byte[] { 1, 0, 0, 0, 0, 0, 0, 2 });

    [Fact]
    public async Task PatchField_AutoSchedulesIncompleteToScheduled()
    {
        var (context, service) = CreateSut();
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Incomplete,
            AssignTo = "user-a",
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        var result = await service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
        {
            Field = WorkOrderBoardFieldNames.ScheduledDate,
            Value = "2026-06-25",
            WorkOrderVersion = ToVersion(wo)
        }, "actor-1");

        Assert.Equal(LifecycleStatus.Scheduled, result.LifecycleStatus);
        Assert.Equal(2, await context.WorkOrderAuditLogs.CountAsync());
    }

    [Fact]
    public async Task PatchField_IncrementsRescheduleCount()
    {
        var (context, service) = CreateSut();
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Scheduled,
            ScheduledDate = new DateTime(2026, 6, 20),
            RescheduleCount = 1,
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        var result = await service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
        {
            Field = WorkOrderBoardFieldNames.ScheduledDate,
            Value = "2026-06-25",
            WorkOrderVersion = ToVersion(wo)
        }, "actor-1");

        Assert.Equal(2, result.RescheduleCount);
    }

    [Fact]
    public async Task PatchField_BlocksStatusChangeWhenPastDue()
    {
        var (context, service) = CreateSut();
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Scheduled,
            ScheduledDate = DateTime.UtcNow.Date.AddDays(-3),
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        var ex = await Assert.ThrowsAsync<WorkOrderBoardValidationException>(() =>
            service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
            {
                Field = WorkOrderBoardFieldNames.LifecycleStatus,
                Value = "Complete",
                WorkOrderVersion = ToVersion(wo)
            }, "actor-1"));

        Assert.Equal("PastDueStatusBlocked", ex.Code);
    }

    [Fact]
    public async Task PatchField_BlocksEditWhenCanceled()
    {
        var (context, service) = CreateSut();
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Canceled,
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        var ex = await Assert.ThrowsAsync<WorkOrderBoardValidationException>(() =>
            service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
            {
                Field = WorkOrderBoardFieldNames.SiteCode,
                Value = "BK5",
                WorkOrderVersion = ToVersion(wo)
            }, "actor-1"));

        Assert.Equal("CanceledReadOnly", ex.Code);
    }

    [Fact]
    public async Task PatchField_VendorCreatesPrimaryDispatch()
    {
        var (context, service) = CreateSut();
        context.Vendors.Add(new Vendor { Id = 5, CompanyName = "Vendor Co" });
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Incomplete,
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        var result = await service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
        {
            Field = WorkOrderBoardFieldNames.VendorId,
            Value = "5",
            WorkOrderVersion = ToVersion(wo)
        }, "actor-1");

        Assert.NotNull(result.PrimaryDispatchId);
        Assert.Equal(5, result.VendorId);
        Assert.Equal(1, await context.Dispatches.CountAsync());
    }

    [Fact]
    public async Task PatchField_CreatesFieldLockAndAuditPerField()
    {
        var (context, service) = CreateSut();
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Scheduled,
            SiteCode = "OLD",
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        await service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
        {
            Field = WorkOrderBoardFieldNames.SiteCode,
            Value = "BK5",
            WorkOrderVersion = ToVersion(wo)
        }, "actor-1");

        Assert.Single(await context.WorkOrderAuditLogs.Where(a => a.Action == AuditActionType.FieldChanged.ToString()).ToListAsync());
        Assert.True(await context.WorkOrderFieldLocks.AnyAsync(l => l.FieldName == "SiteCode"));
    }

    [Fact]
    public async Task PatchField_SecondEditSameField_DoesNotDuplicateFieldLock()
    {
        var (context, service) = CreateSut();
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Scheduled,
            SiteCode = "OLD",
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        await service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
        {
            Field = WorkOrderBoardFieldNames.SiteCode,
            Value = "BK5",
            WorkOrderVersion = ToVersion(wo)
        }, "actor-1");

        var updated = await context.workOrders.FindAsync(1);
        var secondResult = await service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
        {
            Field = WorkOrderBoardFieldNames.SiteCode,
            Value = "BK6",
            WorkOrderVersion = ToVersion(updated!)
        }, "actor-1");

        Assert.Equal("BK6", secondResult.SiteCode);
        Assert.Single(await context.WorkOrderFieldLocks.Where(l => l.FieldName == "SiteCode").ToListAsync());
        Assert.Equal(2, await context.WorkOrderAuditLogs.Where(a => a.FieldName == "SiteCode").CountAsync());
    }

    [Fact]
    public async Task PatchField_NotFound_ThrowsNotFoundCode()
    {
        var (_, service) = CreateSut();

        var ex = await Assert.ThrowsAsync<WorkOrderBoardValidationException>(() =>
            service.PatchFieldAsync(999, new WorkOrderBoardPatchRequestDto
            {
                Field = WorkOrderBoardFieldNames.SiteCode,
                Value = "BK5",
                WorkOrderVersion = Convert.ToBase64String(new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 })
            }, "actor-1"));

        Assert.Equal("NotFound", ex.Code);
    }

    [Fact]
    public async Task PatchField_NormalizesWoNumber()
    {
        var (context, service) = CreateSut();
        var wo = new WorkOrder
        {
            Id = 1,
            InternalWONumber = "00000009999",
            LifecycleStatus = LifecycleStatus.Incomplete,
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        var result = await service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
        {
            Field = WorkOrderBoardFieldNames.WoNumber,
            Value = "12345",
            WorkOrderVersion = ToVersion(wo)
        }, "actor-1");

        Assert.Equal("00000012345", result.WoNumber);
    }

    [Fact]
    public async Task PatchField_RejectsDuplicateWoNumber()
    {
        var (context, service) = CreateSut();
        context.workOrders.AddRange(
            new WorkOrder { Id = 1, InternalWONumber = "00000011111", RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 } },
            new WorkOrder { Id = 2, InternalWONumber = "00000022222", RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 2 } });
        await context.SaveChangesAsync();

        var wo = await context.workOrders.FindAsync(1);
        var ex = await Assert.ThrowsAsync<WorkOrderBoardValidationException>(() =>
            service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
            {
                Field = WorkOrderBoardFieldNames.WoNumber,
                Value = "22222",
                WorkOrderVersion = ToVersion(wo!)
            }, "actor-1"));

        Assert.Equal("DuplicateWoNumber", ex.Code);
    }

    [Fact]
    public async Task PatchField_WeekOnly_DoesNotAutoScheduleOnAssignTo()
    {
        var (context, service) = CreateSut();
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Incomplete,
            ScheduledDate = new DateTime(2026, 6, 25),
            ScheduleWeekOnly = true,
            TargetWeek = new DateOnly(2026, 6, 22),
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        var result = await service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
        {
            Field = WorkOrderBoardFieldNames.AssignTo,
            Value = "user-a",
            WorkOrderVersion = ToVersion(wo)
        }, "actor-1");

        Assert.Equal(LifecycleStatus.Incomplete, result.LifecycleStatus);
        Assert.Equal("user-a", result.DispatcherId);
    }

    [Fact]
    public async Task PatchField_ApptTime_UpdatesScheduledStart()
    {
        var (context, service) = CreateSut();
        var dispatch = new Dispatch
        {
            Id = 10,
            WorkOrderId = 1,
            VendorId = 1,
            ScheduledDate = new DateTime(2026, 6, 25),
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 2 }
        };
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Scheduled,
            PrimaryDispatchId = 10,
            ScheduledDate = new DateTime(2026, 6, 25),
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.Vendors.Add(new Vendor { Id = 1, CompanyName = "Vendor A" });
        context.Dispatches.Add(dispatch);
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        var result = await service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
        {
            Field = WorkOrderBoardFieldNames.ApptTime,
            Value = "09:00-10:00",
            WorkOrderVersion = ToVersion(wo),
            DispatchVersion = ToVersion(dispatch),
            PrimaryDispatchId = 10
        }, "actor-1");

        Assert.Contains("09:00", result.ApptTime);
        var reloaded = await context.workOrders.FindAsync(1);
        Assert.Equal(new DateTime(2026, 6, 25, 9, 0, 0), reloaded!.ScheduledStart);
        Assert.Equal(new DateTime(2026, 6, 25, 10, 0, 0), reloaded.ScheduledEnd);
    }

    [Fact]
    public async Task PatchField_ApptTime_DashPrefixedToken_TreatedAsSingleStart()
    {
        var (context, service) = CreateSut();
        var dispatch = new Dispatch
        {
            Id = 10,
            WorkOrderId = 1,
            VendorId = 1,
            ScheduledDate = new DateTime(2026, 6, 25),
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 2 }
        };
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Scheduled,
            PrimaryDispatchId = 10,
            ScheduledDate = new DateTime(2026, 6, 25),
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.Vendors.Add(new Vendor { Id = 1, CompanyName = "Vendor A" });
        context.Dispatches.Add(dispatch);
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        await service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
        {
            Field = WorkOrderBoardFieldNames.ApptTime,
            Value = "-30",
            WorkOrderVersion = ToVersion(wo),
            DispatchVersion = ToVersion(dispatch),
            PrimaryDispatchId = 10
        }, "actor-1");

        var reloaded = await context.workOrders.FindAsync(1);
        Assert.NotNull(reloaded!.ScheduledStart);
        Assert.Equal(new DateTime(2026, 6, 25), reloaded.ScheduledEnd);
        Assert.Equal(new DateTime(2026, 6, 25).Add(TimeSpan.FromDays(30)), reloaded.ScheduledStart);
    }

    [Fact]
    public async Task PatchField_DispatchFieldWithoutVersion_ThrowsDispatchVersionRequired()
    {
        var (context, service) = CreateSut();
        var dispatch = new Dispatch
        {
            Id = 10,
            WorkOrderId = 1,
            VendorId = 1,
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 2 }
        };
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Scheduled,
            PrimaryDispatchId = 10,
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.Vendors.Add(new Vendor { Id = 1, CompanyName = "Vendor A" });
        context.Dispatches.Add(dispatch);
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        var ex = await Assert.ThrowsAsync<WorkOrderBoardValidationException>(() =>
            service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
            {
                Field = WorkOrderBoardFieldNames.ApptDate,
                Value = "2026-06-26",
                WorkOrderVersion = ToVersion(wo),
                PrimaryDispatchId = 10
            }, "actor-1"));

        Assert.Equal("DispatchVersionRequired", ex.Code);
    }

    [Fact]
    public async Task PatchField_VendorIdNotFound_Throws()
    {
        var (context, service) = CreateSut();
        var dispatch = new Dispatch
        {
            Id = 10,
            WorkOrderId = 1,
            VendorId = 1,
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 2 }
        };
        var wo = new WorkOrder
        {
            Id = 1,
            LifecycleStatus = LifecycleStatus.Scheduled,
            PrimaryDispatchId = 10,
            RowVersion = new byte[] { 1, 0, 0, 0, 0, 0, 0, 1 }
        };
        context.Vendors.Add(new Vendor { Id = 1, CompanyName = "Vendor A" });
        context.Dispatches.Add(dispatch);
        context.workOrders.Add(wo);
        await context.SaveChangesAsync();

        var ex = await Assert.ThrowsAsync<WorkOrderBoardValidationException>(() =>
            service.PatchFieldAsync(1, new WorkOrderBoardPatchRequestDto
            {
                Field = WorkOrderBoardFieldNames.VendorId,
                Value = "999",
                WorkOrderVersion = ToVersion(wo),
                DispatchVersion = ToVersion(dispatch),
                PrimaryDispatchId = 10
            }, "actor-1"));

        Assert.Equal("VendorNotFound", ex.Code);
    }
}