use testing package for container creation

BirthdayResearch · Jan 12, 2022 · add7b3a · add7b3a
1 parent c802e5b
commit add7b3a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/jellyfish-api-core/__tests__/category/wallet/listWallets.test.ts b/packages/jellyfish-api-core/__tests__/category/wallet/listWallets.test.ts
@@ -1,3 +1,4 @@
+import { Testing } from '@defichain/jellyfish-testing'
 import { MasterNodeRegTestContainer, RegTestContainer } from '@defichain/testcontainers'
 import { ContainerAdapterClient } from '../../container_adapter_client'
 
@@ -20,19 +21,18 @@ describe('Wallet without masternode', () => {
 })
 
 describe('Wallet with masternode', () => {
-  const container = new MasterNodeRegTestContainer()
-  const client = new ContainerAdapterClient(container)
+  const { container, rpc: { wallet } } = Testing.create(new MasterNodeRegTestContainer())
 
   beforeAll(async () => {
     await container.start()
-    await client.wallet.createWallet('test')
+    await wallet.createWallet('test')
   })
 
   afterAll(async () => {
     await container.stop()
   })
 
   it('should listWallets', async () => {
-    await expect(client.wallet.listWallets()).resolves.toEqual(['', 'test'])
+    await expect(wallet.listWallets()).resolves.toEqual(['', 'test'])
   })
 })