import { describe, expect, it } from "vitest";
import { CLEAN_ANVIL_CHAIN } from "../../../test/src/chains.js";
import { TEST_CLIENT } from "../../../test/src/test-clients.js";
import { defineChain } from "../../chains/utils.js";
import { fetchPublishedContractMetadata } from "../../contract/deployment/publisher.js";
import {
  getAllDefaultConstructorParamsForImplementation,
  getRequiredTransactions,
} from "./get-required-transactions.js";

describe.runIf(process.env.TW_SECRET_KEY)(
  "getRequiredTransactions",
  {
    timeout: 120000,
  },
  () => {
    it("should count transactions for a published direct deploy contract", async () => {
      const deployMetadata = await fetchPublishedContractMetadata({
        client: TEST_CLIENT,
        contractId: "AccountFactory",
      });
      const results = await getRequiredTransactions({
        chain: CLEAN_ANVIL_CHAIN,
        client: TEST_CLIENT,
        deployMetadata,
      });
      expect(results.length).toBe(1);
    });

    it("should count transactions for a published autofactory contract", async () => {
      const deployMetadata = await fetchPublishedContractMetadata({
        client: TEST_CLIENT,
        contractId: "Airdrop",
      });
      const results = await getRequiredTransactions({
        chain: CLEAN_ANVIL_CHAIN,
        client: TEST_CLIENT,
        deployMetadata,
      });
      expect(results.length).toBe(4);
    });

    it("should count transactions for a modular contract with modules", async () => {
      const deployMetadata = await fetchPublishedContractMetadata({
        client: TEST_CLIENT,
        contractId: "ERC721CoreInitializable",
      });
      const results = await getRequiredTransactions({
        chain: CLEAN_ANVIL_CHAIN,
        client: TEST_CLIENT,
        deployMetadata,
        modules: await Promise.all(
          ["ClaimableERC721", "BatchMetadataERC721", "RoyaltyERC721"].map(
            (contractId) =>
              fetchPublishedContractMetadata({
                client: TEST_CLIENT,
                contractId,
              }).then((m) => ({
                deployMetadata: m,
              })),
          ),
        ),
      });
      expect(results.length).toBe(7);
    });

    it("should count transactions for a dynamic contract", async () => {
      const deployMetadata = await fetchPublishedContractMetadata({
        client: TEST_CLIENT,
        contractId: "EvolvingNFT",
      });
      const results = await getRequiredTransactions({
        chain: CLEAN_ANVIL_CHAIN,
        client: TEST_CLIENT,
        deployMetadata,
      });

      expect(results.length).toBe(8);
    });

    it("should return default constructor params for zksync chains", async () => {
      const params = await getAllDefaultConstructorParamsForImplementation({
        chain: defineChain(300),
        client: TEST_CLIENT,
      });

      expect(params.nativeTokenWrapper).toBeDefined();
    });
  },
);