feat: close vault action

update tests
metaplex-foundation · Dec 1, 2021 · 8f84d48 · 8f84d48
1 parent 7b616c3
commit 8f84d48
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 10 deletions.
diff --git a/test/actions/closeVault.test.ts b/test/actions/closeVault.test.ts
@@ -20,6 +20,8 @@ describe('closing a Vault', () => {
     });
 
     test('closes vault', async () => {
+
+      let vault;
       const vaultResponse = await createVault({
         connection,
         wallet,
@@ -28,7 +30,9 @@ describe('closing a Vault', () => {
       });
 
       await pause(20000);
-      expect(await Vault.load(connection, vaultResponse.vault)).toHaveProperty('data');
+      vault = await Vault.load(connection, vaultResponse.vault);
+      expect(vault).toHaveProperty('data');
+      expect(vault.data.state).toEqual(0);
 
       closeVault({
         connection,
@@ -40,8 +44,9 @@ describe('closing a Vault', () => {
 
       await pause(20000);
 
-      console.log('OK');
-      expect(await Vault.load(connection, vaultResponse.vault)).toHaveProperty('data');
+      vault = await Vault.load(connection, vaultResponse.vault);
+      expect(vault).toHaveProperty('data');
+      expect(vault.data.state).toEqual(2);
     });
   });
 });
diff --git a/test/actions/createVault.test.ts b/test/actions/createVault.test.ts
@@ -27,8 +27,9 @@ describe('creating a Vault', () => {
       });
 
       await pause(20000);
-
-      expect(await Vault.load(connection, vaultResponse.vault)).toHaveProperty('data');
+      const vault = await Vault.load(connection, vaultResponse.vault);
+      expect(vault).toHaveProperty('data');
+      expect(vault.data.state).toEqual(0);
     });
   });
 });