Mock useToken and useFeeToken hooks

Author: JunichiSugiura

packages/utils/src/hooks/tokens.mock.ts

@@ -1,6 +1,7 @@
 import { tokensByAddress, tokensBySymbol } from "@cartridge/utils/mock/data";
 import { fn, Mock } from "@storybook/test";
 import * as actual from "./tokens";
+import { getChecksumAddress } from "starknet";
 
 export * from "./tokens";
 
@@ -11,6 +12,24 @@ export const useTokens: Mock<() => actual.UseTokensResponse> = fn(() => ({
   register: () => {},
 })).mockName("useTokens");
 
-export const useToken: Mock<() => actual.UseTokenResponse> = fn(
-  actual.useToken,
+export const useToken: Mock<(address: string) => actual.UseTokenResponse> = fn(
+  (address) => {
+    const { tokens, isLoading, error } = useTokens();
+    const token = tokens[getChecksumAddress(address)];
+
+    return {
+      token,
+      isLoading,
+      error,
+    };
+  },
 ).mockName("useToken");
+
+export const useFeeToken: Mock<() => actual.UseFeeTokenResponse> = fn(() => {
+  const { feeToken, isLoading, error } = useTokens();
+  return {
+    token: feeToken,
+    isLoading,
+    error,
+  };
+}).mockName("useFeeToken");

packages/utils/src/hooks/tokens.ts

@@ -30,7 +30,13 @@ export function useToken(address: string): UseTokenResponse {
   };
 }
 
-export function useFeeToken() {
+export type UseFeeTokenResponse = {
+  token: ERC20 | undefined;
+  isLoading: boolean;
+  error?: Error;
+};
+
+export function useFeeToken(): UseFeeTokenResponse {
   const { feeToken, isLoading, error } = useTokens();
   return {
     token: feeToken,