fix tests

packages/apps/marmalade-marketplace/src/__tests__/pages/mytokens.test.tsx

@@ -16,6 +16,9 @@ const mockGetTokens = vi.mocked(getTokens);
 vi.mock('@/components/Token', () => ({
   Token: ({ tokenId }: { tokenId: string }) => <div>{tokenId}</div>,
 }));
+vi.mock('@/utils/fetchOnSaleTokens', () => ({
+  fetchOnSaleTokens: console.log,
+}));
 
 const dummyAccountContext = {
   account: null,

packages/apps/marmalade-marketplace/src/pages/mytokens/index.tsx

@@ -2,7 +2,8 @@ import { ListingHeader } from '@/components/ListingHeader';
 import { Token } from '@/components/Token';
 import { getTokens, NonFungibleTokenBalance } from '@/graphql/queries/client';
 import { useAccount } from '@/hooks/account';
-import { database } from '@/utils/firebase';
+import { fetchOnSaleTokens } from '@/utils/fetchOnSaleTokens';
+
 import { ChainId } from '@kadena/client';
 import {
   Badge,
@@ -12,7 +13,7 @@ import {
   ProgressCircle,
   Stack,
 } from '@kadena/kode-ui';
-import { collection, getDocs, query, where } from 'firebase/firestore';
+
 import { useEffect, useState } from 'react';
 
 const MyTokens = () => {
@@ -23,24 +24,6 @@ const MyTokens = () => {
   const [isLoading, setIsLoading] = useState<boolean>(false);
   const { account } = useAccount();
 
-  const fetchOnSaleTokens = async (accountName?: string) => {
-    if (accountName) {
-      //const querySnapshot = await getDocs(query(collection(database, 'sales')));
-      const salesRef = collection(database, 'sales');
-      const q = await query(
-        salesRef,
-        where('seller.account', '==', account?.accountName),
-      );
-      const querySnapshot = await getDocs(q);
-
-      const docs: NonFungibleTokenBalance[] = [];
-      querySnapshot.forEach((doc) => {
-        docs.push(doc.data() as NonFungibleTokenBalance);
-      });
-      setOnSaleTokens(docs);
-    }
-  };
-
   const fetchTokens = async (accountName?: string) => {
     if (accountName) {
       const tokens = await getTokens(accountName);
@@ -52,7 +35,7 @@ const MyTokens = () => {
   useEffect(() => {
     setIsLoading(true);
     fetchTokens(account?.accountName);
-    fetchOnSaleTokens(account?.accountName);
+    fetchOnSaleTokens(account?.accountName!, setOnSaleTokens);
   }, [account?.accountName]);
 
   return (

packages/apps/marmalade-marketplace/src/utils/fetchOnSaleTokens.ts

@@ -0,0 +1,22 @@
+import { NonFungibleTokenBalance } from '@/graphql/queries/client';
+import { database } from '@/utils/firebase';
+import { collection, getDocs, query, where } from 'firebase/firestore';
+import { Dispatch, SetStateAction } from 'react';
+
+export const fetchOnSaleTokens = async (
+  accountName: string,
+  cb: Dispatch<SetStateAction<NonFungibleTokenBalance[]>>,
+) => {
+  if (accountName) {
+    //const querySnapshot = await getDocs(query(collection(database, 'sales')));
+    const salesRef = collection(database, 'sales');
+    const q = await query(salesRef, where('seller.account', '==', accountName));
+    const querySnapshot = await getDocs(q);
+
+    const docs: NonFungibleTokenBalance[] = [];
+    querySnapshot.forEach((doc) => {
+      docs.push(doc.data() as NonFungibleTokenBalance);
+    });
+    cb(docs);
+  }
+};