Gangs/GangsTest/StatTests/ManageTests/StatManagerData.cs

using System.Collections;
using GangsAPI;
using Mock;
using SQLImpl;
using SQLite;

namespace GangsTest.StatTests.ManageTests;

public class StatManagerData : IEnumerable<object[]> {
  private readonly IBehavior[] behaviors = [
    new MockStatManager(),
    new SQLStatManager(
      Environment.GetEnvironmentVariable("DB_CONNECTION_STRING")
      ?? "Host=localhost;User=root;Database=gangs", "gang_unit_test", true),
    new SQLiteStatManager("Data Source=:memory:", "gang_unit_test", true)
  ];

  public StatManagerData() {
    foreach (var behavior in behaviors) behavior.Start();
  }

  public IEnumerator<object[]> GetEnumerator() {
    return behaviors.Select(behavior => (object[]) [behavior]).GetEnumerator();
  }

  IEnumerator IEnumerable.GetEnumerator() { return GetEnumerator(); }
}