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

using DistTestCore.CodexLogsAndMetrics;
using KubernetesWorkflow;
using Logging;

namespace DistTestCore
{
    public class TestLifecycle
    {
        private readonly WorkflowCreator workflowCreator;

        public TestLifecycle(Configuration configuration)
        {
            Log = new TestLog(configuration.GetLogConfig());
            workflowCreator = new WorkflowCreator(configuration.GetK8sConfiguration());

            FileManager = new FileManager(Log, configuration);
            CodexStarter = new CodexStarter(this, workflowCreator);
            PrometheusStarter = new PrometheusStarter(this, workflowCreator);
        }

        public TestLog Log { get; }
        public FileManager FileManager { get; }
        public CodexStarter CodexStarter { get; }
        public PrometheusStarter PrometheusStarter { get; }

        public void DeleteAllResources()
        {
            CodexStarter.DeleteAllResources();
            FileManager.DeleteAllTestFiles();
        }

        public ICodexNodeLog DownloadLog(OnlineCodexNode node)
        {
            var subFile = Log.CreateSubfile();
            var description = node.Describe();
            var handler = new LogDownloadHandler(description, subFile);

            Log.Log($"Downloading logs for {description} to file {subFile.FilenameWithoutPath}");
            CodexStarter.DownloadLog(node.CodexAccess.Container, handler);

            return new CodexNodeLog(subFile);
        }
    }
}