cs-codex-dist-tests/DistTestCore/Metrics/MetricsAccess.cs

using DistTestCore.Helpers;
using KubernetesWorkflow;
using Logging;
using NUnit.Framework;
using NUnit.Framework.Constraints;
using Utils;

namespace DistTestCore.Metrics
{
    public interface IMetricsAccess
    {
        void AssertThat(string metricName, IResolveConstraint constraint, string message = "");
    }

    public class MetricsAccess : IMetricsAccess
    {
        private readonly BaseLog log;
        private readonly ITimeSet timeSet;
        private readonly MetricsQuery query;
        private readonly RunningContainer node;

        public MetricsAccess(BaseLog log, ITimeSet timeSet, MetricsQuery query, RunningContainer node)
        {
            this.log = log;
            this.timeSet = timeSet;
            this.query = query;
            this.node = node;
        }

        public void AssertThat(string metricName, IResolveConstraint constraint, string message = "")
        {
            AssertHelpers.RetryAssert(constraint, () =>
            {
                var metricSet = GetMetricWithTimeout(metricName);
                var metricValue = metricSet.Values[0].Value;

                log.Log($"{node.Name} metric '{metricName}' = {metricValue}");
                return metricValue;
            }, message);
        }

        public Metrics? GetAllMetrics()
        {
            return query.GetAllMetricsForNode(node);
        }

        private MetricsSet GetMetricWithTimeout(string metricName)
        {
            var start = DateTime.UtcNow;

            while (true)
            {
                var mostRecent = GetMostRecent(metricName);
                if (mostRecent != null) return mostRecent;
                if (DateTime.UtcNow - start > timeSet.WaitForMetricTimeout())
                {
                    Assert.Fail($"Timeout: Unable to get metric '{metricName}'.");
                    throw new TimeoutException();
                }

                Time.Sleep(TimeSpan.FromSeconds(2));
            }
        }

        private MetricsSet? GetMostRecent(string metricName)
        {
            var result = query.GetMostRecent(metricName, node);
            if (result == null) return null;
            return result.Sets.LastOrDefault();
        }
    }

    public class MetricsUnavailable : IMetricsAccess
    {
        public void AssertThat(string metricName, IResolveConstraint constraint, string message = "")
        {
            Assert.Fail("Incorrect test setup: Metrics were not enabled for this group of Codex nodes. Add 'EnableMetrics()' after 'SetupCodexNodes()' to enable it.");
            throw new InvalidOperationException();
        }
    }
}