using BenchmarkDotNet.Attributes;
using Speckle.Core.Models;
using Speckle.Core.Serialisation;

namespace Speckle.Core.Tests.Performance.Serialisation;

[MemoryDiagnoser]
[RegressionTestConfig(1, 1, 6)]
public class DeserializationWorkerThreads : IDisposable
{
  public static IEnumerable<int> NumThreadsToTest => Enumerable.Range(0, Environment.ProcessorCount + 1);

  [Params(0, 9)]
  public int DataComplexity { get; set; }

  private TestDataHelper _dataSource;

  [GlobalSetup]
  public async Task Setup()
  {
    _dataSource = new TestDataHelper();
    await _dataSource.SeedTransport(DataComplexity).ConfigureAwait(false);
  }

  [Benchmark]
  [ArgumentsSource(nameof(NumThreadsToTest))]
  public Base RunTest(int numThreads)
  {
    BaseObjectDeserializerV2 sut = new() { WorkerThreadCount = numThreads, ReadTransport = _dataSource.Transport };
    return sut.Deserialize(_dataSource.Transport.GetObject(_dataSource.ObjectId)!);
  }

  [GlobalCleanup]
  public virtual void Cleanup()
  {
    Dispose();
  }

  public void Dispose()
  {
    _dataSource.Dispose();
  }
}