cs-codex-dist-tests/DistTestCore/CodexStarter.cs

84 lines
3.0 KiB
C#
Raw Normal View History

2023-04-12 14:12:04 +00:00
using DistTestCore.Codex;
using KubernetesWorkflow;
2023-04-12 14:06:04 +00:00
namespace DistTestCore
{
2023-04-18 11:45:48 +00:00
public class CodexStarter : BaseStarter
2023-04-12 14:06:04 +00:00
{
2023-04-13 12:36:17 +00:00
public CodexStarter(TestLifecycle lifecycle, WorkflowCreator workflowCreator)
2023-04-18 11:45:48 +00:00
: base(lifecycle, workflowCreator)
2023-04-12 14:06:04 +00:00
{
}
public List<CodexNodeGroup> RunningGroups { get; } = new List<CodexNodeGroup>();
2023-04-13 07:33:10 +00:00
public ICodexNodeGroup BringOnline(CodexSetup codexSetup)
2023-04-12 14:06:04 +00:00
{
LogSeparator();
2023-04-18 11:45:48 +00:00
LogStart($"Starting {codexSetup.Describe()}...");
2023-04-14 08:51:35 +00:00
var gethStartResult = lifecycle.GethStarter.BringOnlineMarketplaceFor(codexSetup);
var startupConfig = new StartupConfig();
startupConfig.Add(codexSetup);
startupConfig.Add(gethStartResult);
2023-04-14 07:54:07 +00:00
2023-04-14 08:51:35 +00:00
var containers = StartCodexContainers(startupConfig, codexSetup.NumberOfNodes, codexSetup.Location);
2023-04-13 12:36:17 +00:00
var metricAccessFactory = lifecycle.PrometheusStarter.CollectMetricsFor(codexSetup, containers);
2023-04-12 14:06:04 +00:00
2023-04-14 08:51:35 +00:00
var codexNodeFactory = new CodexNodeFactory(lifecycle, metricAccessFactory, gethStartResult.MarketplaceAccessFactory);
2023-04-12 14:12:04 +00:00
2023-04-13 12:36:17 +00:00
var group = CreateCodexGroup(codexSetup, containers, codexNodeFactory);
LogEnd($"Started {codexSetup.NumberOfNodes} nodes at '{group.Containers.RunningPod.Ip}'. They are: [{string.Join(",", group.Select(n => n.GetName()))}]");
LogSeparator();
return group;
2023-04-12 14:06:04 +00:00
}
public void BringOffline(CodexNodeGroup group)
2023-04-13 09:07:36 +00:00
{
2023-04-18 11:45:48 +00:00
LogStart($"Stopping {group.Describe()}...");
2023-04-13 09:07:36 +00:00
var workflow = CreateWorkflow();
workflow.Stop(group.Containers);
RunningGroups.Remove(group);
2023-04-18 11:45:48 +00:00
LogEnd("Stopped.");
2023-04-13 09:07:36 +00:00
}
2023-04-12 14:06:04 +00:00
public void DeleteAllResources()
{
2023-04-13 09:07:36 +00:00
var workflow = CreateWorkflow();
2023-04-12 14:06:04 +00:00
workflow.DeleteAllResources();
RunningGroups.Clear();
2023-04-12 14:06:04 +00:00
}
2023-04-13 09:07:36 +00:00
2023-04-13 09:30:19 +00:00
public void DownloadLog(RunningContainer container, ILogHandler logHandler)
{
var workflow = CreateWorkflow();
workflow.DownloadContainerLog(container, logHandler);
}
2023-04-13 12:36:17 +00:00
2023-04-14 08:51:35 +00:00
private RunningContainers StartCodexContainers(StartupConfig startupConfig, int numberOfNodes, Location location)
2023-04-13 12:36:17 +00:00
{
var workflow = CreateWorkflow();
2023-04-14 08:51:35 +00:00
return workflow.Start(numberOfNodes, location, new CodexContainerRecipe(), startupConfig);
2023-04-13 12:36:17 +00:00
}
private CodexNodeGroup CreateCodexGroup(CodexSetup codexSetup, RunningContainers runningContainers, CodexNodeFactory codexNodeFactory)
{
var group = new CodexNodeGroup(lifecycle, codexSetup, runningContainers, codexNodeFactory);
RunningGroups.Add(group);
return group;
}
2023-04-13 09:30:19 +00:00
2023-04-13 09:07:36 +00:00
private StartupWorkflow CreateWorkflow()
{
return workflowCreator.CreateWorkflow();
}
private void LogSeparator()
{
Log("----------------------------------------------------------------------------");
}
2023-04-12 14:06:04 +00:00
}
}