@layr-labs/ecloud-cli 1.0.0-devep7 → 1.0.0-devep8

package/dist/commands/billing/__tests__/top-up.test.js

@@ -312,7 +312,8 @@ async function createBillingClient(flags) {
     walletClient,
     publicClient,
     environment,
-    skipTelemetry: true
+    skipTelemetry: true,
+    privateKey
   });
 }
 
@@ -411,6 +412,11 @@ var BillingTopUp = class _BillingTopUp extends Command {
       default: "compute",
       options: ["compute"],
       env: "ECLOUD_PRODUCT_ID"
+    }),
+    chain: Flags2.string({
+      required: false,
+      description: "Blockchain network for USDC payment: ethereum or base",
+      options: ["ethereum", "base"]
     })
   };
   async run() {
@@ -454,14 +460,25 @@ ${chalk2.bold("Purchase EigenCompute credits")}`);
     });
   }
   async handleUsdc(billing, flags, walletAddress, targetAccount, baselineTotal) {
-    const onChainState = await billing.getTopUpInfo();
+    let selectedChain = "ethereum";
+    if (billing.hasBaseSupport()) {
+      selectedChain = flags.chain ?? await select2({
+        message: "Which network?",
+        choices: [
+          { value: "ethereum", name: "Ethereum" },
+          { value: "base", name: "Base" }
+        ]
+      });
+    }
+    const onChainState = await billing.getTopUpInfo({ chain: selectedChain });
     const { usdcBalance, minimumPurchase } = onChainState;
     const balanceFormatted = formatUnits(usdcBalance, 6);
     this.log(`  ${chalk2.bold("USDC:")}    ${balanceFormatted} USDC`);
     if (usdcBalance === BigInt(0)) {
+      const networkName = selectedChain === "base" ? "Base Sepolia" : "Sepolia";
       this.log(`
 ${chalk2.yellow("  No USDC in wallet.")}`);
-      this.log(`  Send USDC on Sepolia to: ${chalk2.cyan(walletAddress)}`);
+      this.log(`  Send USDC on ${networkName} to: ${chalk2.cyan(walletAddress)}`);
       this.log(`  Then re-run: ${chalk2.cyan("ecloud billing top-up")}
 `);
       return;
@@ -494,7 +511,8 @@ ${chalk2.yellow("  No USDC in wallet.")}`);
   Purchasing ${chalk2.bold(`$${amountFloat.toFixed(2)}`)} in credits...`);
     const { txHash } = await billing.topUp({
       amount: amountRaw,
-      account: targetAccount
+      account: targetAccount,
+      chain: selectedChain
     });
     this.log(`  ${chalk2.green("\u2713")} Transaction confirmed: ${txHash}`);
     await this.pollForCredits(billing, flags, baselineTotal, amountFloat);
@@ -619,8 +637,10 @@ describe("ecloud billing top-up", () => {
       getTopUpInfo: vi.fn(),
       topUp: vi.fn(),
       getPaymentMethods: vi.fn(),
-      purchaseCredits: vi.fn()
+      purchaseCredits: vi.fn(),
+      hasBaseSupport: vi.fn()
     };
+    mockBilling.hasBaseSupport.mockReturnValue(false);
     createBillingClient.mockResolvedValue(mockBilling);
     input3.mockResolvedValue("50");
   });
@@ -680,7 +700,8 @@ describe("ecloud billing top-up", () => {
     expect(fullOutput).toContain("$60.00");
     expect(mockBilling.topUp).toHaveBeenCalledWith({
       amount: BigInt(5e7),
-      account: WALLET_ADDRESS
+      account: WALLET_ADDRESS,
+      chain: "ethereum"
     });
   });
   it("zero USDC balance: exits with fund wallet message", async () => {
@@ -715,7 +736,8 @@ describe("ecloud billing top-up", () => {
     expect(fullOutput).toContain(targetAccount);
     expect(mockBilling.topUp).toHaveBeenCalledWith({
       amount: BigInt(5e7),
-      account: targetAccount
+      account: targetAccount,
+      chain: "ethereum"
     });
   });
   it("billing API poll timeout: shows timeout message", async () => {
@@ -758,6 +780,68 @@ describe("ecloud billing top-up", () => {
     expect(fullOutput).toContain("Transaction confirmed");
     expect(fullOutput).toContain("Credits haven't appeared yet");
   });
+  it("usdc: prompts for chain selection when Base is available", async () => {
+    mockBilling.hasBaseSupport.mockReturnValue(true);
+    setupOnChainState();
+    mockBilling.topUp.mockResolvedValue({ txHash: TX_HASH, walletAddress: WALLET_ADDRESS });
+    mockBilling.getStatus.mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 10 }).mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 60 });
+    select3.mockResolvedValue("base");
+    const cmd = createCommand({ amount: "50", method: "usdc" });
+    const promise = cmd.run();
+    for (let i = 0; i < 10; i++) {
+      await vi.advanceTimersByTimeAsync(5e3);
+    }
+    await promise;
+    expect(select3).toHaveBeenCalledWith(
+      expect.objectContaining({
+        message: "Which network?",
+        choices: expect.arrayContaining([
+          expect.objectContaining({ value: "base" }),
+          expect.objectContaining({ value: "ethereum" })
+        ])
+      })
+    );
+    expect(mockBilling.getTopUpInfo).toHaveBeenCalledWith({ chain: "base" });
+    expect(mockBilling.topUp).toHaveBeenCalledWith({
+      amount: BigInt(5e7),
+      account: WALLET_ADDRESS,
+      chain: "base"
+    });
+  });
+  it("usdc: skips chain prompt when Base is not configured", async () => {
+    mockBilling.hasBaseSupport.mockReturnValue(false);
+    setupOnChainState();
+    mockBilling.topUp.mockResolvedValue({ txHash: TX_HASH, walletAddress: WALLET_ADDRESS });
+    mockBilling.getStatus.mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 10 }).mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 60 });
+    const cmd = createCommand({ amount: "50", method: "usdc" });
+    const promise = cmd.run();
+    for (let i = 0; i < 10; i++) {
+      await vi.advanceTimersByTimeAsync(5e3);
+    }
+    await promise;
+    expect(mockBilling.topUp).toHaveBeenCalledWith({
+      amount: BigInt(5e7),
+      account: WALLET_ADDRESS,
+      chain: "ethereum"
+    });
+  });
+  it("usdc: --chain flag skips network prompt", async () => {
+    mockBilling.hasBaseSupport.mockReturnValue(true);
+    setupOnChainState();
+    mockBilling.topUp.mockResolvedValue({ txHash: TX_HASH, walletAddress: WALLET_ADDRESS });
+    mockBilling.getStatus.mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 10 }).mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 60 });
+    const cmd = createCommand({ amount: "50", method: "usdc", chain: "base" });
+    const promise = cmd.run();
+    for (let i = 0; i < 10; i++) {
+      await vi.advanceTimersByTimeAsync(5e3);
+    }
+    await promise;
+    expect(mockBilling.topUp).toHaveBeenCalledWith({
+      amount: BigInt(5e7),
+      account: WALLET_ADDRESS,
+      chain: "base"
+    });
+  });
   it("credit card: charges selected card on file", async () => {
     mockBilling.getStatus.mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 10 }).mockResolvedValueOnce({ subscriptionStatus: "active", remainingCredits: 35 });
     mockBilling.getPaymentMethods.mockResolvedValue({