using Moq;
using ZB.MOM.WW.ScadaBridge.Commons.Entities.Templates;
using ZB.MOM.WW.ScadaBridge.Commons.Interfaces.Repositories;
using ZB.MOM.WW.ScadaBridge.Commons.Types.Templates;

namespace ZB.MOM.WW.ScadaBridge.TemplateEngine.Tests;

/// <summary>
/// Gitea #66: member add/update now asks the repository for a template's analysis closure
/// rather than the whole analysis graph. Test classes keep seeding the whole graph through
/// <c>GetAllTemplatesForAnalysisAsync</c>; this wires <c>GetTemplateAnalysisClosureAsync</c>
/// to serve the real closure walk over that seeded graph, so every existing scenario runs the
/// service against the reduced set — the same set the production repository returns.
/// </summary>
internal static class AnalysisClosureMock
{
    /// <summary>Wires the closure method to filter whatever the full-graph setup returns.</summary>
    /// <param name="repoMock">The repository mock.</param>
    /// <param name="closureRequests">Optional sink recording each requested closure root.</param>
    public static void ServeClosureFromFullGraph(
        this Mock<ITemplateEngineRepository> repoMock,
        List<int>? closureRequests = null)
    {
        repoMock
            .Setup(r => r.GetTemplateAnalysisClosureAsync(It.IsAny<int>(), It.IsAny<CancellationToken>()))
            .Returns((int templateId, CancellationToken ct) =>
            {
                closureRequests?.Add(templateId);
                return TemplateAnalysisClosure.CollectAsync(
                    templateId,
                    async (ids, token) =>
                    {
                        var all = await repoMock.Object.GetAllTemplatesForAnalysisAsync(token)
                                  ?? Array.Empty<Template>();
                        return all.Where(t => ids.Contains(t.Id)).ToList();
                    },
                    ct);
            });
    }
}