74 lines
2.0 KiB
C#
74 lines
2.0 KiB
C#
using NUnit.Framework;
|
|
|
|
namespace CodexDistTestCore
|
|
{
|
|
public interface IPodLogHandler
|
|
{
|
|
void Log(Stream log);
|
|
}
|
|
|
|
[AttributeUsage(AttributeTargets.Method, AllowMultiple = false)]
|
|
public class DontDownloadLogsAndMetricsOnFailureAttribute : PropertyAttribute
|
|
{
|
|
public DontDownloadLogsAndMetricsOnFailureAttribute()
|
|
: base(Timing.UseLongTimeoutsKey)
|
|
{
|
|
}
|
|
}
|
|
|
|
public class PodLogDownloader
|
|
{
|
|
public const string DontDownloadLogsOnFailureKey = "DontDownloadLogsOnFailure";
|
|
|
|
private readonly TestLog log;
|
|
private readonly IK8sManager k8SManager;
|
|
|
|
public PodLogDownloader(TestLog log, IK8sManager k8sManager)
|
|
{
|
|
this.log = log;
|
|
k8SManager = k8sManager;
|
|
}
|
|
|
|
public CodexNodeLog DownloadLog(OnlineCodexNode node)
|
|
{
|
|
var description = node.Describe();
|
|
var subFile = log.CreateSubfile();
|
|
|
|
log.Log($"Downloading logs for {description} to file {subFile.FilenameWithoutPath}");
|
|
var handler = new PodLogDownloadHandler(description, subFile);
|
|
k8SManager.FetchPodLog(node, handler);
|
|
return handler.CreateCodexNodeLog();
|
|
}
|
|
}
|
|
|
|
public class PodLogDownloadHandler : IPodLogHandler
|
|
{
|
|
private readonly string description;
|
|
private readonly LogFile log;
|
|
|
|
public PodLogDownloadHandler(string description, LogFile log)
|
|
{
|
|
this.description = description;
|
|
this.log = log;
|
|
}
|
|
|
|
public CodexNodeLog CreateCodexNodeLog()
|
|
{
|
|
return new CodexNodeLog(log);
|
|
}
|
|
|
|
public void Log(Stream stream)
|
|
{
|
|
log.Write($"{description} -->> {log.FilenameWithoutPath}");
|
|
log.WriteRaw(description);
|
|
var reader = new StreamReader(stream);
|
|
var line = reader.ReadLine();
|
|
while (line != null)
|
|
{
|
|
log.WriteRaw(line);
|
|
line = reader.ReadLine();
|
|
}
|
|
}
|
|
}
|
|
}
|