import { expect, Locator, test } from "@playwright/test";
import {
  CommandBarButton,
  DataExplorer,
  ONE_MINUTE_MS,
  TEST_AUTOSCALE_MAX_THROUGHPUT_RU_2K,
  TEST_MANUAL_THROUGHPUT_RU_2K,
  TestAccount,
} from "../../fx";
import { createTestSQLContainer, TestContainerContext } from "../../testData";

test.describe("Autoscale throughput", () => {
  let context: TestContainerContext = null!;
  let explorer: DataExplorer = null!;

  test.beforeAll("Create Test Database & Open Scale tab", async ({ browser }) => {
    context = await createTestSQLContainer();
    const page = await browser.newPage();
    explorer = await DataExplorer.open(page, TestAccount.SQL);

    // Click Scale & Settings and open Scale tab
    await explorer.openScaleAndSettings(context);
    const scaleTab = explorer.frame.getByTestId("settings-tab-header/ScaleTab");
    await scaleTab.click();

    await switchManualToAutoscaleThroughput();
  });

  if (!process.env.CI) {
    test.afterAll("Delete Test Database", async () => {
      await context?.dispose();
    });
  }


  test("Update autoscale max throughput", async () => {
    // Update autoscale max throughput
    await getThroughputInput(explorer, "autopilot").fill(TEST_AUTOSCALE_MAX_THROUGHPUT_RU_2K.toString());

    // Save
    await explorer.commandBarButton(CommandBarButton.Save).click();

    // Read console message
    await expect(explorer.getConsoleHeaderStatus()).toContainText(
      `Successfully updated offer for collection ${context.container.id}`,
      {
        timeout: 2 * ONE_MINUTE_MS,
      },
    );
  });

  test("Update autoscale max throughput passed allowed limit", async () => {
    // Get soft allowed max throughput and remove commas
    const softAllowedMaxThroughputString = await explorer.frame
      .getByTestId("soft-allowed-maximum-throughput")
      .innerText();
    const softAllowedMaxThroughput = Number(softAllowedMaxThroughputString.replace(/,/g, ""));

    // Try to set autoscale max throughput above allowed limit
    await getThroughputInput(explorer, "autopilot").fill((softAllowedMaxThroughput * 10).toString());
    await expect(explorer.commandBarButton(CommandBarButton.Save)).toBeDisabled();
    await expect(getThroughputInputErrorMessage(explorer, "autopilot")).toContainText(
      "This update isn't possible because it would increase the total throughput",
    );
  });

  test("Update autoscale max throughput with invalid increment", async () => {
    // Try to set autoscale max throughput with invalid increment
    await getThroughputInput(explorer, "autopilot").fill("1100");
    await expect(explorer.commandBarButton(CommandBarButton.Save)).toBeDisabled();
    await expect(getThroughputInputErrorMessage(explorer, "autopilot")).toContainText(
      "Throughput value must be in increments of 1000",
    );
  });

  const switchManualToAutoscaleThroughput = async (): Promise<void> => {
    const autoscaleRadioButton = explorer.frame.getByText("Autoscale", { exact: true });
    await autoscaleRadioButton.click();
    await expect(explorer.commandBarButton(CommandBarButton.Save)).toBeEnabled();
    await explorer.commandBarButton(CommandBarButton.Save).click();
    await expect(explorer.getConsoleHeaderStatus()).toContainText(
      `Successfully updated offer for collection ${context.container.id}`,
      {
        timeout: 2 * ONE_MINUTE_MS,
      },
    );
  };
});

test.describe("Manual throughput", () => {
  let context: TestContainerContext = null!;
  let explorer: DataExplorer = null!;

  test.beforeAll("Create Test Database & Open scale tab", async ({ browser }) => {
    context = await createTestSQLContainer();
    const page = await browser.newPage();
    explorer = await DataExplorer.open(page, TestAccount.SQL);

    // Click Scale & Settings and open Scale tab
    await explorer.openScaleAndSettings(context);
    const scaleTab = explorer.frame.getByTestId("settings-tab-header/ScaleTab");
    await scaleTab.click();
  });

  test.afterAll("Delete Test Database", async () => {
    await context?.dispose();
  });

  test("Update manual throughput", async () => {
    await getThroughputInput(explorer, "manual").fill(TEST_MANUAL_THROUGHPUT_RU_2K.toString());
    await explorer.commandBarButton(CommandBarButton.Save).click();
    await expect(explorer.getConsoleHeaderStatus()).toContainText(
      `Successfully updated offer for collection ${context.container.id}`,
      {
        timeout: 2 * ONE_MINUTE_MS,
      },
    );
  });

  test("Update manual throughput passed allowed limit", async () => {
    // Get soft allowed max throughput and remove commas
    const softAllowedMaxThroughputString = await explorer.frame
      .getByTestId("soft-allowed-maximum-throughput")
      .innerText();
    const softAllowedMaxThroughput = Number(softAllowedMaxThroughputString.replace(/,/g, ""));

    // Try to set manual throughput above allowed limit
    await getThroughputInput(explorer, "manual").fill((softAllowedMaxThroughput * 10).toString());
    await expect(explorer.commandBarButton(CommandBarButton.Save)).toBeDisabled();
    await expect(getThroughputInputErrorMessage(explorer, "manual")).toContainText(
      "This update isn't possible because it would increase the total throughput",
    );
  });
});

// Helper methods
const getThroughputInput = (explorer: DataExplorer, type: "manual" | "autopilot"): Locator => {
  return explorer.frame.getByTestId(`${type}-throughput-input`);
};

const getThroughputInputErrorMessage = (explorer: DataExplorer, type: "manual" | "autopilot"): Locator => {
  return explorer.frame.getByTestId(`${type}-throughput-input-error`);
};