fix sdk-kit e2e test

rooch-network · Dec 20, 2024 · f0c71ae · f0c71ae
1 parent 4a7bbb2
commit f0c71ae
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 34 deletions.
diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml
diff --git a/sdk/typescript/rooch-sdk-kit/package.json b/sdk/typescript/rooch-sdk-kit/package.json
@@ -79,7 +79,8 @@
     "react-dom": "^18.3.1",
     "size-limit": "^11.1.5",
     "typescript": "^5.6.2",
-    "vite": "^5.4.4"
+    "vite": "^5.4.4",
+    "vitest": "2.1.0"
   },
   "dependencies": {
     "@roochnetwork/rooch-sdk": "workspace:*",

diff --git a/sdk/typescript/rooch-sdk-kit/test/case/use-default-client.test.tsx b/sdk/typescript/rooch-sdk-kit/test/case/use-default-client.test.tsx
@@ -17,7 +17,7 @@ import {
   useConnectWallet,
   useCreateSessionKey,
   useCurrentSession,
-  useSession,
+  useSessions,
 } from '../../src/index.js'
 
 describe('useDefaultClient', () => {
@@ -38,7 +38,7 @@ describe('useDefaultClient', () => {
     function useTestHook() {
       const { mutateAsync: connectWallet } = useConnectWallet()
       const { mutateAsync: createSessionKey } = useCreateSessionKey()
-      const sessions = useSession()
+      const sessions = useSessions()
       const currentSession = useCurrentSession()
       const networks = { test: { url: 'http://localhost:6767' } }
       const defaultClient = useDefaultClient({ currentNetwork: 'test', networks })

diff --git a/sdk/typescript/rooch-sdk-kit/test/mocks/mock-wallet.ts b/sdk/typescript/rooch-sdk-kit/test/mocks/mock-wallet.ts
@@ -18,6 +18,7 @@ import {
 import { SupportChain } from '../../src/feature/index.js'
 import { Wallet } from '../../src/wellet/wallet.js'
 import { Mock, vi } from 'vitest'
+import { WalletNetworkType } from '../../src/index.js'
 
 export class MockBitcoinWallet extends Wallet {
   private kp: Keypair
@@ -55,8 +56,8 @@ export class MockBitcoinWallet extends Wallet {
     return 'mock'
   }
 
-  getNetwork(): string {
-    return 'testnet'
+  getNetwork(): Promise<WalletNetworkType> {
+    return Promise.resolve('testnet')
   }
 
   getPublicKey(): PublicKey<Address> {
@@ -67,7 +68,7 @@ export class MockBitcoinWallet extends Wallet {
     return this.kp.getRoochAddress()
   }
 
-  getSupportNetworks(): string[] {
+  getSupportNetworks(): WalletNetworkType[] {
     return ['testnet', 'livenet']
   }
 
@@ -98,7 +99,9 @@ export class MockBitcoinWallet extends Wallet {
 
   switchAccount(_: string): void {}
 
-  switchNetwork(_: string): void {}
+  switchNetwork(_: string): Promise<void> {
+    return Promise.resolve()
+  }
 
   getDescription(): string {
     return ''
@@ -111,4 +114,16 @@ export class MockBitcoinWallet extends Wallet {
   getInstallUrl(): string {
     return ''
   }
+
+  sendBtc(input: {
+    toAddress: string
+    satoshis: number
+    options?: { feeRate: number }
+  }): Promise<string> {
+    throw new Error('Method not implemented.')
+  }
+
+  getBalance(): Promise<{ confirmed: number; unconfirmed: number; total: string }> {
+    throw new Error('Method not implemented.')
+  }
 }