import { describe, it, expect, beforeEach, afterEach } from "vitest";
import { orchestratorConfig } from "./orchestrator.config";

describe("orchestratorConfig", () => {
  const originalEnv = process.env;

  beforeEach(() => {
    process.env = { ...originalEnv };
  });

  afterEach(() => {
    process.env = originalEnv;
  });

  describe("sandbox.enabled", () => {
    it("should be enabled by default when SANDBOX_ENABLED is not set", () => {
      delete process.env.SANDBOX_ENABLED;

      const config = orchestratorConfig();

      expect(config.sandbox.enabled).toBe(true);
    });

    it("should be enabled when SANDBOX_ENABLED is set to 'true'", () => {
      process.env.SANDBOX_ENABLED = "true";

      const config = orchestratorConfig();

      expect(config.sandbox.enabled).toBe(true);
    });

    it("should be disabled only when SANDBOX_ENABLED is explicitly set to 'false'", () => {
      process.env.SANDBOX_ENABLED = "false";

      const config = orchestratorConfig();

      expect(config.sandbox.enabled).toBe(false);
    });

    it("should be enabled for any other value of SANDBOX_ENABLED", () => {
      process.env.SANDBOX_ENABLED = "yes";

      const config = orchestratorConfig();

      expect(config.sandbox.enabled).toBe(true);
    });

    it("should be enabled when SANDBOX_ENABLED is empty string", () => {
      process.env.SANDBOX_ENABLED = "";

      const config = orchestratorConfig();

      expect(config.sandbox.enabled).toBe(true);
    });
  });

  describe("other config values", () => {
    it("should use default port when ORCHESTRATOR_PORT is not set", () => {
      delete process.env.ORCHESTRATOR_PORT;

      const config = orchestratorConfig();

      expect(config.port).toBe(3001);
    });

    it("should use provided port when ORCHESTRATOR_PORT is set", () => {
      process.env.ORCHESTRATOR_PORT = "4000";

      const config = orchestratorConfig();

      expect(config.port).toBe(4000);
    });

    it("should use default valkey config when not set", () => {
      delete process.env.VALKEY_HOST;
      delete process.env.VALKEY_PORT;
      delete process.env.VALKEY_URL;

      const config = orchestratorConfig();

      expect(config.valkey.host).toBe("localhost");
      expect(config.valkey.port).toBe(6379);
      expect(config.valkey.url).toBe("redis://localhost:6379");
    });
  });

  describe("spawner config", () => {
    it("should use default maxConcurrentAgents of 20 when not set", () => {
      delete process.env.MAX_CONCURRENT_AGENTS;

      const config = orchestratorConfig();

      expect(config.spawner.maxConcurrentAgents).toBe(20);
    });

    it("should use provided maxConcurrentAgents when MAX_CONCURRENT_AGENTS is set", () => {
      process.env.MAX_CONCURRENT_AGENTS = "50";

      const config = orchestratorConfig();

      expect(config.spawner.maxConcurrentAgents).toBe(50);
    });

    it("should handle MAX_CONCURRENT_AGENTS of 10", () => {
      process.env.MAX_CONCURRENT_AGENTS = "10";

      const config = orchestratorConfig();

      expect(config.spawner.maxConcurrentAgents).toBe(10);
    });
  });
});