//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();
//        }
//    }
//}