using NUnit.Framework;

namespace DistTestCore
{
    public class AutoBootstrapDistTest : DistTest
    {
        public override IOnlineCodexNode SetupCodexBootstrapNode(Action<ICodexSetup> setup)
        {
            throw new Exception("AutoBootstrapDistTest creates and attaches a single bootstrap node for you. " +
                "If you want to control the bootstrap node from your test, please use DistTest instead.");
        }

        public override ICodexNodeGroup SetupCodexNodes(int numberOfNodes, Action<ICodexSetup> setup)
        {
            var codexSetup = CreateCodexSetup(numberOfNodes);
            setup(codexSetup);
            codexSetup.WithBootstrapNode(BootstrapNode);
            return BringOnline(codexSetup);
        }

        [SetUp]
        public void SetUpBootstrapNode()
        {
            var setup = CreateCodexSetup(1).WithName("BOOTSTRAP");
            BootstrapNode = BringOnline(setup)[0];
        }

        protected IOnlineCodexNode BootstrapNode { get; private set; } = null!;
    }
}